CF 118D Caesar‘s Legions(记忆化搜索)

CF 118D Caesar‘s Legions(记忆化搜索)_第1张图片
CF 118D Caesar‘s Legions(记忆化搜索)_第2张图片
CF 118D Caesar‘s Legions(记忆化搜索)_第3张图片
直接上代码和注释:

//记忆化搜索
//题意:
//有n个步兵,m个骑兵,步兵最多连续站k1个,骑兵最多连续站k2个,问多少种站法
#include

using namespace std;
#define int long long
const int N = 110, mod = 100000000;
int memo[2][N][N];//记忆化数组 记录步兵剩余j个 骑兵剩余k个 最后为i兵种 的方案数
int n, m, k1, k2;
int ans;

int dfs(int x)//x代表当前这个位置准备放什么
{
	int& res = memo[x][n][m];	//简化写法,设置一个记忆化数组的引用变量
	if (~res) {//若当前不为空则返回已经有的数值
		return res;
	}
	if (!n && !m) return res = 1;	//若两者均为空,则表示有一种情况了
	res = 0;
	int cnt = (!x ? min(n, k1) : min(m, k2));
	for (int i = 1; i <= cnt; ++i) {//循环枚举这个位置连续放多少个相同的士兵
		int& one = (!x ? n : m);
		one -= i;
		res += dfs(x ^ 1), res %= mod;
		one += i;
	}
	return res;
}

signed main()
{
	cin >> n >> m >> k1 >> k2;
	memset(memo, -1, sizeof memo);
	//将所有方案分为两类,0表示第一个位置即将放置footman,1表示第一个位置即将放置horseman
	ans = (dfs(0) + dfs(1)) % mod;
	cout << ans << '\n';
	
	return 0;
}

你可能感兴趣的:(动态规划,深度优先,动态规划)