洛谷P1352没有上司的舞会
思路:
树形dp的模板题。
dp[i][0]表示不i结点后i子树能得到得最大值,dp[i][1]表示选了i子树后能得到得最大值。
用dfs,回溯时通过子节点向上更新信息。
dp[u][0]=∑max{dp[v][1],dp[v][0]};
dp[u][1]=a[u]+dp[v][0];
对于找根节点,我们可以把{u,v}作为一个有向边,根节点即为入度为0的点。
代码:
#include<bits/stdc++.h>
const int N=1e5+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
int dp[N][2],a[N],in[N];
vector<int> son[N];
void dfs(int u)
{
dp[u][0]=0;
dp[u][1]=a[u];
for(int i=0;i<son[u].size();i++)
{
int v=son[u][i];
dfs(v);
dp[u][0]+=max(dp[v][0],dp[v][1]);
dp[u][1]+=dp[v][0];
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,i;
cin>>n;
for(i=1;i<=n;i++)
cin>>a[i];
for(i=1;i<=n-1;i++)
{
int u,v;
cin>>u>>v;
son[v].push_back(u);
in[u]=1;
}
int root;
for(i=1;i<=n;i++)
if(!in[i])
{
root=i;
break;
}
dfs(root);
int ans=max(dp[root][0],dp[root][1]);
cout<<ans<<endl;
return 0;
}