Codeforces 354B 博弈， DP，记忆化搜索

Posted 2022-03-02 pkgunboat

题意：现在有一个字符矩阵，从左上角出发，每个人交替选择一个字符。如果最后字符a数目大于字符b，那么第一个人获胜，否则b获胜，否则平均。现在双方都不放水，问最后结果是什么？

思路：这题需要注意，选择的字符串不是一条单纯的路径，而是包括这个字符串的所有路径的并。

比如:

abc

bac

ccc

ab字符串其实是(1, 2)和(2, 1)的并，abc是(1, 3)和(3, 1)的并。

因为n最大只有20，那么对角线长度最长只有20，我们可以考虑状压，通过状压来更新状态，转移的时候枚举26个字母转移。

代码：

#include <bits/stdc++.h>
#define pii pair<int, int>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 20;
vector<pii> a[maxn * 2];
int id[maxn][maxn];
bool v[40][1 << 20];
int dp[40][1 << 20];
char s[25][25];
int n;
int cal(char ch) 
	if(ch == ‘a‘) return 1;
	else if(ch == ‘b‘) return -1;
	return 0;

int dfs(int deep, int mask) 
	if(deep == n * 2 - 2) return 0;
	if(v[deep][mask]) return dp[deep][mask];
	int sz = a[deep].size(); 
	int ans;
	if(deep & 1) ans = -INF;
	else ans = INF;
	for (char c = ‘a‘; c <= ‘z‘; c++) 
		int add = cal(c);
		int Next_mask = 0;
		for (int j = 0; j < sz; j++) 
			if((mask >> j) & 1) 
				pii tmp = a[deep][j];
				int x = tmp.first, y = tmp.second;
				if(x < n - 1 && s[x + 1][y] == c)
					 Next_mask |= (1 << id[x + 1][y]);
				if(y < n - 1 && s[x][y + 1] == c)
					 Next_mask |= (1 << id[x][y + 1]);
			
		
		if(Next_mask) 
			if(deep & 1) ans = max(ans, add + dfs(deep + 1, Next_mask));
			else ans = min(ans, add + dfs(deep + 1, Next_mask));
		
	
	v[deep][mask] = 1;
	dp[deep][mask] = ans;
	return ans;

int main() 
	scanf("%d", &n);
	for (int i = 0; i < n; i++) 
		scanf("%s", s[i]);
	
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++) 
			id[i][j] = a[i + j].size();
			a[i + j].push_back(make_pair(i, j));
		
	int ans = dfs(0, 1);
	ans += cal(s[0][0]);
	if(ans > 0) printf("FIRST\n");
	else if(ans < 0) printf("SECOND\n");
	else printf("DRAW\n");