雨天的尾巴 bzoj-3307
题目大意:N个点,形成一个树状结构。有M次发放,每次选择两个点x,y对于x到y的路径上(含x,y)每个点发一袋Z类型的物品。完成所有发放后,每个点存放最多的是哪种物品。
想法:我们像主席树一样,维护桶然后对应节点合并即可。
最后,附上丑陋的代码... ...
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #define N 100050 using namespace std; int t[N*60],n,m,head[N],to[N<<1],nxt[N<<1],cnt,tot,mx[N*60],xx[N],yy[N],zz[N]; int fa[N],top[N],son[N],siz[N],dep[N],root[N],ls[N*60],rs[N*60],ans[N],v[N]; inline void add(int u,int v) { to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; } void dfs1(int x,int y) { int i; siz[x]=1; fa[x]=y; dep[x]=dep[y]+1; for(i=head[x];i;i=nxt[i]) { if(to[i]!=y) { dfs1(to[i],x); siz[x]+=siz[to[i]]; if(siz[to[i]]>siz[son[x]]) son[x]=to[i]; } } } void dfs2(int x,int t) { top[x]=t; if(son[x]) dfs2(son[x],t); int i; for(i=head[x];i;i=nxt[i]) { if(to[i]!=son[x]&&to[i]!=fa[x]) { dfs2(to[i],to[i]); } } } int lca(int x,int y) { while(top[x]!=top[y]) { if(dep[top[x]]>dep[top[y]]) swap(x,y); y=fa[top[y]]; } return dep[x]<dep[y]?x:y; } void pushup(int p) { if(t[ls[p]]>=t[rs[p]]) mx[p]=mx[ls[p]],t[p]=t[ls[p]]; else mx[p]=mx[rs[p]],t[p]=t[rs[p]]; } int merge(int l,int r,int x,int y) { if(!x) return y; if(!y) return x; if(l==r) { t[x]+=t[y]; return x; } int mid=(l+r)>>1; ls[x]=merge(l,mid,ls[x],ls[y]); rs[x]=merge(mid+1,r,rs[x],rs[y]); pushup(x); return x; } void update(int l,int r,int x,int v,int &p) { if(!p) p=++tot; if(l==r) { t[p]+=v; mx[p]=l; return ; } int mid=(l+r)>>1; if(x<=mid) update(l,mid,x,v,ls[p]); else update(mid+1,r,x,v,rs[p]); pushup(p); } void dfs(int x) { int i; for(i=head[x];i;i=nxt[i]) { if(to[i]!=fa[x]) { dfs(to[i]); root[x]=merge(1,maxn,root[x],root[to[i]]); } } if(t[root[x]]) ans[x]=v[mx[root[x]]]; } int main() { scanf("%d%d",&n,&m); int i,x,y,z; for(i=1;i<n;i++) { scanf("%d%d",&x,&y); add(x,y); add(y,x); } dfs1(1,0); dfs2(1,1); for(i=1;i<=m;i++) { scanf("%d%d%d",&xx[i],&yy[i],&zz[i]); v[i]=zz[i]; } sort(v+1,v+m+1); for(i=1;i<=m;i++) { zz[i]=lower_bound(v+1,v+m+1,zz[i])-v,x=xx[i],y=yy[i]; int l=lca(x,y); update(1,maxn,zz[i],1,root[x]); update(1,maxn,zz[i],1,root[y]); update(1,maxn,zz[i],-1,root[l]); if(fa[l]) update(1,maxn,zz[i],-1,root[fa[l]]); } dfs(1); for(i=1;i<=n;i++) { printf("%d\n",ans[i]); } }
小结:当时做这个题的时候感觉这辈子不会再碰了,但是学完主席树之后觉得就一傻逼题... ...
原文地址:https://www.cnblogs.com/ShuraK/p/9338840.html
时间: 2024-11-09 02:16:24