分析
看完T1T2以后看T3
噔 噔 咚 心 肺 停 止
又™是树链剖分?有了T1的经验我仔细再看了两眼
的确是链剖,只是当时没想到奇妙的堆乱搞,然后用链剖打了个暴力……(排序z暴力做链剖,O(nlog+nm)有70?!)
堆的写法好奇妙,但是我选择简单的树上差分和线段树合并……
差分就把x,y的z+1,lca的z-1,fa[lca]的z-1即可
最后自底向上统计答案,顺便合并线段树
(居然要我写BFS用深度来处理防爆栈?)
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#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 ",ans[i]); }