poj 2342 hdu 1520【树形dp】

poj 2342

给出每个顶点的happy值，还有若干组两个顶点L，K关系，表示K是L的上司。求当K、L不同时出现时获得的happy值的最大和。

设dp[u][0]表示不选u结点时获得的最大值，dp[u][1]表示选u结点时获得的最大值。则有：

　　　dp[u][0]+=max(dp[v][0],dp[v][1]),dp[u][1]+=dp[v][0](u为v的父节点)

当父亲节点用有向边连向子节点时，会形成一颗树，自然就只有一个根。那么从根开始dfs就行了。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<vector>
 #include<algorithm>
 using namespace std;
 const int maxn = ;
 int dp[maxn][], vis[maxn], a[maxn];
 vector<int> tree[maxn];
 int N;

 void Init()
 {
     for (int i = ; i <= N; i++) {
         cin >> a[i];
         tree[i].clear(), vis[i] = ;
     }
     int L, K;
     while (cin>>L>>K)
     {
         if (L ==  && K == ) break;
         tree[K].push_back(L);
         vis[L] = ;
     }
     tree[].clear();//找根
     for(int i=;i<=N;i++)
         if (!vis[i]) {
             tree[].push_back(i);
             break;//只有一个根，找到后就break
         }
 }

 void dfs(int u)
 {
     dp[u][] = ;
     dp[u][] = a[u];
     for (int i = ; i < tree[u].size(); i++)
     {
         int v = tree[u][i];
         dfs(v);
         dp[u][] += max(dp[v][], dp[v][]);
         dp[u][] += dp[v][];
     }
 }

 void Solve()
 {
     int root = tree[][];
     dfs(root);
     cout << max(dp[root][], dp[root][]) << endl;
 }

 int main()
 {
     while (cin>>N)
     {
         Init();
         Solve();
     }
     return ;
 }

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<vector>
 #include<cmath>
 #include<algorithm>
 using namespace std;
 const int maxn=;
 int dp[maxn][],rat[maxn],f[maxn],vis[maxn];
 vector<int> tree[maxn];

 void dfs(int u)
 {
     vis[u]=;
     dp[u][]=;
     dp[u][]=rat[u];
     int len=tree[u].size();
     for(int i=;i<len;i++){
         int v=tree[u][i];
         if(vis[v]) continue;
         dfs(v);
         dp[u][]+=max(dp[v] [],dp[v][]);
         dp[u][]+=dp[v][];
     }
 }

 int main()
 {
     int N;
     while(scanf("%d",&N)==)
     {
         for(int i=;i<=N;i++){
             scanf("%d",&rat[i]);
             vis[i]=;f[i]=-;
             tree[i].clear();
         }
         int a,b;
         while(scanf("%d%d",&a,&b))
         {
             if(a==&&b==) break;
             f[a]=b;
             tree[b].push_back(a);
         }
         int root;
         for(int i=;i<=N;i++){
             if(f[i]==-){
                 root=i;
                 break;
             }
         }
         dfs(root);
         cout<<max(dp[root][],dp[root][])<<endl;
     }
     return ;
 }

又写了一遍