【每日一题】乌龟棋

乌龟棋：记忆化搜索

https://ac.nowcoder.com/acm/problem/16590

思路：看数据范围是350，每一步的选择有四种，可以采用dp递推，也可以记忆化搜索。

从图中可以看出到达位置3的时候用了1个1，1个2，可以是先1后2，或者先2后1，但其实是重复了的，所有采用记忆化搜索。

记忆化搜索：设置状态dp[a][b][c][d] 表示用了a个1，b个2，c个3，d个4的最优解。

然后对于每一步搜索四种情况

#include <bits/stdc++.h>
#define ull unsigned long long
#define ll long long
const int inf = 0x3f3f3f3f;
const int mod = 10007;
const int N = 3e5+7;
const ll ds = 1e15+7;
const double PI = 3.141592653589793238462643383;
 
using namespace std;

int cnt[5],dp[125][125][125][125],n,m;
int bb[555];
int dfs(int num,int a,int b,int c,int d)
{
	//int ans = 0;
	if(num >= n) return bb[n];
	if(dp[a][b][c][d]) return dp[a][b][c][d];
	int ans = 0;
	if(cnt[1] > a) ans = max(ans,dfs(num+1,a+1,b,c,d));
	if(cnt[2] > b) ans = max(ans,dfs(num+2,a,b+1,c,d));
	if(cnt[3] > c) ans = max(ans,dfs(num+3,a,b,c+1,d));
	if(cnt[4] > d) ans = max(ans,dfs(num+4,a,b,c,d+1));
	return dp[a][b][c][d] = ans+bb[num];
}

void solve() {
	int x;
	cin >> n >> m;
	for(int i = 1; i <= n; i++) cin >> bb[i];
	for(int i = 1; i <= m; i++) {
		cin >> x;
		cnt[x]++;
	}
	cout << dfs(1,0,0,0,0);
}

int main() { 
//	int t;
//	scanf("%d",&t);
//	while(t--)
		solve();
	return 0;
}