五子棋-判断输赢

#include<iostream>
using namespace std;
#define maxn 100
int a[maxn][maxn],n;
int dir[8][2]={1,1,1,-1,-1,-1,-1,1,0,1,0,-1,1,0,-1,0};
int check(int x,int y)
{
	if(x<0||y<0||x>=n||y>=n)
	return 0;
	return 1;
}
int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			cin>>a[i][j];
			 
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			if(!a[i][j])
				continue;
			for(int k=0;k<8;k++)//共8个方向
			{
				int x=i,y=j,ct=1,nx,ny;
				for(int l=0;l<4;l++) //向同一个方向走四步
				{
					nx=x+dir[k][0];
					ny=y+dir[k][1];
					x=nx,y=ny;
					if(check(nx,ny)&&a[i][j]==a[nx][ny])
						ct++;
					else
						break;
				}
				if(ct==5)
				{
					if(a[i][j]==1)
					{
						cout<<"黑棋赢"<<endl; 
						return 0;
					}
					else 
					{
						cout<<"白棋赢"<<endl; 
						return 0;
					}
				}
			}
		}	
	}
	cout<<"不分胜负"<<endl; 
}