Javaで行列どうしの足し算と引き算をするメソッドを書いてみた

標準ライブラリしか使えない環境のJavaで、行列（２次元配列）を扱わないといけなくなったで以下略です。

というわけで、まずは行列どうしの足し算をするメソッドから。ソースは以下な感じで。

// 行列どうしの足し算の結果を返す public static double[][] additionOfMatrices(double[][] a, double[][] b){ int aY = a.length; int aX = a[0].length; int bY = b.length; int bX = b[0].length; double[][] c = new double[aY][aX]; // 計算できる条件が満たされていないと、初期状態の行列を返す if(aY != bY || aX != bX){ return c; } for(int i=0; i<aY; i++){ for(int j=0; j<aX; j++){ c[i][j] = a[i][j] + b[i][j]; } } return c; }

対して引き算の方はというと、足し算の場合の 20行目の + を - にしてやれば OK。

// 引数の行列 a から引数の行列 b を引いた結果を返す public static double[][] subtractionOfMatrices(double[][] a, double[][] b){ int aY = a.length; int aX = a[0].length; int bY = b.length; int bX = b[0].length; double[][] c = new double[aY][aX]; // 計算できる条件が満たされていないと、初期状態の行列を返す if(aY != bY || aX != bX){ return c; } for(int i=0; i<aY; i++){ for(int j=0; j<aX; j++){ c[i][j] = a[i][j] - b[i][j]; } } return c; }

Enjoy!