C++ - 連立方程式解法(ガウス・ジョルダン(ピボット選択)法)!

Updated:


先日は、連立方程式を「ガウス・ジョルダン法」で解くアルゴリズムを C++ で実装したことを紹介しました。

今回は、連立方程式を「ガウス・ジョルダン法」を応用した「ガウス・ジョルダン(ピボット選択)法」で解くアルゴリズムを C++ で実装してみました。

以下、簡単な説明と C++ ソースコードの紹介です。

0. 前提条件

  • Linux Mint 14 Nadia (64bit) での作業を想定。
  • g++ (Ubuntu/Linaro 4.7.2-2ubuntu1) 4.7.2

1. ガウス・ジョルダン(ピボット選択)法による連立方程式の解法について(簡単に)

(数式が多いので、\(\LaTeX\) で記載)

GAUSS_JORDEN_PIVOT

2. C++ ソース作成

File: gauss_jorden_pivot.cpp

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/*********************************************
 * 連立方程式の解法 ( ガウス・ジョルダン法(ピボット選択法) )
 *********************************************/
#include <iostream>  // for cout
#include <math.h>    // for fabs()
#include <stdio.h>   // for printf()
#include <stdlib.h>  // for exit()

// 元の数定義
#define N 3  // 4

using namespace std;

/*
 * 計算クラス
 */
class Calc
{
    double a[N][N + 1];  // 係数用配列

    // 各種変数
    double p, d;         // ピボット係数、ピボット行x係数
    double max, dummy;   // 最大絶対値、入れ替え時ダミー
    int i, j, k;         // LOOP インデックス
    int s;               // 入替行

    public:
        // 連立方程式を解く(ガウス・ジョルダン法(ピボット選択法))
        void calcGaussJordenPivot();
};

/*
 * 連立方程式を解く(ガウス・ジョルダン法(ピボット選択法))
 */
void Calc::calcGaussJordenPivot()
{
    // 係数
    static double a[N][N + 1] = {
        { 2.0, -3.0,  1.0,  5.0},
        { 1.0,  1.0, -1.0,  2.0},
        { 3.0,  5.0, -7.0,  0.0}
        //{ 1.0, -2.0,  3.0, -4.0,  5.0},
        //{-2.0,  5.0,  8.0, -3.0,  9.0},
        //{ 5.0,  4.0,  7.0,  1.0, -1.0},
        //{ 9.0,  7.0,  3.0,  5.0,  4.0}
    };

    // 元の連立方程式をコンソール出力
    for (i = 0; i < N; i++) {
        for (j = 0; j < N; j++)
            printf("%+fx%d ", a[i][j], j + 1);
        printf("= %+f\n", a[i][N]);
    }

    for (k = 0; k < N; k++) {
        // 行入れ替え
        max = 0; s = k;
        for (j = k; j < N; j++) {
            if (fabs(a[j][k]) > max) {
                max = fabs(a[j][k]);
                s = j;
            }
        }
        if (max == 0) {
            printf("解けない!");
            exit(1);
        }
        for (j = 0; j <= N; j++) {
            dummy   = a[k][j];
            a[k][j] = a[s][j];
            a[s][j] = dummy;
        }

        // ピボット係数
        p = a[k][k];

        // ピボット行を p で除算
        for (j = k; j < N + 1; j++)
            a[k][j] /= p;

        // ピボット列の掃き出し
        for (i = 0; i < N; i++) {
            if (i != k) {
                d = a[i][k];
                for (j = k; j < N + 1; j++)
                    a[i][j] -= d * a[k][j];
            }
        }
    }

    // 結果出力
    for (k = 0; k < N; k++)
        printf("x%d = %f\n", k + 1, a[k][N]);
}

/*
 * メイン処理
 */
int main()
{
    try
    {
        // 計算クラスインスタンス化
        Calc objCalc;

        // 連立方程式を解く(ガウス・ジョルダン法(ピボット選択法))
        objCalc.calcGaussJordenPivot();
    }
    catch (...) {
        cout << "例外発生!" << endl;
        return -1;
    }

    // 正常終了
    return 0;
}

3. C++ ソースコンパイル

-Wall は警告出力、-O2 最適化のオプション)

$ g++ -Wall -O2 -o gauss_jorden_pivot gauss_jorden_pivot.cpp

何も出力されなければ成功。

4. 実行

実際に、次の連立方程式を解いてみる。

EQUATION_1

$ ./gauss_jorden_pivot
+2.000000x1 -3.000000x2 +1.000000x3 = +5.000000
+1.000000x1 +1.000000x2 -1.000000x3 = +2.000000
+3.000000x1 +5.000000x2 -7.000000x3 = +0.000000
x1 = 3.000000
x2 = 1.000000
x3 = 2.000000

コンソールには元の連立方程式と解が出力される。


通常のガウス・ジョルダン法で連立方程式を解くより精度は向上するはずです。(特に解が実数になるような場合に)

※ちなみに、以前 C 言語によるアルゴリズムに関する書物を参考に作成していた C 言語プログラムを、 C++ に移植した形態となっています。

以上。





 

Sponsored Link

 

Comments