AcWing 1080：骑士 ← 树形DP+基环树

【题目来源】
https://www.acwing.com/problem/content/1082/

【题目描述】
Z 国的骑士团是一个很有势力的组织，帮会中聚集了来自各地的精英。
他们劫富济贫，惩恶扬善，受到了社会各界的赞扬。
可是，最近发生了一件很可怕的事情：邪恶的 Y 国发起了一场针对 Z 国的侵略战争。
战火绵延五百里，在和平环境中安逸了数百年的 Z 国又怎能抵挡得住 Y 国的军队。
于是人们把所有希望都寄托在了骑士团身上，就像期待有一个真龙天子的降生，带领正义打败邪恶。
骑士团是肯定具备打败邪恶势力的能力的，但是骑士们互相之间往往有一些矛盾。
每个骑士有且仅有一个他自己最厌恶的骑士（当然不是他自己），他是绝对不会与最厌恶的人一同出征的。
战火绵延，人们生灵涂炭，组织起一个骑士军团加入战斗刻不容缓！
国王交给你了一个艰巨的任务：从所有骑士中选出一个骑士军团，使得军团内没有矛盾的两人，即不存在一个骑士与他最痛恨的人一同被选入骑士军团的情况，并且使这支骑士军团最富有战斗力。
为描述战斗力，我们将骑士按照 1 至 N 编号，给每位骑士一个战斗力的估计，一个军团的战斗力为所有骑士的战斗力之和。

【输入格式】
输入第一行包含一个正整数 N，描述骑士团的人数；
接下来 N 行每行两个正整数，按顺序描述每一名骑士的战斗力和他最痛恨的骑士。

【输出格式】
输出包含一行，一个整数，表示你所选出的骑士军团的战斗力。

【数据范围】
N≤10^6,
每名骑士的战斗力都是不大于 10^6 的正整数。

【输入样例】
3
10 2
20 3
30 1

【输出样例】
30

【算法分析】
● 链式前向星：https://blog.csdn.net/hnjzsyjyj/article/details/139369904
● 并查集：https://blog.csdn.net/hnjzsyjyj/article/details/120147618
● 环形结构上的动态规划问题的基本处理方式：https://blog.csdn.net/hnjzsyjyj/article/details/100418268
环形结构上的动态规划问题，是一种特殊的区间动态规划问题。
由于存在“环形后效性”，所以不满足动态规划算法的“无后效性”约束条件。故常将环形结构上的动态规划问题，通过“断环为链”策略转化为线性动态规划问题求解。所谓“无后效性”，即“未来与过去无关”。所谓“断环为链”策略，即“把环断开为链，然后复制一倍接在末尾”。示意图如下：

● 没有上司的舞会：https://blog.csdn.net/hnjzsyjyj/article/details/119890782
本题在去掉“基环树”的环之后，代码与“没有上司的舞会”代码基本一致。即为经典的树形DP问题。

【算法代码】

/* 链式前向星存图
val[idx]：存储编号为 idx 的边的值
e[idx]：存储编号为 idx 的结点的值
ne[idx]：存储编号为 idx 的结点指向的结点的编号
h[a]：存储头结点 a 指向的结点的编号
*/

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef pair<int, int> PII;

const int N=1e6+5;
const int M=N<<1;
int pre[N];
int val[N];
LL f[N][2];
vector<PII> roots;
int n;
int e[M],ne[M],h[N],idx;

void add(int a,int b) {
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

int find(int x) {
    if(x!=pre[x]) pre[x]=find(pre[x]);
    return pre[x];
}

/*void merge(int x,int y) {
    int a=find(x);
    int b=find(y);
    if(a!=b) pre[a]=b;
}*/

LL dfs(int u,int fa) {
    f[u][1]=val[u];
    f[u][0]=0;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        if(j==fa) continue;
        dfs(j,u);
        f[u][1]+=f[j][0];
        f[u][0]+=max(f[j][0],f[j][1]);
    }
    return f[u][0];
}

int main() {
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    for(int i=1; i<=n; i++) pre[i]=i;

    for(int i=1; i<=n; i++) {
        int x;
        scanf("%d%d", &val[i], &x);
        int a=find(i);
        int b=find(x);
        if(a!=b) {
            pre[a]=b;
            add(i,x),add(x,i);
        } else roots.push_back({x,i});
    }

    LL sum=0;
    for(PII t:roots) {
        int x=t.first, y=t.second;
        LL ans=dfs(x,-1);
        ans=max(ans,dfs(y,-1));
        sum+=ans;
    }

    printf("%lld", sum);

    return 0;
}

/*
in:
3
10 2
20 3
30 1

out:
30
*/

【参考文献】
https://www.acwing.com/solution/content/133721/
https://www.acwing.com/solution/content/13182/