JAVA解N元一次方程组(矩阵)

今天写了一个解N元一次方程组的JAVA程序(矩阵),毕竟不是专业搞算法的,用的是最基本的消元法, 模拟人的计算过程,记录一下.

public class Matrix {

 /**
  * @param args
  * 必须为N行N+1列
  */

 public static void main(String[] args) {

  int n = 4;
  float[][] matrix = { { 6, 0, 7, 9, 20 }, { 0, 3, 0, 10, 50 },
    { 0, 4, 0, 11, 31 }, { 0, 8, 9, 0, 10 } };
  

//  float[][] matrix = { { 6, 0, 7, 9, 20 ,99}, { 0, 3, 0, 10, 50 ,56},
//    { 0, 4, 0, 11, 31 ,4}, { 0, 8, 9, 0, 10 ,78} ,{3,6,23,76,34,0}};

  // 消元前
  System.out.println("消元前");
  printMatrix(n, matrix);
  simple(n, matrix);
  // 消元后
  System.out.println("消元后");
  printMatrix(n, matrix);

  getResult(n, matrix);

 }

 private static void simple(int n, float[][] matrix) {
  for (int k = 0; k < n; k++) {
   if (matrix[k][k] == 0) {
    changeRow(n, k, matrix);
   }

   for (int i = 0; i < n; i++) {
    // 记录对角线元素，作为除数
    float temp = matrix[i][k];
    for (int j = 0; j < n + 1; j++) {
     // i<k时,i行已经计算完成
     if (i < k)
      break;
     if (temp == 0)
      continue;
     if (temp != 1) {
      matrix[i][j] /= temp;
     }

     if (i > k)
      matrix[i][j] -= matrix[k][j];
    }
   }
  }

 }

 private static float[] getResult(int n, float[][] matrix) {
  float[] result = new float[n];
  for (int i = n - 1; i >= 0; i--) {
   float temp = matrix[i][n];
   for (int j = n - 1; j >= 0; j--) {
    if (i < j && matrix[i][j] != 0) {
     temp = temp - result[j] * matrix[i][j];
    }
   }
   temp /= matrix[i][i];
   result[i] = temp;

  }

  for (int k = 0; k < result.length; k++) {
   System.out.println("X" + (k + 1) + " = " + result[k]);
  }

  return result;
 }

 // 对角线上元素为０时候和下行交换
 private static void changeRow(int n, int k, float[][] matrix) {
  float[] temp = new float[n + 1];
  // if()
  for (int i = k; i < n; i++) {
   // 已到最后一列,不能继续交换
   if (i + 1 == n && matrix[k][k] == 0) {
    System.out.println("无解或有不唯一解！");
    System.exit(1);
   }

   for (int j = 0; j < n + 1; j++) {
    temp[j] = matrix[k][j];
    matrix[k][j] = matrix[i + 1][j];
    matrix[i + 1][j] = temp[j];
   }
   if (matrix[k][k] != 0)
    return;

  }
 }

 private static void printMatrix(int n, float[][] matrix) {
  // 打印矩阵
  System.out.println("============================================================================");
  for (int i = 0; i < n; i++) {
   for (int j = 0; j < n + 1; j++) {
    if (j == n)
     System.out.print(" = " + matrix[i][j]);
    else
     System.out.print("(" + matrix[i][j] + ") * X" + (j + 1)+ " + ");
   }
   System.out.println();
  }
  System.out.println("============================================================================");
 }
}