P1352 没有上司的舞会

P1352 没有上司的舞会
一道树形DP的模板题。树形DP一般用递归实现,将节点从小到大的顺序作为动态规划的阶段,先递归子节点,再回溯从子节点到另一节点转移。

#include
using namespace std;

vector son[1010010];
int dp[10001][101],vis[100001],tree[100101];
int n;

void dptree(int x){
dp[x][0] = 0;
dp[x][1] = tree[x];
for(int i=0;i int Tree = son[x][i];
dptree(Tree);
dp[x][0]+=max(dp[Tree][0],dp[Tree][1]);
dp[x][1]+=dp[Tree][0];
}
}

int main(){

cin >> n;
for(int i=1;i<=n;i++){
	cin >> tree[i];
}
for(int i=1;i> x >>y;
 	vis[x] = 1;
 	son[y].push_back(x);
}
int root;
for(int i=1;i<=n;i++){
	if(!vis[i]){
		root = i;
	}
}
dptree(root);
int ans = max(dp[root][0],dp[root][1]);
cout << ans;
return 0;

}```



你可能感兴趣的:(洛谷之旅)