其他分享
首页 > 其他分享> > 【树形DP】ZJOI2008 骑士

【树形DP】ZJOI2008 骑士

作者:互联网

题目内容

洛谷链接
有\(n\)位骑士,每个人的战力可能不同,并且每一个人都有且仅有一个憎恨的人,互相憎恨的人不能在同一队中。
求组合为一个骑士队的最大战斗力。
PS:可以去看看题目背景学学历史(雾)

输入格式

第一行包含一个正整数\(n\),描述骑士团的人数。
接下来\(n\)行,每行两个正整数,按顺序描述每一名骑士的战斗力和他最痛恨的骑士。

数据范围

\(n ≤ 1 000 000\),每名骑士的战斗力都是不大于$ 1 000 000$的正整数。

输出格式

输出最大战斗力。

样例输入

3
10 2
20 3
30 1

样例输出

30

思路

参考没有上司的舞会
不过还是稍有不同,因为没有上司的舞会是个树,而此题是个有向图,可能存在环之类的。
可以把环断开,两个端点分别深搜。
数组为\(dp[1000000][2]\),第二维若为1表示选\(i\)点,0则表示不选。用\(j\)表示\(i\)的憎恨者。
转移方程:
\( \begin{cases} f[i][1]+=f[j][0],\\ f[i][0]+=max(f[j][0],f[j][1]).\\ \end{cases} \)

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+10;
const int INF=0x3f3f3f3f;
int n;
ll ans;//可能超int
ll hater[maxn],fight[maxn],dp[maxn][2];
bool vis[maxn];

struct Node{
    int x,y;
}c[maxn];

int cnt,head[maxn];
void add(int x,int y){
    c[++cnt].x=head[x];
    c[cnt].y=y;
    head[x]=cnt;
}

void dfs(int x,int g){
    vis[x]=1;
    dp[x][1]=fight[x];
    dp[x][0]=0;
    for(int i=head[x];i;i=c[i].x)
      if(c[i].y!=g){
        dfs(c[i].y,g);
        dp[x][1]+=dp[c[i].y][0];
        dp[x][0]+=max(dp[c[i].y][0],dp[c[i].y][1]);
      }
      else dp[c[i].y][1]=-INF;
}

void find(int x){
    while(!vis[x]){
        vis[x]=1;
        x=hater[x];
    }
    dfs(x,x);
    ll temp=max(dp[x][1],dp[x][0]);
    x=hater[x];
    dfs(x,x);
    ans+=max(temp,max(dp[x][1],dp[x][0]));
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%lld%d",&fight[i],&x);
        add(x,i);
        hater[i]=x;
    }
    for(int i=1;i<=n;i++)
        if(!vis[i])find(i);
    printf("%lld",ans);
    return 0;
}

标签:cnt,int,max,ll,树形,maxn,DP,ZJOI2008,dp
来源: https://www.cnblogs.com/Midoria7/p/12766718.html