2021年度训练联盟热身训练赛第四场 B Connect3 棋盘状态 DFS暴力搜索

【2021年度训练联盟热身训练赛第四场】
B Connect3 （棋盘状态 DFS暴力搜索）

【题目描述】


【输入描述】

【输出描述】

测试样例1

输入

2
2 3

输出

516

测试样例2

输入

3
4 4

输出

177

【题目分析】

这道题是一个类似三子棋的游戏，双方谁先在一行、一列或对角线上达到连续三个棋子即获胜。只不过和一般下棋的区别是，棋盘是竖着放的，也就是每次放入棋子只能落入当前一列最底下的空位。

题目输入给了黑方第一手放入棋子的列数x（即落在(1,x)处），以及白方最后一手落子的位置(a,b)，要求可能的棋盘布局情况的总数 。

注意，如果在下到最后一步之前游戏已经分出胜负了，则不符合题目要求。即算入总数的情况都要到白方下完最后一个子才恰好结束。

棋盘一共有4*4=16个位置，显然每个位置有空着、黑子、白子三种情况，总情况数并不多，我们直接使用DFS暴力搜索就行。黑白双方交替落子，注意判断当前列是否还能放入、状态是否重复、游戏是否提前结束等，细节较多要考虑清楚。

【AC代码】

#include <iostream>
#include <cstdio>
#include <set>
using namespace std;
int x,a,b,ans=0,qp[5][5],pos[5];
set<int> st;

bool check(int colour)//判断游戏是否结束 
{
	if(qp[1][1]==colour&&qp[1][1]==qp[1][2]&&qp[1][2]==qp[1][3]||qp[1][2]==colour&&qp[1][2]==qp[1][3]&&qp[1][3]==qp[1][4]||
	   qp[2][1]==colour&&qp[2][1]==qp[2][2]&&qp[2][2]==qp[2][3]||qp[2][2]==colour&&qp[2][2]==qp[2][3]&&qp[2][3]==qp[2][4]||
	   qp[3][1]==colour&&qp[3][1]==qp[3][2]&&qp[3][2]==qp[3][3]||qp[3][2]==colour&&qp[3][2]==qp[3][3]&&qp[3][3]==qp[3][4]||
	   qp[4][1]==colour&&qp[4][1]==qp[4][2]&&qp[4][2]==qp[4][3]||qp[4][2]==colour&&qp[4][2]==qp[4][3]&&qp[4][3]==qp[4][4]||
	   qp[1][1]==colour&&qp[1][1]==qp[2][1]&&qp[2][1]==qp[3][1]||qp[2][1]==colour&&qp[2][1]==qp[3][1]&&qp[3][1]==qp[4][1]||
	   qp[1][2]==colour&&qp[1][2]==qp[2][2]&&qp[2][2]==qp[3][2]||qp[2][2]==colour&&qp[2][2]==qp[3][2]&&qp[3][2]==qp[4][2]||
	   qp[1][3]==colour&&qp[1][3]==qp[2][3]&&qp[2][3]==qp[3][3]||qp[2][3]==colour&&qp[2][3]==qp[3][3]&&qp[3][3]==qp[4][3]||
	   qp[1][4]==colour&&qp[1][4]==qp[2][4]&&qp[2][4]==qp[3][4]||qp[2][4]==colour&&qp[2][4]==qp[3][4]&&qp[3][4]==qp[4][4]||
	   qp[1][1]==colour&&qp[1][1]==qp[2][2]&&qp[2][2]==qp[3][3]||qp[2][2]==colour&&qp[2][2]==qp[3][3]&&qp[3][3]==qp[4][4]||
	   qp[1][2]==colour&&qp[1][2]==qp[2][3]&&qp[2][3]==qp[3][4]||qp[2][1]==colour&&qp[2][1]==qp[3][2]&&qp[3][2]==qp[4][3]||
	   qp[1][4]==colour&&qp[1][4]==qp[2][3]&&qp[2][3]==qp[3][2]||qp[2][3]==colour&&qp[2][3]==qp[3][2]&&qp[3][2]==qp[4][1]||
	   qp[1][3]==colour&&qp[1][3]==qp[2][2]&&qp[2][2]==qp[3][1]||qp[2][4]==colour&&qp[2][4]==qp[3][3]&&qp[3][3]==qp[4][2]) 
		return 1;
	return 0;
}

int state()//状态编码，每格有三种情况：空、黑、白 
{
	int temp=0;
	for(int i=1;i<=4;i++)
	for(int j=1;j<=4;j++)
		temp=temp*3+qp[i][j];
	return temp;
}

void dfs(int colour)//搜索可能的情况 
{
	int sta=state();
	if(st.count(sta)) return;//状态重复 
	st.insert(sta);
	
	if(qp[a][b]==2)//到达最终状态 
	{
		if(check(2))
			ans++;
		return;
	}
	if(check(1)||check(2)||qp[a][b]!=0)
		return;

	for(int i=1;i<=4;i++)
	{
		if(pos[i]==4)//当前列不能再放入石块 
			continue;
		if(colour==1)
		{
			pos[i]++;
			qp[pos[i]][i]=1;
			dfs(2);
			qp[pos[i]][i]=0;
			pos[i]--;
		}
		if(colour==2)
		{
			pos[i]++;
			qp[pos[i]][i]=2;
			dfs(1);
			qp[pos[i]][i]=0;
			pos[i]--;
		}
	}
}

int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>x>>a>>b;
	qp[1][x]=1;
	pos[x]++;//在x列放入黑色石块 
	dfs(2);//搜索 
	cout<<ans<<endl;
	return 0;
}