【BZOJ4919】[Lydsy六月月赛]大根堆
Description
给定一棵n个节点的有根树,编号依次为1到n,其中1号点为根节点。每个点有一个权值v_i。
你需要将这棵树转化成一个大根堆。确切地说,你需要选择尽可能多的节点,满足大根堆的性质:对于任意两个点i,j,如果i在树上是j的祖先,那么v_i>v_j。
请计算可选的最多的点数,注意这些点不必形成这棵树的一个连通子树。
Input
第一行包含一个正整数n(1<=n<=200000),表示节点的个数。
接下来n行,每行两个整数v_i,p_i(0<=v_i<=10^9,1<=p_i<i,p_1=0),表示每个节点的权值与父亲。
Output
输出一行一个正整数,即最多的点数。
Sample Input
6
3 0
1 1
2 1
3 1
4 1
5 1
Sample Output
5
题解:考虑用f[i][j]表示在i节点的子树中,最大值<=j,最多能选择多少点。如何转移呢?父亲节点的f数组可以看成儿子节点的f数组对应位置相加。然后再用 当前点权值-1处的f值 +1 来更新当前点权值后面的所有f值。
为此,我们可以考虑用线段树+标记永久化维护,我们要实现的有:区间加,维护区间最大值。然后转移的时候可以直接用线段树合并搞定。细节还是比较多的。
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; const int maxn=200010; int n,m,cnt,tot,ans; int to[maxn],next[maxn],head[maxn],p[maxn],val[maxn],v[maxn],rt[maxn]; inline void add(int a,int b) { to[cnt]=b,next[cnt]=head[a],head[a]=cnt++; } bool cmp(const int &a,const int &b) { return val[a]<val[b]; } struct sag { int ls,rs,tag,sum; }s[maxn<<6]; inline void pushdown(int x) { if(s[x].ls) s[s[x].ls].sum+=s[x].sum,s[s[x].ls].tag=max(s[s[x].ls].tag+s[x].sum,s[x].tag); if(s[x].rs) s[s[x].rs].sum+=s[x].sum,s[s[x].rs].tag=max(s[s[x].rs].tag+s[x].sum,s[x].tag); s[x].sum=0; } int merge(int a,int b) { if(!a||!b) return a^b; pushdown(a),pushdown(b); if(!s[a].ls) s[a].ls=s[b].ls,s[s[a].ls].tag+=s[a].tag,s[s[a].ls].sum+=s[a].tag+s[a].sum; else if(!s[b].ls) s[s[a].ls].tag+=s[b].tag,s[s[a].ls].sum+=s[b].tag+s[b].sum; else s[a].ls=merge(s[a].ls,s[b].ls); if(!s[a].rs) s[a].rs=s[b].rs,s[s[a].rs].tag+=s[a].tag,s[s[a].rs].sum+=s[a].tag+s[a].sum; else if(!s[b].rs) s[s[a].rs].tag+=s[b].tag,s[s[a].rs].sum+=s[b].tag+s[b].sum; else s[a].rs=merge(s[a].rs,s[b].rs); s[a].tag+=s[b].tag; return a; } void updata(int l,int r,int &x,int a,int b,int c) { if(!x) x=++tot; if(a<=l&&r<=b) { s[x].tag=max(s[x].tag,c); return ; } pushdown(x); int mid=(l+r)>>1; if(a<=mid) updata(l,mid,s[x].ls,a,b,c); if(b>mid) updata(mid+1,r,s[x].rs,a,b,c); } int query(int l,int r,int x,int a) { if(!x||!a) return 0; if(l==r) return s[x].tag; pushdown(x); int mid=(l+r)>>1; if(a<=mid) return max(s[x].tag,query(l,mid,s[x].ls,a)); return max(s[x].tag,query(mid+1,r,s[x].rs,a)); } void dfs(int x) { for(int i=head[x];i!=-1;i=next[i]) dfs(to[i]),rt[x]=merge(rt[x],rt[to[i]]); updata(1,m,rt[x],v[x],m,query(1,m,rt[x],v[x]-1)+1); } void find(int x) { ans=max(ans,s[x].tag); pushdown(x); if(s[x].ls) find(s[x].ls); if(s[x].rs) find(s[x].rs); } inline int rd() { int ret=0,f=1; char gc=getchar(); while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘) f=-f; gc=getchar();} while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+(gc^‘0‘),gc=getchar(); return ret*f; } int main() { n=rd(); int i,a; memset(head,-1,sizeof(head)); for(i=1;i<=n;i++) { val[i]=rd(),a=rd(),p[i]=i; if(i!=1) add(a,i); } sort(p+1,p+n+1,cmp); for(i=1;i<=n;i++) { if(i==1||val[p[i]]>val[p[i-1]]) m++; v[p[i]]=m; } dfs(1),find(rt[1]); printf("%d",ans); return 0; }
时间: 2024-11-05 21:46:41