CCF CSP JAVA消除类游戏

import java.util.Scanner;
public class Main {
    public static void main(String[] args){
        Scanner input=new Scanner(System.in);
        int row=input.nextInt();
        int col=input.nextInt();
        int a[][]=new int[row][col];
        int b[][]=new int[row][col];
        for(int i=0;i<row;i++){
            for(int j=0;j<col;j++){
                a[i][j]=input.nextInt();
            }
        }
        int con=0;
        for(int i=0;i<row;i++){
            for(int j=0;j<col-1;j++){
                if(a[i][j]==a[i][j+1]){
                    if(j==0){
                        con=0;
                    }
                    con++;
                    if(con>=2){
                    b[i][j-1]=1;
                    b[i][j]=1;
                    b[i][j+1]=1;
                    }
                }else {
                    con=0;
                }
            }
        }
        con=0;
        for(int j=0;j<col;j++){
            for(int i=0;i<row-1;i++){
                if(a[i][j]==a[i+1][j]){
                    if(i==0){
                        con=0;
                    }
                    con++;
                    if(con>=2){
                        b[i-1][j]=1;
                        b[i][j]=1;
                        b[i+1][j]=1;
                    }
                }else {
                    con=0;
                }
            }
        }
        for(int i=0;i<row;i++){
            for(int j=0;j<col;j++){
                if(b[i][j]==1){
                    a[i][j]=0;
                }
                System.out.print(a[i][j]+" ");
            }
            System.out.println();
        }

    }

}