五子棋问题

NowCoder最近爱上了五子棋。
现在给你一个棋局，请你帮忙判断其中有没有五子连珠（超过五颗也算）。
输入描述：
输入有多组数据，每组数据为一张20x20的棋盘。
其中黑子用“*”表示，白子用“+”表示，空白位置用“.”表示。
输出描述：
如果棋盘上存在五子连珠（无论哪种颜色的棋子），输入“Yes”，否则输出“No”。

解题思路

结构设计：dir代表当前位置的8个方向，其中上下、左右、左上右下、右上左下为必须放在一起检测。
获取一个棋盘，按照行列检测棋盘中的每个位置，当拿到一个位置后，按照以下步骤进行操作：
以该位置为中心，依次检测该位置的上下、左右、左上右下、右上左下，比如左上。
从该位置开始向上检测，找出连在一起的同种棋子个数，再向下检测同种棋子的个数并累计，注意在检测时，中心位置统计了2次，上下统计完时，需要给结果减去1。
按照2统计完上下、左右、左上右下、右上左下各个方向，找出最大的同种棋子个数。
检测3中统计出的最大同种棋子个数，如果大于等于5，输出Yes，否则取下一个位置继续1。
如果所有的位置检测完，没有超过5个的相同棋子，则输出No。

#include <iostream>
#include <fstream>
#include <string>

using namespace std;

#define N 20
int count(string table[], char ch, int x, int y){
	int maxc = 0;
	int dir[4][2][2] = { {{ -1,0 },{ 1,0 }},{{ 0,-1 },{ 0,1 }},{{ -1,-1 },{ 1,1 }},{{ -1,1
	},{ 1,-1 }} };
	for (int i = 0; i < 4; ++i) { // 四种方向
		int c = 0;
		for (int j = 0; j < 2; ++j) {// 两个小方向
			int nx = x, ny = y;
			while (nx >= 0 && nx < N && ny >= 0 && ny < N && table[nx][ny] == ch){
				nx += dir[i][j][0];
				ny += dir[i][j][1];
				++c;
			}
		}
		maxc = (maxc > c ? maxc : c);
	}
	return maxc - 1; //统计两个方向（如横向的左右两个方向）的时候，当前棋子被计算了两次
}
bool solve(string table[]){
	for (int i = 0; i < N; ++i){
		for (int j = 0; j < N; ++j){
			if (table[i][j] == '*' || table[i][j] == '+')
				if (count(table, table[i][j], i, j) >= 5)
					return true;
		}
	}
	return false;
}
int main(){
	string table[N];
	while (cin >> table[0]){
		for (int i = 1; i < N; ++i) cin >> table[i];
		cout << (solve(table) ? "Yes" : "No") << endl;
	}
	return 0;
}