BZOJ4919:大根堆
Description:
题目描述
给定一棵n个节点的有根树,编号依次为1到n,其中1号点为根节点。每个点有一个权值v_i。 你需要将这棵树转化成一个大根堆。确切地说,你需要选择尽可能多的节点,满足大根堆的性质:对于任意两个点i,j,如果i在树上是j的祖先,那么v_i>v_j。 请计算可选的最多的点数,注意这些点不必形成这棵树的一个连通子树。
输入格式
第一行包含一个正整数n(1<=n<=200000),表示节点的个数。 接下来n行,每行两个整数v_i,p_i(0<=v_i<=10^9,1<=p_i<i,p_1=0),表示每个节点的权值与父亲。
输出格式
输出一行一个正整数,即最多的点数。
样例
样例输入
6
3 0
1 1
2 1
3 1
4 1
5 1
样例输出
5
思路:
我们可以先深搜到叶子,在叶子建树,然后直接向上合并到根,然后算出选或不选根节点的答案并比较,然后确定根节点要不要选,详见代码。
#include<bits/stdc++.h>
using namespace std;
namespace Decleration
{
#define ll long long
#define rr register
const int SIZE=200004;
int n,cnt;
int v_[SIZE],v[SIZE];
struct edge{int t,last;}a[SIZE];
int last[SIZE];
//快读
static char buf[100000],*pa=buf,*pd=buf;
#define gc pa==pd&&(pd=(pa=buf)+fread(buf,1,100000,stdin),pa==pd)?EOF:*pa++
inline int read()
{
register int x(0);register char c(gc);
while(c<'0'||c>'9')c=gc;
while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=gc;
return x;
}
//建边
inline void add(int f,int t)
{
a[++cnt].t=t;
a[cnt].last=last[f];
last[f]=cnt;
}
int cmp(const int a_cmp,const int b_cmp){return a_cmp<b_cmp;}
//二分,离散化
}
using namespace Decleration;
namespace Line_Forest
{
int num;
int rt[SIZE],sum[SIZE*20];
int lc[SIZE*20],rc[SIZE*20];
//插入
void updata(int &p,int L,int R,int l,int r,int z)
{
if(!p)
p=++num;
if(l<=L&&R<=r)
{
sum[p]+=z;
return;
}
int mid=(L+R)>>1;
if(l<=mid)
updata(lc[p],L,mid,l,r,z);
if(r>mid)
updata(rc[p],mid+1,R,l,r,z);
}
//合并
int merge(int p,int q,int l,int r)
{
if(!p)
return q;
if(!q)
return p;
sum[p]+=sum[q];
int mid=(l+r)>>1;
lc[p]=merge(lc[p],lc[q],l,mid);
rc[p]=merge(rc[p],rc[q],mid+1,r);
return p;
}
//查询小于等于val的个数
int query(int p,int l,int r,int z)
{
if(!p)
return 0;
rr int mid=(l+r)>>1;
rr int o=sum[p];
if(z<=mid)
o+=query(lc[p],l,mid,z);
else
o+=query(rc[p],mid+1,r,z);
return o;
}
//深搜
void dfs(int now)
{
for(rr int i=last[now];i!=-1;i=a[i].last)
{
dfs(a[i].t);
rt[now]=merge(rt[now],rt[a[i].t],1,cnt);
}
rr int ans1=query(rt[now],1,cnt,v[now]-1)+1;
rr int ans2=query(rt[now],1,cnt,v[now]);
if(ans1<=ans2) return;
int l=v[now],r=cnt,pos=v[now];
//假如说根节点要选的话,可能面临着权值比自己大的的节点的可选节点数
//比现有的最新的答案小,所以要二分一个满足这个条件的最大的区间,给这个区间加一
//这实际上是向后更新
//有DP那味了
while(l<=r)
{
int mid=(l+r)>>1;
if(query(rt[now],1,cnt,mid)<ans1)
{
l=mid+1;
pos=mid;
}
else r=mid-1;
}
updata(rt[now],1,cnt,v[now],pos,1);
}
};
using namespace Line_Forest;
int main()
{
//freopen("1.in","r",stdin);
//freopen("C.OUT","w",stdout);
n=read();
memset(last,-1,sizeof(last));
for(rr int i=1;i<=n;i++)
{
v[i]=v_[i]=read();
int fa=read();
if(!fa) continue;
add(fa,i);
}
sort(v_+1,v_+1+n,cmp);
cnt=unique(v_+1,v_+1+n)-v_-1;
for(rr int i=1;i<=n;i++)
v[i]=lower_bound(v_+1,v_+1+cnt,v[i])-v_;
dfs(1);
printf("%d\n",query(rt[1],1,cnt,cnt));
//fclose(stdout);
//fclose(stdin);
}
这道题一开始交代码时TLE70后来将快读改成了更快的,并把register开满才过了,这里吐槽一下HZ的评测机,着实慢的一批啊。。。。。
另外蒟蒻有一个问题:板子与板子之间有区别吗??同样是板子,为什么我一开始用下面的板子过不了然后换了个板子就过了。。。
void insert(int &id,int l,int r,int val)
{
if(!id) id=++num;
if(l==r)
{
sum[id]++;
return;
}
int mid=(l+r)>>1;
if(val<=mid) insert(lc[id],l,mid,val);
else insert(rc[id],mid+1,r,val);
sum[id]=sum[lc[id]]+sum[rc[id]];
}
int query_sum(int id,int l,int r,int val)
{
if(!id) return 0;
if(l==r) return sum[id];
int mid=(l+r)>>1;
if(val<=mid) return query_sum(lc[id],l,mid,val);
else return query_sum(rc[id],mid+1,r,val)+sum[lc[id]];
}
int merge(int id1,int id2,int l,int r)
{
if(!id1||!id2) return id1+id2;
if(l==r)
{
sum[id1]+=sum[id2];
return id1;
}
int mid=(l+r)>>1;
lc[id1]=merge(lc[id1],lc[id2],l,mid);
rc[id1]=merge(rc[id1],rc[id2],mid+1,r);
sum[id1]=sum[lc[id1]]+sum[rc[id1]];
return id1;
}
请求路过的大神解释,蒟蒻能力有限,自己想不明白。。。。。