CSP考试 2015年12月第2题 消除类游戏 C语言实现

#include <stdio.h>
int main()
{
	int NUM1;
	int NUM2;
	scanf("%d",&NUM1);
	scanf("%d",&NUM2);

	int a[NUM1][NUM2];
	int b[NUM1][NUM2];

	int i;
	int j;
	for(i=0;i<NUM1;i++)
	{
		for(j=0;j<NUM2;j++)
		{
			scanf("%d",&a[i][j]);
		}
	}

	for(i=0;i<NUM1;i++)
	{
		for(j=0;j<NUM2;j++)
		{
			b[i][j]=1;
		}
	}

	for(i=0;i<NUM1;i++)
	{
		for(j=0;j<NUM2;j++)
		{
			if(((j+2)<NUM2)&&(a[i][j]==a[i][j+1])&&(a[i][j]==a[i][j+2]))
			{
				b[i][j]=0;
				b[i][j+1]=0;
				b[i][j+2]=0;
			}
			if(((i+2)<NUM1)&&(a[i][j]==a[i+1][j])&&(a[i][j]==a[i+2][j]))
			{
				b[i][j]=0;
				b[i+1][j]=0;
				b[i+2][j]=0;
			}
		}
	}
	for(i=0;i<NUM1;i++)
	{
		for(j=0;j<NUM2;j++)
		{
			if(b[i][j]==0)
			{
				a[i][j]=0;
			}
		}
	}
	//printf("\n");
	for(i=0;i<NUM1;i++)
	{
		for(j=0;j<NUM2;j++)
		{
			printf("%d ",a[i][j]);
		}
		printf("\n");
	}
	return 0;
}