雨天的尾巴 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]);
}
}
小结:当时做这个题的时候感觉这辈子不会再碰了,但是学完主席树之后觉得就一傻逼题... ...