【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;
}