题面
题解
我们发现,这是个贪心问题,我们可以每次选当前权值最大的一条合法链。
为什么呢,因为它完全就是个拟阵。
好,那么有这么一种模拟做法:
- 找到权值和最大的合法链,加入答案。
- 将该链上的权值清零。
- 重复 k k k 次。
好,如果是一般题,到这里就结束了,因为我们可以根据 d f s \tt dfs dfs 序用线段树维护最值,时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间复杂度 O ( n ) O(n) O(n)。
但是,这是道卡常题,空间开成这样(必要数组至少 32 n b y t e {\rm 32n}~{\tt byte} 32n byte,线段树至少 36 n b y t e {\rm 36n}~{\tt byte} 36n byte)过不了,直接 0 分。
我们先从卡空间入手吧,至少 45 分是拿得到的。
卡空间做法:暴力
我们注意到,这道题的树是种子生成的,满足随机生成的性质。
那么,可得:树高不超过 log \log log 。
而且,贪心思路下,我们只关心叶子,于是——
直接把原树当成线段树使!空间砍大半!
我们每次修改权值,就从叶子跑到根,一路更新每个点子树内的最大合法链就完了。
空间解决! 时间复杂度 O ( n log n ) O(n\log n) O(nlogn) 。
时间过不去了,得换个方向了。
卡时间做法:长链剖分
我们观察选出的链都有哪些性质:
很容易发现:这不就是长链剖分吗?
我们把这棵树进行长链剖分,每个叶子记录它所在的一条链的权值和,然后放到一个数组里。
接下来我们要对该数组的前 k k k 大求和。
注意到我们并不需要整个数组变得有序,我们只想得到前 k k k 大是那些。
nth_element()
足矣,时间复杂度
O
(
n
)
O(n)
O(n) ,空间常数小。
最后,光是复杂度 O ( n ) 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;
}