Work Group
Work Group
https://ac.nowcoder.com/acm/problem/110906
思路
定义表示为根节点的子树选取奇数/偶数个数所获得的最大权值是多少.然后直接dfs转移即可.
代码
#include <bits/stdc++.h> using namespace std; typedef long long ll; const int N=2e5+5,M=2; const ll inf=-(1ll<<62ll); int val[N]; vector<int>g[N]; ll f[N][M]; void dfs(int u,int fa) { f[u][1]=inf; for(int v:g[u]) { if(v==fa) continue;dfs(v,u); ll x0=f[u][0],x1=f[u][1]; f[u][0]=max(x0+f[v][0],x1+f[v][1]); f[u][1]=max(x1+f[v][0],x0+f[v][1]); }f[u][1]=max(f[u][1],f[u][0]+val[u]); } int main() { int n,root=0;cin>>n; for(int i=1;i<=n;i++) { int u,w;cin>>u>>w; if(u==-1) root=i; g[u].push_back(i); g[i].push_back(u); val[i]=w; }dfs(root,-1); cout<<max(f[root][0],f[root][1])<<'\n'; return 0; }
牛客每日一题推介(裙子计划~) 文章被收录于专栏
应该是高质量的题解.