wikioi1004 四子连棋 解题报告

http://codevs.cn/problem/1004/

为了锻炼自己弱不禁风的代码能力，挑了一道搜索来做。

思路很简单，数据范围也很小，根本无需任何优化。

两个问题：

轮流走棋。我做了两遍BFS来解决。

步数的记录。我觉得在结构体里开一个变量记录这个很浪费，是愚蠢的行为。然后发现大家都这么写……有更简单的方法吗？

详见代码：

#include<cstdio>
#include<queue>
#include<algorithm>
#include<set>
using namespace std;
//1 black 0 kong -1 white
const int p=40700017;
struct node{int a[5][5],cnt,now;};//cnt步数，now先手

queue<node> q;//STL依赖症晚期
node st;
set<int> s;
int ans=0x3f3f3f3f;

int hash(node x){
	int ret=0;
	int sd=5;
	for(int i=1;i<=4;i++)
		for(int j=1;j<=4;j++){
			ret=(ret+(x.a[i][j]+31)*sd*(4*i+j))%p;//自己乱写的怪异的hash函数
			sd=(5*sd)%p;
		}
	return ret;
}
	
bool is(node x){//丑爆了
	for(int i=1;i<=4;i++)
	if (x.a[i][1]==x.a[i][2]&&x.a[i][2]==x.a[i][3]&&x.a[i][3]==x.a[i][4]) return 1;
	for(int j=1;j<=4;j++)
	if (x.a[1][j]==x.a[2][j]&&x.a[2][j]==x.a[3][j]&&x.a[3][j]==x.a[4][j]) return 1;
	if (x.a[1][1]==x.a[2][2]&&x.a[2][2]==x.a[3][3]&&x.a[3][3]==x.a[4][4]) return 1;
	if (x.a[1][4]==x.a[2][3]&&x.a[2][3]==x.a[3][2]&&x.a[3][2]==x.a[4][1]) return 1;
	return 0;
}

void swap(int &x,int &y){
	int tmp=x;
	x=y;
	y=tmp;
}

void bfs(node st){
	q.push(st);
	s.insert(hash(st));
	while(!q.empty()){
		node k=q.front(),k1=q.front(),k2=q.front(),
		k3=q.front(),k4=q.front();
		
		q.pop();
		/*调试信息
		for(int i=1;i<=4;i++){
			for(int j=1;j<=4;j++)
				printf("%d ",k.a[i][j]);
			printf("\n");
		}
		printf("\n");
		*/
		for(int i=4;i>=1;i--)//这里十分奇特：我正着枚举WA三个点，倒着枚举就A了……如果我还有心情修改的话就会把两个空位单独存放下来
			for(int j=4;j>=1;j--)//同上
				if (k.a[i][j]==0) {//四个方向蠢到写四遍……复制粘贴时还出错了-_-
					if (i>1&&k1.now==k1.a[i-1][j])
					{
						swap(k1.a[i][j],k1.a[i-1][j]);
						k1.now=-k1.now;
						k1.cnt++;
						if (is(k1)){ans=min(ans,k1.cnt); return;}
						if (s.find(hash(k1))==s.end()){q.push(k1);s.insert(hash(k1));}
					}
					if (i<4&&k2.now==k2.a[i+1][j])
					{
						swap(k2.a[i][j],k2.a[i+1][j]);
						k2.now=-k2.now;
						k2.cnt++;
						if (is(k2)){ans=min(ans,k2.cnt); return;}
						if (s.find(hash(k2))==s.end()){q.push(k2);s.insert(hash(k2));}
					}
					if (j>1&&k3.now==k3.a[i][j-1])
					{
						swap(k3.a[i][j],k3.a[i][j-1]);
						k3.now=-k3.now;
						k3.cnt++;
						if (is(k3)){ans=min(ans,k3.cnt); return;}
						if (s.find(hash(k3))==s.end()){q.push(k3);s.insert(hash(k3));}
					}
					if (j<4&&k4.now==k4.a[i][j+1])
					{
						swap(k4.a[i][j],k4.a[i][j+1]);
						k4.now=-k4.now;
						k4.cnt++;
						if (is(k4)){ans=min(ans,k4.cnt); return;}
						if (s.find(hash(k4))==s.end()){q.push(k4);s.insert(hash(k4));}
					}
				}
	}
}

bool equal(node x,node y){
	for(int i=1;i<=4;i++)
		for(int j=1;j<=4;j++)
			if (x.a[i][j]!=y.a[i][j]) return false;
	return true;
}

int main(){
	for(int i=1;i<=4;i++){
		char s[4];
		scanf("%s",s);
		for(int j=0;j<4;j++)
			if (s[j]=='B') st.a[i][j+1]=1;
			else if(s[j]=='W') st.a[i][j+1]=-1;
			else st.a[i][j+1]=0;
	}

	st.cnt=0;
	st.now=-1;
	bfs(st);
	while(!q.empty()) q.pop();
	s.clear();
	st.now=1;
	bfs(st);//两次BFS，先手不同
	printf("%d",ans);
}