【LOJ】#2111. 「JLOI2015」战争调度

题解

记录一个数组dp[i][S][k]表示第i个点，它上面所有的点的状态（参军或者后勤）可以用状态S来表示，一共有k个平民参军的最大收益，当然数组开不下，可以用vector动态开

我们对于每个平民枚举它上面所有贵族的状态来计算如果对于这个平民，上面所有贵族状态为S的时候，收益是多少

然后对于每个贵族只要枚举它上面的二进制状态S，左子树参军人数a，右子树参军状态b，在左右儿子分别取较大的状态来转移即可

代码

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define enter putchar('\n')
#define space putchar(' ')
//#define ivorysi
using namespace std;
typedef long long int64;
typedef long double db;
typedef unsigned int u32;
template<class T>
void read(T &res) {
    res = 0;char c = getchar();T f = 1;
    while(c < '0' || c > '9') {
	if(c == '-') f = -1;
	c = getchar();
    }
    while(c >= '0' && c <= '9') {
	res = res * 10 + c - '0';
	c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {putchar('-');x = -x;}
    if(x >= 10) out(x / 10);
    putchar('0' + x % 10);
}
int N,M;
int w[4005][15],f[4005][15],len[2005];
vector<int> dp[1035][1035];
void Solve() {
    read(N);read(M);
    for(int i = 0 ; i < (1 << N - 1) ; ++i) {
	for(int j = 1 ; j <= N - 1 ; ++j) read(w[i][j]);
    }
    for(int i = 0 ; i < (1 << N - 1) ; ++i) {
	for(int j = 1 ; j <= N - 1 ; ++j) read(f[i][j]);
    }
    len[0] = len[1] = 1;
    for(int i = 2 ; i < (1 << N) ; ++i) len[i] = len[i / 2] + 1;
    for(int k = (1 << N - 1) - 1 ; k >= 0 ; --k) {

	for(int S = 0 ; S < (1 << N) ; ++S) {
	    dp[k + (1 << N - 1)][S].resize(2);
	    for(int j = 1 ; j <= N - 1 ; ++j) {
		if((S & 1) && (S >> j & 1)) dp[k + (1 << N - 1)][S][S & 1] += w[k][j];
		if(!(S & 1) && !(S >> j & 1)) dp[k + (1 << N - 1)][S][S & 1] += f[k][j];
	    }
	}
    }
    for(int k = (1 << N - 1) - 1; k >= 1 ; --k) {
	int s = dp[k << 1][0].size();
	for(int S = 0 ; S < (1 << len[k]) ; ++S) {
	    dp[k][S].resize(s * 2);
	    for(int a = 0 ; a < s ; ++a) {
		int t1 = max(dp[k << 1][S << 1][a],dp[k << 1][S << 1 | 1][a]);
		for(int b = 0 ; b < s ; ++b) {
		    int t2 = max(dp[k << 1 | 1][S << 1][b],dp[k << 1 | 1][S << 1 | 1][b]);
		    dp[k][S][a + b] = max(dp[k][S][a + b],t1 + t2);
		}
	    }
	}
    }
    int ans = 0;
    for(int i = 0 ; i <= M ; ++i) ans = max(ans,max(dp[1][0][i],dp[1][1][i]));
    out(ans);enter;
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Solve();
}