bzoj4719/洛谷1600 noip2016天天爱跑步

题外话

呃，话说当年我好像连暴力25分都没拿满哈？学OI几个月的萌新直接上刑场考场也是没有办法的事情。
现在我改了noip2016的所有题目，祝愿我noip2017rp++，不要再被大佬们往死里踩了。

题目分析

对于每一条路径(s,t),我们令o=lca(s,t),则路径可以拆成s->o和o->t，最后处理o被重复算的那一次。dep表示某个节点在树上的深度，w表示某个节点上观察员出现的时间。
对于s->o路径上的某个点i,我们会发现一个人被观测到的条件是：w[i]=dep[s]-dep[i]，即w[i]+dep[i]=dep[s];
对于o->t路径上的某个点j，我们会发现一个人被观测到的条件是：w[j]=lon-dep[t]+dep[j]，（lon表示(s,t)的路径长度），即dep[j]-w[j]=dep[t]-lon
好的好的，现在我们用dfs来解决问题。
准确的来说，是用桶。用两个桶分别保存dep[s]和dep[t]-lon，注意后者可能为负数，要加上一个300000存入桶中。
对于某一个点x,s->o的路径可以对其造成贡献，则s是x的后代。同理，如果o->t的路径可以对其造成贡献，则t是x的后代。所以我们预先记录两个桶里不是x后代的dep[s]和dep[t]-lon的贡献，然后处理x的所有后代，最后获取答案。这一段核心语言好像真的讲不太清楚，看代码吧，总之是标记每一个点上两个桶的增加量和减少量来实现的，注意o被重复计算的次数。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define pb push_back
int read(){
    int q=0;char ch=' ';
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9')q=q*10+ch-'0',ch=getchar();
    return q;
}
const int N=300000;
int n,m,tot;
int h[N],ne[N<<1],to[N<<1],ss[N<<1],tt[N<<1];
int f[N][22],ans[N],dep[N],w[N],js[N];
vector<int>a[N],b[N],c[N];
void add(int x,int y){to[++tot]=y,ne[tot]=h[x],h[x]=tot;}
void dfs(int x,int las,int d){
    int i;dep[x]=d;
    f[x][0]=las;for(i=1;i<=20;++i)f[x][i]=f[f[x][i-1]][i-1];
    for(i=h[x];i!=-1;i=ne[i])if(to[i]!=las)dfs(to[i],x,d+1);
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    for(int i=20;i>=0;--i)if(dep[f[x][i]]>=dep[y])x=f[x][i];
    if(x==y)return x;
    for(int i=20;i>=0;--i)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
    return f[x][0];
}
void work(int x,int las){
    int i,sz,skl=ss[w[x]+dep[x]],tkl=tt[dep[x]-w[x]+N];
    //skl,tkl:把不会对x造成影响的桶里东西预先存一下
    ss[dep[x]]+=js[x];
    sz=a[x].size();for(i=0;i<sz;++i)++tt[a[x][i]+N];//往桶里添加元素
    for(i=h[x];i!=-1;i=ne[i])if(to[i]!=las)work(to[i],x);//处理子树
    ans[x]=ss[w[x]+dep[x]]+tt[dep[x]-w[x]+N]-skl-tkl;//计算答案
    sz=b[x].size();
    for(i=0;i<sz;++i){
        --ss[b[x][i]];
        if(w[x]+dep[x]==b[x][i])--ans[x];//lca会重复计算一次
    }
    sz=c[x].size();for(i=0;i<sz;++i)--tt[c[x][i]+N];//从桶里去除元素
}
int main(){
    int i,j,x,y,len,o;
    n=read(),m=read();
    for(i=1;i<=n;++i)h[i]=-1;
    for(i=1;i<n;++i)x=read(),y=read(),add(x,y),add(y,x);
    dfs(1,0,1);
    for(i=1;i<=n;++i)w[i]=read();
    for(i=1;i<=m;++i){
        x=read(),y=read(),o=lca(x,y),++js[x];//js:以x为起点的跑步者个数
        len=dep[x]+dep[y]-dep[o]*2;
        a[y].pb(dep[y]-len),b[o].pb(dep[x]),c[o].pb(dep[y]-len);
        //a:dfs时在y上要往桶tt中添加dep[y]-len，b:在o上要从桶中去除dep[x],c:在o上要从桶中去除dep[y]-len
    }
    work(1,0);
    for(i=1;i<n;++i)printf("%d ",ans[i]);printf("%d",ans[n]);
    return 0;
}