算法筆記_087:藍橋杯練習 9-1九宮格（Java）

目錄

1 問題描述

2 解決方案

 

---

1 問題描述

問題描述

　　九宮格。輸入1-9這9個數字的一種任意排序，構成3*3二維數組。如果每行、每列以及對角線之和都相等，打印1。否則打印0。

樣例輸出

與上面的樣例輸入對應的輸出。
例：

數據規模和約定

　　輸入1-9這9個數字的一種任意排序。

 

---

2 解決方案

 

具體代碼如下：

import java.util.Scanner;

public class Main {
    
    public boolean isOK(int[] A) {
        int sum = A[0] + A[1] + A[2];
        //判斷行
        for(int i = 3;i <= 6;i = i + 3) {
            if(sum != A[i] + A[i + 1] + A[i + 2])
                return false;
        }
        //判斷列
        for(int i = 0;i <= 2;i++) {
            if(sum != A[i] + A[i + 3] + A[i + 6])
                return false;
        }
        //判斷對角線
        if(sum != A[0] + A[4] + A[8])
            return false;
        if(sum != A[2] + A[4] +A[6])
            return false;
        return true;
    }
    
    public static void main(String[] args) {
        Main test = new Main();
        Scanner in = new Scanner(System.in);
        int[] A = new int[9];
        for(int i = 0;i < 9;i++)
            A[i] = in.nextInt();
        if(test.isOK(A)) {
            System.out.print("1");
        } else {
            System.out.print("0");
        }
    }
}