[洛谷 P4556] 雨天的尾巴
阿新 • • 發佈:2018-12-08
eight get max n) bits ref ret getch clu
傳送門
Solution
線段樹合並的入門題
lca可以在dfs的時候離線求(用並查集)
更新的點有每條鏈的兩個端點,它們的lca和dad[lca]
為了節省空間,lca和dad[lca]
的更新可以先存儲下來,等搜完該點後再做
Code
#include<bits/stdc++.h> #define ll long long #define max(a,b) ((a)>(b)?(a):(b)) #define min(a,b) ((a)<(b)?(a):(b)) inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();} return x*f; } #define MN 100005 #define N 100000 int n,m,z[MN],ans[MN]; struct ques{int b,nex;}q[MN<<1]; int qhr[MN],cnt=1; struct edge{int to,nex;}e[MN<<1]; int hr[MN],en; inline void qins(int f,int t) { q[++cnt]=(ques){t,qhr[f]};qhr[f]=cnt; if(f==t) ++cnt; else q[++cnt]=(ques){f,qhr[t]},qhr[t]=cnt; } inline void ins(int f,int t) { e[++en]=(edge){t,hr[f]};hr[f]=en; e[++en]=(edge){f,hr[t]};hr[t]=en; } int fa[MN],dad[MN]; bool vis[MN]; inline int getf(int x){return fa[x]==x?x:fa[x]=getf(fa[x]);} #define mid ((l+r)>>1) int ls[MN*40],rs[MN*40],rt[MN],t[MN*40],seg; void merge(int &x,int y,int l,int r) { if(!x||!y){x=x+y;return;} if(l==r){t[x]+=t[y];return;} merge(ls[x],ls[y],l,mid); merge(rs[x],rs[y],mid+1,r); t[x]=max(t[ls[x]],t[rs[x]]); } int query(int x,int l,int r) { if(!x||!t[x]) return 0; if(l==r) return l; return t[rs[x]]>t[ls[x]]?query(rs[x],mid+1,r):query(ls[x],l,mid); } void update(int &x,int l,int r,int p,int v) { if(!x) x=++seg; if(l==r){t[x]+=v;return;} if(p>mid) update(rs[x],mid+1,r,p,v); if(p<=mid) update(ls[x],l,mid,p,v); t[x]=max(t[ls[x]],t[rs[x]]); } std::map<int,int> mar[MN]; void dfs(int x,int f) { register int i; fa[x]=x;dad[x]=f; for(i=hr[x];i;i=e[i].nex)if(f^e[i].to) { dfs(e[i].to,x); fa[e[i].to]=x; merge(rt[x],rt[e[i].to],1,N); } vis[x]=true; for(i=qhr[x];i;i=q[i].nex)if(vis[q[i].b]==true) { int lca=getf(q[i].b); mar[lca][z[i>>1]]--;mar[dad[lca]][z[i>>1]]--; } std::map<int,int>::iterator it; for(it=mar[x].begin();it!=mar[x].end();++it) update(rt[x],1,N,it->first,it->second); ans[x]=query(rt[x],1,N); } int main() { n=read(),m=read(); register int i,j,k; for(i=1;i<n;++i) ins(read(),read()); for(i=1;i<=m;++i) { j=read(),k=read(),z[i]=read(); qins(j,k),update(rt[j],1,N,z[i],1),update(rt[k],1,N,z[i],1); } dfs(1,0); for(i=1;i<=n;++i) printf("%d\n",ans[i]); return 0; }
Blog來自PaperCloud,未經允許,請勿轉載,TKS!
[洛谷 P4556] 雨天的尾巴