CCF201512-2消除类游戏

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
int main(int argc, char** argv) 
{
    int n,m;
    cin>>n>>m;
    int arr[n][m],brr[n][m];
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            cin>>arr[i][j];
        }
    }
    memset(brr,0,sizeof(arr));
    //进行行标记
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m-2;j++)
        {
            if(arr[i][j]==arr[i][j+1]&&arr[i][j+1]==arr[i][j+2])
            {
                brr[i][j]=brr[i][j+1]=brr[i][j+2]=1;
            }
        }
     } 
     //列标记 
    for(int j=0;j<m;j++)
    {
        for(int i=0;i<n-2;i++)
        {
            if(arr[i][j]==arr[i+1][j]&&arr[i+1][j]==arr[i+2][j])
            {
                brr[i][j]=brr[i+1][j]=brr[i+2][j]=1;
             } 
        }
     } 
     //重置矩阵arr 
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(brr[i][j])
            {
                arr[i][j]=0;
            }
        }
    }
    //输出结果
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(j!=0)
                 cout<<" ";
            cout<<arr[i][j];
        }
        cout<<endl;
     } 
    return 0;
}