SGU 507 Treediff (线段树合并)
2017-08-20 19:12
239 查看
求每个子树的叶子节点差值的最小值,合并时用右子树的最小值和左子树的最大值的差更新答案
#include<stdio.h> #include<string.h> #include<iostream> #include<algorithm> #include<vector> using namespace std; vector<int>ve; const int MAXN=(int)5e4+10; const int INF =INT_MAX; int root[MAXN],tot,ls[MAXN*17],rs[MAXN*17],sz[MAXN*17]; int d[MAXN*17],vl[MAXN],ma[MAXN*17],mi[MAXN*17]; void insert(int l,int r,int x,int &rt){ if(rt==0)rt=++tot; ma[rt]=mi[rt]=x;sz[rt]=1;d[rt]=INF; if(l==r)return ; int mid=l+r>>1; if(mid>=x)insert(l,mid,x,ls[rt]); else insert(mid+1,r,x,rs[rt]); } int merge(int u,int v){ if(u==0||v==0)return u|v; ls[u]=merge(ls[u],ls[v]); rs[u]=merge(rs[u],rs[v]); sz[u]+=sz[v]; mi[u]=min(mi[u],mi[v]); ma[u]=max(ma[u],ma[v]); d[u]=min(d[u],d[v]); if(ls[u]!=0&&rs[u]!=0)d[u]=min(d[u],ve[mi[rs[u]]-1]-ve[ma[ls[u]]-1]); if(ls[u]!=0)d[u]=min(d[u],d[ls[u]]); if(rs[u]!=0)d[u]=min(d[u],d[rs[u]]); if(ls[u]==0&&rs[u]==0&&sz[u]>1)d[u]=0; return u; } struct edge{int to,nxt;}ed[MAXN<<1]; int head[MAXN],cnt; int ans[MAXN]; void addedge(int u,int v){ ed[cnt].to=v; ed[cnt].nxt=head[u]; head[u]=cnt++; } int n,m,vsz; void dfs(int u){ for(int i=head[u];i!=-1;i=ed[i].nxt){ int v=ed[i].to; dfs(v); root[u]=merge(root[u],root[v]); } if(u+m>n)insert(1,vsz,vl[u],root[u]); else ans[u]=d[root[u]]; } int main() { memset(head,-1,sizeof(head)); scanf("%d%d",&n,&m); for(int i=2;i<=n;i++){ int x; scanf("%d",&x); addedge(x,i); } for(int i=1;i<=m;i++){ scanf("%I64d",&vl[n-m+i]); ve.push_back(vl[n-m+i]); } sort(ve.begin(),ve.end()); ve.erase(unique(ve.begin(),ve.end()),ve.end()); vsz=ve.size(); for(int i=n-m+1;i<=n;i++)vl[i]=lower_bound(ve.begin(),ve.end(),vl[i])-ve.begin()+1; dfs(1); for(int i=1;i<=n-m;i++)printf("%d ",ans[i]); printf("\n"); return 0; }
相关文章推荐
- SGU 507 Treediff <启发式合并>
- SGU 507 Treediff(平衡树)
- sgu 507 (启发式合并)
- 【bzoj1977】[BeiJing2010组队]次小生成树 Tree 最小生成树+权值线段树合并
- hdu_4718_The LCIS on the Tree(树链剖分+线段树合并)
- hdu_4718_The LCIS on the Tree(树链剖分+线段树合并)
- 【HDU 4718】 The LCIS on the Tree 【树链剖分+线段树合并】
- SPOJ COT3 Combat on a tree(Trie树、线段树的合并)
- POI - 18 - Tree Rotation (线段树合并)
- HDU 4718 The LCIS on the Tree (树链剖分+线段树区间合并)
- AOJ 2450 Do use segment tree (树链剖分 + 线段树区间合并)
- HDU 4718 The LCIS on the Tree (树链剖分 + 线段树区间合并)
- BZOJTree Rotations(线段树的合并)
- Implementation:Segment Tree 线段树
- poj3667 (线段树区间合并)
- [HDU 3308]LCIS[线段树][区间合并]
- poj 3237 Tree 树链剖分 线段树
- hdoj 3308 LCIS 【线段树单点更新 + 区间合并】【求解最长递增序列 的长度】
- [BZOJ]2653: middle 线段树合并+二分
- 2017ACM/ICPC广西邀请赛 K- Query on A Tree trie树合并