卡常树（贪心，长链剖分）

最新推荐文章于 2023-02-13 15:01:24 发布

DD(XYX)

最新推荐文章于 2023-02-13 15:01:24 发布

阅读量127

点赞数 1

分类专栏：树鬼贪心文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_43960414/article/details/120210674

版权

树同时被 2 个专栏收录

57 篇文章 0 订阅

订阅专栏

鬼贪心

22 篇文章 0 订阅

订阅专栏

题面

在这里插入图片描述

题解

我们发现，这是个贪心问题，我们可以每次选当前权值最大的一条合法链。

为什么呢，因为它完全就是个拟阵。

好，那么有这么一种模拟做法：

找到权值和最大的合法链，加入答案。
将该链上的权值清零。
重复 $k$ 次。

好，如果是一般题，到这里就结束了，因为我们可以根据 $\tt dfs$ 序用线段树维护最值，时间复杂度 $O(n\log n)$ ，空间复杂度 $O (n)$ 。

但是，这是道卡常题，空间开成这样（必要数组至少 ${\rm 32n}~{\tt byte}$ ，线段树至少 ${\rm 36n}~{\tt byte}$ ）过不了，直接 0 分。

我们先从卡空间入手吧，至少 45 分是拿得到的。

卡空间做法：暴力

我们注意到，这道题的树是种子生成的，满足随机生成的性质。

那么，可得：树高不超过 $\log$ 。

而且，贪心思路下，我们只关心叶子，于是——

直接把原树当成线段树使！空间砍大半！

我们每次修改权值，就从叶子跑到根，一路更新每个点子树内的最大合法链就完了。

空间解决！时间复杂度 $O(n\log n)$ 。

时间过不去了，得换个方向了。

卡时间做法：长链剖分

我们观察选出的链都有哪些性质：
在这里插入图片描述
很容易发现：这不就是长链剖分吗？

我们把这棵树进行长链剖分，每个叶子记录它所在的一条链的权值和，然后放到一个数组里。

接下来我们要对该数组的前 $k$ 大求和。

注意到我们并不需要整个数组变得有序，我们只想得到前 $k$ 大是那些。

nth_element() 足矣，时间复杂度 $O (n)$ ，空间常数小。

最后，光是复杂度 $O (n)$ 还不行，我们常数还非常大。原因在于，递归遍历。

我们观察数据，发现每个点的父亲编号一定比自己小，于是，可以把长链剖分用两个分开的循环代替。

这才能跑进 1 s 。

CODE

#include<set>
#include<map>
#include<stack>
#include<cmath>
#include<ctime>
#include<queue>
#include<bitset>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 4000005
#define LL long long
#define ULL unsigned long long
#define UI unsigned int
#define DB double
#define ENDL putchar('\n')
#define lowbit(x) (-(x) & (x))
#define FI first
#define SE second
#define eps (1e-4)
#define SI(x) set<x>::iterator
LL read() {
	LL f=1,x=0;char s = getchar();
	while(s < '0' || s > '9') {if(s=='-')f = -f;s = getchar();}
	while(s >= '0' && s <= '9') {x=x*10+(s-'0');s = getchar();}
	return f*x;
}
void putpos(LL x) {
	if(!x) return ;
	putpos(x/10); putchar('0'+(x%10));
}
void putnum(LL x) {
	if(!x) putchar('0');
	else if(x < 0) putchar('-'),putpos(-x);
	else putpos(x);
}
void AIput(LL x,char c) {putnum(x);putchar(c);}

const int MOD = 1000000007;
int n,m,s,o,k;
int hd[MAXN],nx[MAXN];
void ins(int x,int y) {
	nx[y] = hd[x]; hd[x] = y;
}
int val[MAXN],fa[MAXN];
int son[MAXN];
LL dp[MAXN],siz[MAXN];
LL as[MAXN];
int cn;
int main() {
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	n = read();m = read();
	int sed = read();
	for(int i = 1;i <= n;i ++) {
		val[i] = sed;
		if(i>1) {
			s = (sed ^ 23333333) % (i-1) + 1;
			ins(s,i);
			fa[i] = s;
		}
		sed = (sed * 23333333ll + 6666666ll) % MOD;
	}
	for(int x = n;x > 0;x --) {
		siz[x] = val[x];
		for(int i = hd[x];i;i = nx[i]) {
			if(siz[i] + val[x] > siz[x]) {
				son[x] = i; siz[x] = siz[i] + val[x];
			}
		}
	}
	for(int x = 1;x <= n;x ++) {
		if(son[fa[x]] == x) dp[x] = dp[fa[x]] + val[x];
		else dp[x] = val[x];
		if(!hd[x]) as[++ cn] = dp[x];
	}
	m = min(m,cn);
	nth_element(as + 1,as + (cn-m+1),as + 1 + cn);
	LL ans = 0;
	for(int i = cn;i > 0 && i > cn-m;i --) {
		ans += as[i];
	}
	AIput(ans,'\n');
	return 0;
}