「FJOI2018」领导集团问题 解题报告

「FJOI2018」领导集团问题

题意:给你一颗(n)个点的带点权有根树,选择一个点集(S),使得点集中所有祖先的点权$le (子孙的点权,最大化)|S|$(出题人语死早...)


  • 一个显然的(dp)

    (dp_{i,j})代表子树(i)中选择的点集中最小的点权为(j)时的最大值。

  • 一个朴素的转移需要维护一个后缀最大值

    设为(f_{i,j}=maxlimits_{kge i} dp_{i,k})

  • 一些显然的事实

    • 后缀最大值中仅有最多(siz_v)(指子树大小)个权值不同的位置
    • 后缀最大值单调不增

于是我们考虑如何合并两颗子树有限的后缀最大值

在不考虑关键点时,我们只需要对应位置两两加和就可以了,即

[dp_{i,j}=sum_vf_{v,j} ]

下面我们仅考虑关键点

如图所示,蓝色的关键点是子树(u)的关键点,黄色的是子树(v)的关键点,注意到关键点是左边位置的值与它相等。

考虑启发式合并,但是我们注意到关键点是互相影响的,就是黄影响蓝,蓝影响黄。

其实这样可以做,但是比较麻烦。

注意到一个事实,我们只需要求(f_{rt,0}),所以我们考虑维护后缀最大值(f)的差分数组。

这样我们合并两个关键点集合,直接对关键点进行相加就可以了,考虑直接用线段树合并维护。

然后我们还需要计算根(i)自己的贡献,设(i)的权值为(w_i)

考虑到根直接的贡献是使(dp_{i,w_i}=f_{i,w_i}+1)

然后从这个(w_i)往左边走,找到第一个(f_{i,p})(f_{i,w_i})(1)的位置,然后更新这一段的贡献,就是一个区间加。具体操作就是在(p)打个(-1),在(w_i)打一个(1)

形象一点就是把这一段铺平了,就是那(dp)再回去贡献(f)

找第一个位置就是在差分意义下就是找一个非(0)位置,可以在线段树上进行二分操作找到。


Code:

#include <cstdio>
#include <cctype>
#include <vector>
#include <algorithm>
using std::min;
using std::max;
const int SIZE=1<<21;
char ibuf[SIZE],*iS,*iT;
//#define gc() (iS==iT?(iT=(iS=ibuf)+fread(ibuf,1,SIZE,stdin),iS==iT?EOF:*iS++):*iS++)
#define gc() getchar()
template <class T>
void read(T &x)
{
	int f=0;x=0;char c=gc();
	while(!isdigit(c)) f|=c=='-',c=gc();
	while(isdigit(c)) x=x*10+c-'0',c=gc();
	if(f) x=-x;
}
const int N=4e5+10;
int n,m,w[N],yuy[N];
int head[N],to[N],Next[N],cnt;
void add(int u,int v)
{
	to[++cnt]=v,Next[cnt]=head[u],head[u]=cnt;
}
int root[N],ch[N*40][2],sum[N*40],tot;
#define ls ch[now][0]
#define rs ch[now][1]
void upt(int &now,int l,int r,int p,int d)
{
	if(!now) now=++tot;
	if(l==r)
	{
		sum[now]+=d;
		return;
	}
	int mid=l+r>>1;
	if(p<=mid) upt(ls,l,mid,p,d);
	else upt(rs,mid+1,r,p,d);
	sum[now]=sum[ls]+sum[rs];
}
int qry(int now,int l,int r,int p)
{
	if(!now||!p) return 0;
	if(l==r) return sum[now]?l:0;
	int mid=l+r>>1;
	if(p<=mid) return qry(ls,l,mid,p);
	else 
	{
		int ret=qry(rs,mid+1,r,p);
		if(ret) return ret;
		else return qry(ls,l,mid,p);
	}
}
int Merge(int x,int y,int l,int r)
{
	if(!x||!y) return x^y;
	if(l==r)
	{
		sum[x]+=sum[y];
		return x;
	}
	int mid=l+r>>1;
	ch[x][0]=Merge(ch[x][0],ch[y][0],l,mid);
	ch[x][1]=Merge(ch[x][1],ch[y][1],mid+1,r);
	sum[x]=sum[ch[x][0]]+sum[ch[x][1]];
	return x;
}
void dfs(int now)
{
	for(int v,i=head[now];i;i=Next[i])
	{
		dfs(v=to[i]);
		root[now]=Merge(root[now],root[v],0,m);
	}
	upt(root[now],1,m,w[now],1);
	int pos=qry(root[now],1,m,w[now]-1);
	if(pos) upt(root[now],1,m,pos,-1);
}
int main()
{
	read(n);
	for(int i=1;i<=n;i++) read(w[i]),yuy[i]=w[i];
	std::sort(yuy+1,yuy+1+n);
	m=std::unique(yuy+1,yuy+1+n)-yuy-1;
	for(int i=1;i<=n;i++) w[i]=std::lower_bound(yuy+1,yuy+1+m,w[i])-yuy;
	for(int p,i=2;i<=n;i++) read(p),add(p,i);
	dfs(1);
	printf("%d
",sum[root[1]]);
	return 0;
}

2019.5.21

原文地址:https://www.cnblogs.com/butterflydew/p/10901869.html