【HDU】1520 Anniversary party（树形dp）

题目

题目

---

分析

带权值的树上最大独立集

---

代码

 #include <bits/stdc++.h>
 using namespace std;
 const int maxn=;
 int a[maxn], n, fa[maxn];
 vector<int> son[maxn];

 int dp(int u)
 {
     int sum1=a[u],sum2=;
     if(son[u].empty())
     {
         return a[u];
     }
     for(int i=;i<son[u].size();i++)
     {
         for(int j=;j<son[son[u][i]].size();j++)
         sum1+=dp(son[son[u][i]][j]);
         sum2+=dp(son[u][i]);
     }
     int ans=max(sum1,sum2);
     return ans;
 }
 int main()
 {
     int k,l;
     while(scanf("%d",&n)==)
     {
         for(int i=; i<=n;i++) scanf("%d",&a[i]);
         while(scanf("%d%d",&k,&l) && k)
         {
             son[l].push_back(k);
             fa[k]=;
         }
         int Be=;
         for(int i=;i<=n;i++)
             if(!fa[i]) { Be=i; break; }

         printf("%d\n",dp(Be));

         for(int i=;i<=n+;i++) son[i].clear();
         memset(fa,,sizeof(fa));
     }

     return ;
 }