洛谷：P1352 没有上司的舞会（普及/提高- ，树的dp）

题目：

分析：和紫书上树的dp一样，依稀记得。

代码：

#include<bits/stdc++.h>
using namespace std;
int n;
int A[6002];
long long D[6002];
vector<vector<int> > vv;
long long f(int x)
{
 if(D[x]!=-1) return D[x];
 D[x]=0;
 int a=0; 
 //包含本身。
 a=A[x];
 for(int i=0;i<vv[x].size();i++)
 {
  for(int j=0;j<vv[vv[x][i]].size();j++)
  {
   a+=f(vv[vv[x][i]][j]);
  }
 }
 int b=0; 
 //不包含本身。
  for(int i=0;i<vv[x].size();i++) b+=f(vv[x][i]);
 D[x]=max(a,b);
 return D[x];
}
int main()
{
 cin>>n;
 for(int i=1;i<=n;i++) cin>>A[i];
 vector<int> v;
 for(int i=0;i<=n;i++) vv.push_back(v);
 for(int i=1;i<n;i++)
 {
  int a,b;
  cin>>a>>b;
  vv[b].push_back(a);
 }
 memset(D,-1,sizeof(D));
 long long maxx=-1;
 for(int i=1;i<=n;i++) maxx=max(f(i),maxx);
 cout<<maxx;
}