【题解】LuoGu2607:[ZJOI2008]骑士

原题传送门
基环树的树形dp
对于处理基环树,我本来认为只能先把树的部分处理好之后,在环上把信息综合起来,不过这题非常麻烦
基环树有另一种处理信息的方法,就是断环
把环找出来之后,从中间断掉他,分别以两个断点为根做树形dp,取答案大的那个累加到最终答案里
Code:

#include <bits/stdc++.h>
#define LL long long
#define maxn 1000010
using namespace std;
struct Edge{
	int to, next;
}edge[maxn << 1];
int num, head[maxn], a[maxn], f[maxn], vis[maxn], n;
LL dp[maxn][2], ans;

inline int read(){
	int s=  0,w = 1;
	char c = getchar();
	for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
	for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
	return s * w;
}

void addedge(int x, int y){ edge[++num] = (Edge){y, head[x]}, head[x] = num; }

void dfs(int u, int x){
	vis[u] = 1;
	dp[u][0] = 0, dp[u][1] = a[u];
	for (int i = head[u]; i; i = edge[i].next){
		int v = edge[i].to;
		if (v != x){
			dfs(v, x);
			dp[u][0] += max(dp[v][0], dp[v][1]);
			dp[u][1] += dp[v][0];
		} else dp[v][1] = -1e9;
	}
}

void solve(int u){ 
	while (!vis[u]) vis[u] = 1, u = f[u];
	dfs(u, u);
	LL sum = max(dp[u][0], dp[u][1]);
	u = f[u];
	dfs(u, u);
	ans += max(sum, max(dp[u][0], dp[u][1]));
}

int main(){
	n = read();
	for (int i = 1; i <= n; ++i){
		a[i] = read(), f[i] = read();
		addedge(f[i], i);
	}
	for (int i = 1; i <= n; ++i)
		if (!vis[i]) solve(i);
	printf("%lld\n", ans);
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值