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

「FJOI2018」领导集团问题

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


  • 一个显然的\(dp\)

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

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

    设为\(f_{i,j}=\max\limits_{k\ge 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\n",sum[root[1]]);
	return 0;
}

2019.5.21

posted @ 2019-05-21 19:34  露迭月  阅读(363)  评论(0编辑  收藏  举报