3 x 3 的幻方是一个填充有从 1 到 9 的不同数字的 3 x 3 矩阵,其中每行,每列以及两条对角线上的各数之和都相等。
给定一个由整数组成的 grid,其中有多少个 3 × 3 的 “幻方” 子矩阵?(每个子矩阵都是连续的)。
示例:
输入: [[4,3,8,4],
[9,5,1,9],
[2,7,6,2]]
输出: 1
解释:
下面的子矩阵是一个 3 x 3 的幻方:
438
951
276
而这一个不是:
384
519
762
总的来说,在本示例所给定的矩阵中只有一个 3 x 3 的幻方子矩阵。
方法
class Solution {
public int numMagicSquaresInside(int[][] grid) {
int count=0;
int row=grid.length,col=grid[0].length;
for(int i=0;i<=row-3;i++){
for(int j=0;j<=col-3;j++){
if(judge33(grid,i,j)){
count++;
}
}
}
return count;
}
public boolean judge33(int[][] grid,int row,int col){
//判断是否为1-9
Set<Integer> set=new HashSet<Integer>();
for(int i=row;i<row+3;i++){
for(int j=col;j<col+3;j++){
if(set.contains(grid[i][j]) ||grid[i][j]<0 ||grid[i][j]>9){
System.out.println("执行了");
return false ;
}else{
set.add(grid[i][j]);
}
}
}
//判断各数之和
if(grid[row][col]+grid[row][col+1]+grid[row][col+2]!=15){
return false;
}
if(grid[row+1][col]+grid[row+1][col+1]+grid[row+1][col+2]!=15){
return false;
}
if(grid[row+2][col]+grid[row+2][col+1]+grid[row+2][col+2]!=15){
return false;
}
if(grid[row][col]+grid[row+1][col]+grid[row+2][col]!=15){
return false;
}
if(grid[row][col+1]+grid[row+1][col+1]+grid[row+2][col+1]!=15){
return false;
}
if(grid[row][col+2]+grid[row+1][col+2]+grid[row+2][col+2]!=15){
return false;
}
if(grid[row][col]+grid[row+1][col+1]+grid[row+2][col+2]!=15){
return false;
}
if(grid[row][col+2]+grid[row+1][col+1]+grid[row+2][col]!=15){
return false;
}
return true;
}
}