CCF之消除类游戏

这个题真心累呀，出现了各种问题，最大的问题就是代码不简洁！！！边界值处理的不好！、
注意：memset函数：作用是在一段内存块中填充某个给定的值，它是对较大的结构体或数组进行清零操作的一种最快方法。
对于二维数组int a[n][m]清零,如下：
memset(a,0,sizeof(a));

如果对数组全都赋1，是不对的。
如果用memset(a,1,sizeof(a))，就是对a指向的内存的所有字节进行赋值，每个字节都用数字1(实际上是二进制0000 0001)去填充，。一个int类型的元素占4个字节，拼接在一起是0000 0001,0000 0001,0000 0001,0000 0001，转化成十六进制就是0x01010101，就等于16843009，这就完成了对一个int类型的元素的赋值了。

#include <iostream>
#include <string.h>
using namespace std;
int main()
{
   int n,m,a[31][31],flags[31][31],flag=1;
   cin>>n>>m;
    for(int i=0;i<n;i++)
  for(int j=0;j<m;j++)
    flags[i][j]=1;
   for(int i=0;i<n;i++)
   for(int j=0;j<m;j++)
        cin>>a[i][j];
   //消除行
   for(int i=0;i<n;i++)
   {
       for(int j=0;j<m;j++)
       {
           if(a[i][j]==a[i][j+1]&&j<m-1)
                flag++;
            else
            {
                if(flag>=3)
                {
                    while(flag)
                    {
                    flags[i][j+1-flag]=0;
                    flag--;
                    }
                }
               flag=1;
            }
       }
   }
    //消除列
    flag=1;
   for(int i=0;i<m;i++)
   {
       for(int j=0;j<n;j++)
       {
           if(a[j][i]==a[j+1][i]&&j<n-1)//边界
                flag++;
            else
            {
                if(flag>=3)
                {
                    while(flag)    //边界
                    {
                    flags[j+1-flag][i]=0;
                    flag--;
                    }
                }
                flag=1;
            }
       }
   }
   for(int i=0;i<n;i++)
   for(int j=0;j<m;j++)
        if(flags[i][j]==0)
            a[i][j]=0;
   for(int i=0;i<n;i++)
   {
       for(int j=0;j<m-1;j++)
       cout<<a[i][j]<<" ";
       cout<<a[i][m-1]<<endl;
   }
   return 0;
}

对memset函数的理解可以参考：
http://baike.baidu.com/link?url=XtBeuHIoji3-9MN0QBzW-lcaHtSz5L-i4CpOXTpyBFpANb1nbOetuKUGhHq0NuXNjyvR8P4zRda6yOzCOG7Wsa