【洛谷P1352】没有上司的舞会 树形dp第二题
第一种做法 我们把每个上司之间存一个 n e ne ne数组 和 p o po po数组 po数组是这个点的最后一个一个儿子 ne数组是这个儿子的上一个兄弟 递归到0 类似邻接链表
然后我们知道如果你取这个点 那么
dp[1][x] = max(max(dp[1][x],dp[1][x]+dp[0][i]),dp[0][i]);
取这个点的最大值为 和它本身加上不取他儿子的最大值 和他儿子不取的最大值取个max
dp[0][x] = max(max(dp[0][x],max(dp[0][x]+dp[1][i],dp[0][x]+dp[0][i])),max(dp[0][i],dp[1][i]));
不取这个点的最大值为他本身加上取儿子 和 他本身加上不取儿子 和取他儿子 和不取他儿子的值取max
/*
if you can't see the repay
Why not just work step by step
rubbish is relaxed
to ljq
*/
#include #define dbg3(x1,x2,x3) cout<<#x1<<" = "<#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
typedef pair<int,int> pll;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int _inf = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll _INF = 0xc0c0c0c0c0c0c0c0;
const ll mod = (int)1e9+7;
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll ksm(ll a,ll b,ll mod){int ans=1;while(b){if(b&1) ans=(ans*a)%mod;a=(a*a)%mod;b>>=1;}return ans;}
ll inv2(ll a,ll mod){return ksm(a,mod-2,mod);}
void exgcd(ll a,ll b,ll &x,ll &y,ll &d){if(!b) {d = a;x = 1;y=0;}else{exgcd(b,a%b,y,x,d);y-=x*(a/b);}}//printf("%lld*a + %lld*b = %lld\n", x, y, d);
const int MAX_N = 6025;
int ne[MAX_N],po[MAX_N],indegre[MAX_N];
ll dp[2][MAX_N];
void DP(int x)
{
for(int i = po[x];i;i=ne[i])
{
DP(i);
dp[1][x] = max(max(dp[1][x],dp[1][x]+dp[0][i]),dp[0][i]);
dp[0][x] = max(max(dp[0][x],max(dp[0][x]+dp[1][i],dp[0][x]+dp[0][i])),max(dp[0][i],dp[1][i]));
}
}
int main()
{
//ios::sync_with_stdio(false);
//freopen("a.txt","r",stdin);
//freopen("b.txt","w",stdout);
int a,b,n,xb;
scanf("%d",&n);
for(int i = 1;i<=n;++i) scanf("%lld",&dp[1][i]);
for(int i = 1;i<=n;++i)
{
scanf("%d%d",&b,&a);
if(b==0) continue;
indegre[b]++;
ne[b] = po[a];
po[a] = b;
}
for(int i = 1;i<=n;++i)
if(!indegre[i])
{
xb = i;
break;
}
DP(xb);
printf("%lld\n",max(dp[1][xb],dp[0][xb]));
//fclose(stdin);
//fclose(stdout);
//cout << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
return 0;
}