战争调度（动态规划、记忆化搜索）

战争调度

题目描述

脸哥最近来到了一个神奇的王国，王国里的公民每个公民有两个下属或者没有下属，这种关系刚好组成一个 n 层的完全二叉树。

公民 i 的下属是 2 * i 和 2 * i +1。最下层的公民即叶子节点的公民是平民，平民没有下属，最上层的是国王，中间是各级贵族。

现在这个王国爆发了战争，国王需要决定每一个平民是去种地以供应粮食还是参加战争，每一个贵族（包括国王自己）是去管理后勤还是领兵打仗。

一个平民会对他的所有直系上司有贡献度，若一个平民 i 参加战争，他的某个直系上司 j 领兵打仗，那么这个平民对上司的作战贡献度为 wij。

若一个平民i 种地，他的某个直系上司 j 管理后勤，那么这个平民对上司的后勤贡献度为 fij，若 i 和 j 所参加的事务不同，则没有贡献度。

为了战争需要保障后勤，国王还要求不多于 m 个平民参加战争。国王想要使整个王国所有贵族得到的贡献度最大，并把这件事交给了脸哥。但不幸的是，脸哥还有很多 deadline 没有完成，他只能把这件事又转交给你。你能帮他安排吗？

输入格式

第一行两个数 n;m。

接下来 2^(n-1) 行，每行n-1 个数，第 i 行表示编号为 2^(n-1)-1+ i 的平民对其n-1直系上司的作战贡献度，其中第一个数表示对第一级直系上司，即编号为 (2^(n-1)-1+ i)/2 的贵族的作战贡献度 wij，依次往上。

接下来 2^(n-1)行，每行n-1个数，第i行表示编号为 2^(n-1)-1+ i的平民对其n-1个直系上司的后勤贡献度，其中第一个数表示对第一级直系上司，即编号为 (2^(n-1)-1+ i)/2 的贵族的后勤贡献度 fij ，依次往上。

输出格式

一行一个数表示满足条件的最大贡献值

解法

这道题从平民角度想，可以想出O(2^1023)的暴力，从官僚角度想，反正我是没想出来。
但是如果我们连起来想，所有所有的贡献其唯一来源都是平民，而平民的贡献是以整个官僚阶梯的结构存在的，每一个平民的这条贡献链都是互不相干独立存在的，当然也就方便直接相加来计算。
这里我是用记忆化搜索做的，递归是从国王开始，但dp实际上是从平民开始算。
dp[ i ][ j ] 表示 “ 以 i 为根节点的子数的所有叶子节点中，满足有 j 个人打仗，然后把所有平民的贡献链相加起来的值 ” 。
dp[ x ][ i ] = max{ dp[ x * 2 ][ j ] + dp[ x * 2 + 1 ][ i - j ] } (j <= i)
算到平民时就是算他~~(她)~~ 的贡献链。

CODE

本蒟蒻代码↓

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#define LL long long
using namespace std;

LL read() {
	LL f = 1,x = 0;char s = getchar();
	while(s < '0' || s > '9') {if(s == '-')f = -1;s = getchar();}
	while(s >= '0' && s <= '9') {x = x * 10 + s - '0';s = getchar();}
	return x * f;
}
LL n,m,i,j,s,o,k,cnt;
int w[1030][2][15];
int dp[1030][1030];
bool f[15];
LL min(LL a,LL b) {
	return a < b ? a : b;
}
void dfs(int x,int y) {
	for(int i = 0;i <= (1<<y);i ++) dp[x][i] = 0;
	if(y == 0) {
		for(int i = 1;i < n;i ++) {
			if(f[i]) dp[x][1] += w[x][1][i];
			else dp[x][0] += w[x][0][i]; 
		}
		return ;
	}
	f[y] = 0;dfs(x * 2,y - 1);dfs(x * 2 + 1,y - 1);
	for(int i = 0;i <= (1<<(y-1));i ++) {
		for(int j = 0;j <= (1<<(y-1));j ++) {
			dp[x][i + j] = max(dp[x][i + j],dp[x * 2][i] + dp[x * 2 + 1][j]);
		}
	}
	f[y] = 1;dfs(x * 2,y - 1);dfs(x * 2 + 1,y - 1);
	for(int i = 0;i <= (1<<(y-1));i ++) {
		for(int j = 0;j <= (1<<(y-1));j ++) {
			dp[x][i + j] = max(dp[x][i + j],dp[x * 2][i] + dp[x * 2 + 1][j]);
		}
	}
}
int main() {
	n = read();k = read();
	cnt = (1<<(n - 1)) - 1;
	for(int i = 1;i <= (1<<(n-1));i ++) {
		for(int j = 1;j < n;j ++) {
			w[i + cnt][1][j] = read();
		}
	}
	int as1 = 0;
	for(int i = 1;i <= (1<<(n-1));i ++) {
		for(int j = 1;j < n;j ++) {
			w[i + cnt][0][j] = read();
		}
	}
	dfs(1,n - 1);
	int ans = 0;
	for(int i = 0;i <= k;i ++) {
		ans = max(ans,dp[1][i]);
	}
	printf("%d\n",ans);
	return 0; 
}

某位大佬的代码↓（逃）

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream> 
#include<algorithm>
#include<stack>
#include<vector>
using namespace std;
typedef long long LL;
const LL INF=0x3f3f3f3f;
LL n,m,ans=0;
LL w[1500][20],f[1500][20];
LL dp[1500][1500];
bool vis[15];
/*#define gc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<22,stdin),p1 == p2)?EOF:*p1++)
char buf[(1<<22)],*p1=buf,*p2=buf;*/
#define gc() getchar()
template<typename _T>
inline void read(_T &x)
{
	_T f=1;x=0;char s=gc();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=gc();}
	while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=gc();}
	x*=f;
}//读优
void dfs(LL x,LL y)
{
	//printf("%d %d\n",x,y);
	for(LL i=0;i<=1<<y;i++) dp[x][i]=0;
	if(!y)
	{
		for(LL i=1;i<=n;i++)
			if(vis[i]) dp[x][1]+=w[x][i];
			else dp[x][0]+=f[x][i];
		return ;
	}//更新当前叶节点
	vis[y]=0;dfs(x<<1,y-1);dfs(x<<1|1,y-1);//不打仗
	for(LL i=0;i<=1<<(y-1);i++)
		for(LL j=0;j<=1<<(y-1);j++)
			dp[x][i+j]=max(dp[x][i+j],dp[x<<1][i]+dp[x<<1|1][j]);//返回更新
	vis[y]=1;dfs(x<<1,y-1);dfs(x<<1|1,y-1);//打仗
	for(LL i=0;i<=1<<(y-1);i++)
		for(LL j=0;j<=1<<(y-1);j++)
			dp[x][i+j]=max(dp[x][i+j],dp[x<<1][i]+dp[x<<1|1][j]);//返回更新
}
int main()
{
	read(n);read(m);n--;
	for(LL i=0;i<(1<<n);i++)
		for(LL j=1;j<=n;j++)
			read(w[i+(1<<n)][j]);
	for(LL i=0;i<(1<<n);i++)
		for(LL j=1;j<=n;j++)
			read(f[i+(1<<n)][j]);
	dfs(1,n);
	for(LL i=0;i<=m;i++)
		ans=max(dp[1][i],ans);//找答案
	printf("%lld",ans);
	return 0;
}