AcWing 359 创世纪 基环树,树形dp

这道题bzoj上也有,可惜我没权限,只在AcWing上过了

题目:https://www.acwing.com/problem/content/361/

首先是建边问题,一个点可不可以选取决于有没有限制它的点没有被选上,因此建立边的方式就是从 被限制的点指向限制它的点,单向边,此时假如需要选一个点,那么必定有一个以上它的子节点没有被选

用dp[i][option] 来做dp数组,其中i为哪一个点,option为0的话表示不选这个点,option为1的话表示选这个点。

假设to是now的子节点,那么

不选now这个点,那么其子节点怎么选都无所谓,直接都选最大的

假如选now这个点,其子节点就必定有一个不能被选上,此时如何表达呢?

首先我们先选子节点最大的那个状态:

假如此时有一个子节点的dp[to][0]>=dp[to][1],那就符合条件,但是假如所有子节点都是dp[to][1]>dp[to][0]

此时我们需要找到一个dp[to][1]-dp[to][0] 最小的,然后让dp[now][1]减去这个最小值即可

其中sto初始化为正无穷,假如有一个dp[to][0]>=dp[to][1],那么sto的结果将会是0

最后,对于环形问题,有时候是将它切割成线性问题,然后分类讨论完成。

本题就是需要找到环上的一个点,将它和它限制的那个点断开,先后进行两次树形dp,第一次是假设环上的这个点对其限制的点不起限制作用,另外一次是强制环上那个点已经限制了其可以限制的点(也就是环上那个点不选)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 7;
int n, tot, root;
int fat[maxn], dp[maxn][2];
int Head[maxn], Nxt[maxn << 1], To[maxn << 1];
int vis[maxn];
int flag;
void add_edge(int fro,int to) {
	Nxt[++tot] = Head[fro];
	To[tot] = to;
	Head[fro] = tot;
}
void tree_dp(int now) {
	dp[now][0] = 0;
	vis[now] = 1;
	int res = 0, sto = 0x3f3f3f3f;
	for (int i = Head[now]; i; i = Nxt[i]) {
		int &to = To[i];
		if (to == root) continue; 
		tree_dp(to);
		sto = min(sto, max(dp[to][1], dp[to][0]) - dp[to][0]);
		dp[now][0] += max(dp[to][0], dp[to][1]);
	}
	dp[now][1] = dp[now][0] - sto + 1;
	if (now == fat[root] && flag) dp[now][1] = dp[now][0] + 1; //假如起强制限制作用,那么那个点的其他子节点就都可以随便选了
}
int main() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		register int to;
		scanf("%d", &to);
		add_edge(to, i);
		fat[i] = to;
	}
	int ans = 0;
	for (int i = 1; i <= n; i++) {
		if (vis[i]) continue;
		int res = 0;
		root = i;
		while (!vis[fat[root]]) { //如此可以找到环上的一个点
			vis[root] = 1;
			root = fat[root];
		}
		flag = 0; //flag=0表示不起限制作用
		tree_dp(root);
		res = max(dp[root][0], dp[root][1]);
		flag = 1; //表示起强制限制作用
		tree_dp(root);
		res = max(res, dp[root][0]);
		ans += res;
	}
	cout << ans << endl;
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值