Anniversary party
题解:没什么好讲的,设dp[u][0]代表不选这个节点,dp[u][1]代表选这个节点。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 6005;
int dp[maxn][2],w[maxn];
vector<int>G[maxn];
int dfs(int u,int fa)
{
dp[u][1] = w[u];
dp[u][0] = 0;
for(auto v:G[u])
if(v!=fa) {
dfs(v,u);
dp[u][0] += max(dp[v][1],dp[v][0]);
dp[u][1] += dp[v][0];
}
return max(dp[u][0],dp[u][1]);
}
int main()
{
int n,u,v;
while(~scanf("%d",&n))
{
for(int i=0;i<maxn;i++)
G[i].clear();
for(int i=1;i<=n;i++)
scanf("%d",&w[i]);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&u,&v);
if(!u) continue;
G[u].push_back(v);
G[v].push_back(u);
}
printf("%d\n",dfs(1,0));
}
}
Computer
题解:换根dp入门
我有一道好题也是换根dp+树直径
题目
#include<bits/stdc++.h>
#define ll long long
#define pi pair<int,int>
#define mk make_pair
using namespace std;
const int maxn = 1e4+10;
vector<pi>G[maxn];
vector<ll>sf[maxn];
ll dp[maxn], ans[maxn];
void dfs(int u,int fa)
{
dp[u] = 0;
for(auto tmp:G[u])
{
int v = tmp.first , w = tmp.second ;
if(v==fa) continue;
dfs(v,u);
dp[u] = max(dp[u],dp[v]+w);
}
}
void dfs1(int u,int fa)
{
sf[u].push_back(0);
dp[u] = 0;
for(int i=G[u].size()-1;i>=0;i--)
{
int v = G[u][i].first , w = G[u][i].second;
dp[u] = max(dp[u],dp[v]+w);
sf[u].push_back(dp[u]);
}
ans[u] = dp[u];
ll x = 0;
int j = sf[u].size()-2;
for(auto tmp:G[u])
{
int v = tmp.first , w = tmp.second;
dp[u] = max(x,sf[u][j]);
x = max(x,dp[v]+w);
if(v!=fa) dfs1(v,u);
j--;
}
}
int main()
{
int n;
while(~scanf("%d",&n))
{
int v,w;
for(int i=1;i<=n;i++)
G[i].clear(),sf[i].clear();
for(int i=2;i<=n;i++)
{
scanf("%d%d",&v,&w);
G[i].push_back(mk(v,w));
G[v].push_back(mk(i,w));
}
dfs(1,0);
dfs1(1,0);
for(int i=1;i<=n;i++)
printf("%lld\n",ans[i]);
}
}