201512-2 消除类游戏

#include<iostream>
using namespace std;

int main(){
    int n,m,i,j;
    int a[50][50]={0},b[50][50]={0};
    cin>>n>>m;
    for(i=0;i<n;i++){
        for(j=0;j<m;j++){
            cin>>a[i][j];
        }
    }
    for(i=0;i<n;i++){
        for(j=0;j<m-2;j++){
            if(a[i][j]==a[i][j+1] && a[i][j]==a[i][j+2]){
                b[i][j]=1;
                b[i][j+1]=1;
                b[i][j+2]=1;
            }
        }
    }
    for(j=0;j<m;j++){
        for(i=0;i<n-2;i++){
            if(a[i][j]==a[i+1][j] && a[i][j]==a[i+2][j]){
                b[i][j]=1;
                b[i+1][j]=1;
                b[i+2][j]=1;
            }
        }
    }
    for(i=0;i<n;i++){
        for(j=0;j<m;j++){
            if(b[i][j]==1) a[i][j]=0;
        }
    }
    for(i=0;i<n;i++){
        for(j=0;j<m;j++){
            if(j==0) cout<<a[i][j];
            else cout<<" "<<a[i][j];
        }
        cout<<endl;
    }
    return 0;
}