2342-Anniversary party-树dp

xiaoxiao2021-02-27  435

Anniversary party 题目链接: http://poj.org/problem?id=2342 题意:给定每个人的快乐值,邀请一部分人去参加宴会,被邀请的人和该人的上司只能去一个,使得最终的快乐值最大。 应该是一道很简单的树dp吧,正在学习当中 ~ing,树dp,紫薯上的最大独立集啊。 包括两部分,建树部分和dp部分,,,建树、、、、、和dp(找到转移方程啊) 用1,0分别表示去和不去的状态,dp[i][0],dp[i][1],到第i位上司为止的最大的快乐值(这个人去或不去)。 dp[u][1]+=dp[v][0];    //上司去,下属不去 dp[u][0]+=max(dp[v][0],dp[v][1]);//上司不去,下属去或不去的最大值 用dfs依次递归,如果dfs(1,-1)则是从1开始,最终返回到最大值在1,处,如果dfs(n,-1)开始,最终返回到n处最大值,理解递归。 参考代码: #include <stdio.h> #include <algorithm> #include <math.h> #include <string.h> #include <vector> #include <queue> #include <stack> #include <iostream> #define INF 0x3f3f3f3f using namespace std; const int maxn=6010; struct Edge { int v,next; }edge[maxn*2]; int n,k,cnt[maxn],head[maxn],dp[maxn][2]; void add_edge(int u,int v) { edge[k].v=v; edge[k].next=head[u]; head[u]=k++; } void dfs(int u,int fa) { for(int i=head[u];i!=-1;i=edge[i].next) { int v=edge[i].v; if(v!=fa) { dfs(v,u); dp[u][1]+=dp[v][0]; //去 dp[u][0]+=max(dp[v][0],dp[v][1]);//不去 } } } int main() { //freopen("C:\\Users\\Administrator\\Desktop\\a.txt","r",stdin); k=0; int a,b; for(int i=0;i<maxn;i++) head[i]=-1,dp[i][0]=0,dp[i][1]=0; while(~scanf("%d",&n)) { for(int i=1;i<=n;i++) scanf("%d",&dp[i][1]); while(scanf("%d%d",&a,&b),a&&b) add_edge(a,b),add_edge(b,a); dfs(1,-1); cout<<max(dp[1][0],dp[1][1])<<endl; } return 0; }
转载请注明原文地址: https://www.6miu.com/read-745.html

最新回复(0)