[树上差分][线段树合并]JZOJ 3397 雨天的尾巴
Posted mastervan
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[树上差分][线段树合并]JZOJ 3397 雨天的尾巴相关的知识,希望对你有一定的参考价值。
分析
看完T1T2以后看T3
噔 噔 咚 心 肺 停 止
又™是树链剖分?有了T1的经验我仔细再看了两眼
的确是链剖,只是当时没想到奇妙的堆乱搞,然后用链剖打了个暴力……(排序z暴力做链剖,O(nlog+nm)有70?!)
堆的写法好奇妙,但是我选择简单的树上差分和线段树合并……
差分就把x,y的z+1,lca的z-1,fa[lca]的z-1即可
最后自底向上统计答案,顺便合并线段树
(居然要我写BFS用深度来处理防爆栈?)
#include <iostream> #include <cstdio> #include <algorithm> #include <queue> #define lson ls[x] #define rson rs[x] using namespace std; const int N=1e5+10; const int Inf=1e9; struct Graph int v,nx; g[2*N]; int cnt,list[N]; int t[100*N],ls[100*N],rs[100*N],scnt,root[N],f[N][20],dep[N]; int n,m,ans[N],a[N]; void Add(int u,int v) g[++cnt]=(Graph)v,list[u];list[u]=cnt; g[++cnt]=(Graph)u,list[v];list[v]=cnt; void BFS(int v0) queue<int> q; while (!q.empty()) q.pop(); q.push(v0); while (!q.empty()) int u=q.front();q.pop(); for (int i=list[u];i;i=g[i].nx) if (g[i].v!=f[u][0]) f[g[i].v][0]=u;dep[g[i].v]=dep[u]+1; q.push(g[i].v); int LCA(int u,int v) if (dep[u]<dep[v]) swap(u,v); for (int i=19;i>=0;i--) if (dep[f[u][i]]>=dep[v]) u=f[u][i]; if (u==v) return v; for (int i=19;i>=0;i--) if (f[u][i]!=f[v][i]) u=f[u][i],v=f[v][i]; return f[u][0]; void Insert(int &x,int l,int r,int k,int z) if (!x) x=++scnt; if (l==r) t[x]+=z; return; int mid=l+r>>1; if (k<=mid) Insert(lson,l,mid,k,z); else Insert(rson,mid+1,r,k,z); t[x]=max(t[lson],t[rson]); void Merge(int &x,int y,int l,int r) if (!y) return; if (!x) t[x=++scnt]=t[y],lson=ls[y],rson=rs[y]; return; if (l==r) t[x]+=t[y]; return; int mid=l+r>>1; Merge(lson,ls[y],l,mid);Merge(rson,rs[y],mid+1,r); t[x]=max(t[lson],t[rson]); int Locate(int x,int l,int r,int mx) if (l==r) return l; int mid=l+r>>1; if (t[lson]==mx) return Locate(lson,l,mid,mx); return Locate(rson,mid+1,r,mx); bool CMP(int a,int b) return dep[a]>dep[b]; int main() scanf("%d%d",&n,&m); for (int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),Add(u,v),a[i]=i;a[n]=n; dep[1]=1;BFS(1);sort(a+1,a+n+1,CMP); for (int i=1;i<20;i++) for (int j=1;j<=n;j++) f[j][i]=f[f[j][i-1]][i-1]; for (int i=1;i<=m;i++) int x,y,z,lca; scanf("%d%d%d",&x,&y,&z); lca=LCA(x,y); Insert(root[x],0,Inf,z,1);Insert(root[y],0,Inf,z,1); Insert(root[lca],0,Inf,z,-1);if (f[lca][0]) Insert(root[f[lca][0]],0,Inf,z,-1); for (int i=1;i<n;i++) int u=a[i]; int mx=t[root[u]]; ans[u]=Locate(root[u],0,Inf,mx); Merge(root[f[u][0]],root[u],0,Inf); int mx=t[root[1]]; ans[1]=Locate(root[1],0,Inf,mx); for (int i=1;i<=n;i++) printf("%d\n",ans[i]);
以上是关于[树上差分][线段树合并]JZOJ 3397 雨天的尾巴的主要内容,如果未能解决你的问题,请参考以下文章
P4556 [Vani有约会]雨天的尾巴 (线段树合并 + 树上差分)
P4556 [Vani有约会]雨天的尾巴 (线段树合并 + 树上差分)
P4556 [Vani有约会]雨天的尾巴 (线段树合并 + 树上差分)