题面
一般的基环树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;
}