BZOJ1040: [ZJOI2008]骑士 基环树DP

题面
一般的基环树DP 处理环时的思路是断环成链，转为序列上的dp 其他地方同树形DP
但这样写起来挺麻烦
这道题可以以一种更简洁的方法处理：
把图建成无向图，用dfs找环。对于每个环，任意切断其环上的一条边，并记录两端点编号。
一颗基环树砍掉一条环上的边后即变成了一颗树，可以在上面做树形dp
由于我们建的是无向图，可以从一个端点出发做一次树形dp，另一个端点再做一次
两端点的答案中取最优解即为这颗基环树的答案。
实现时要注意一个细节：如果dfs时记录的是父节点，会无法正确处理二元环，直接做出一颗没有环的树。
应该记录走到当前节点的边的编号。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define Min(_A,_B) (_A<_B?_A:_B)
#define Max(_A,_B) (_A>_B?_A:_B)
#define ll long long
const int maxn = 1e6 + 7;
int read(){
    int s = 0, f = 1; char c = getchar();
    while (c > '9' || c < '0') {
        if (c == '-') f = -1; c = getchar();
    }
    while (c >= '0' && c <= '9') {
        s = s * 10 + c - '0';
        c = getchar();
    }
    return s * f;
}
struct edge{
    int v, nxt;
}e[maxn << 1];
int head[maxn], eid, tot, n, cnt, s, t;
ll f[maxn], g[maxn], d[maxn], ans, tmp;
bool vis[maxn], book[maxn << 1];
void init(){memset(head, -1, sizeof(head)); eid = 0;}
void insert(int u, int v){
    e[eid].v = v; e[eid].nxt = head[u]; head[u] = eid++;
}
void dfs(int u, int fi){
    vis[u] = true;
    for (int i = head[u]; ~i; i = e[i].nxt) {
        int v = e[i].v;
        if (i == (fi ^ 1)) continue;
        if (vis[v]) {
            book[i] = book[i ^ 1] = true;   //标记无向边
            s = u, t = v; 
        }
        else dfs(v, i);
    }
}
void dp(int u, int fi) {
    f[u] = d[u]; g[u] = 0;
    for (int i = head[u]; ~i; i = e[i].nxt) {
        int v = e[i].v;
        if (i == (fi ^ 1) || book[i] || book[i ^ 1]) continue;
        dp(v, i); 
        f[u] += g[v]; 
        g[u] += Max(f[v], g[v]);
    }
}
int main(){
    n = read(); init();
    for (int i = 1; i <= n; i++) {
        d[i] = read(); int x = read();
        insert(i, x); insert(x, i);
    }
    for (int i = 1; i <= n; i++) {
        if (!vis[i]) {
            dfs(i, 0x3f3f3f3f);  //每棵子树内dfs+dp
            dp(s, 0x3f3f3f3f);
            tmp = g[s];
            dp(t, 0x3f3f3f3f);
            ans += Max(g[t], tmp);
        }
    }
    printf("%lld\n", ans);
    return 0;
}