小美的平衡矩阵 java代码

1.

小美的平衡矩阵

小美拿到了一个

n∗n的矩阵，其中每个元素是 0 或者 1。

小美认为一个矩形区域是完美的，当且仅当该区域内 0 的数量恰好等于 1 的数量。

现在，小美希望你回答有多少个i∗i

的完美矩形区域。你需要回答

1≤1≤i≤n

的所有答案。

输入描述：

第一行输入一个正整数，代表矩阵大小。接下来的行，每行输入一个长度为的 01 串，用来表示矩阵。

输出描述：

输出行，第行输出的完美矩形区域的数量。

示例1

输入例子：

4
1010
0101
1100
0011

输出例子：

0
7
0
1

import java.util.Scanner; 

import java.util.Arrays;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n=in.nextInt();
        int[][] nums=new int[n+2][n+2];
        for(int i=1;i<=n;++i){
            String line=in.next();
            for(int j=1;j<=n;++j){
                if(line.charAt(j-1)=='0') nums[i][j]=0;
                else nums[i][j]=1;
                nums[i][j]+=nums[i-1][j]+nums[i][j-1]-nums[i-1][j-1];
            }
        }

        for(int size=1;size<=n;++size){
            int count=0;
            if(size%2!=0) {
                System.out.println(0);
                continue;
            }
            for(int i=1;i+size-1<=n;++i){
                for(int j=1;j+size-1<=n;++j){
                    int temp=nums[i+size-1][j+size-1]-nums[i+size-1][j-1]
							-nums[i-1][j+size-1]+nums[i-1][j-1];
                    if(temp==size*size/2) ++count;
                }
            }
            System.out.println(count);
        }
      
    }
}