记录编号 471571 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 [NOIP 2016]天天爱跑步 最终得分 100
用户昵称 Gravatar~玖湫~ 是否通过 通过
代码语言 C++ 运行时间 4.238 s
提交时间 2017-11-06 17:25:51 内存使用 158.25 MiB
显示代码纯文本
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const int M=300000+10;
#define mem(a) memset(a,0,sizeof(a))
int n,m,cnt,num,dfn;
int head[2*M],dep[M],in[M],out[M],tim[M],ans[M],fa[M][22];
int sum[25*M],lc[25*M],rc[25*M],root[25*M];
struct DATE{int fr,to,last;}date[2*M];
struct P{int s,t,lca;}p[M];
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*10+ch-'0';ch=getchar();}
    return x*f;
}
inline void add(int x,int y){
    date[++num]=(DATE){x,y,head[x]};
    head[x]=num;
}
void dfs(int x){
    in[x]=++dfn;
    for(int i=1;(1<<i)<=dep[x];i++)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int i=head[x];i;i=date[i].last){
        int to=date[i].to;
        if(to!=fa[x][0]){
            fa[to][0]=x;
            dep[to]=dep[x]+1;
            dfs(to);
        }
    }out[x]=dfn;
}
inline int getlca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=20;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            x=fa[x][i];
    if(x==y) return x;
    for(int i=20;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void update(int pos,int zhi,int &rt,int l,int r){
    if(!pos)  return ;
    if(!rt) rt=++cnt;
    sum[rt]+=zhi;
    if(l==r) return ;
    int mid=l+r>>1;
    if(pos<=mid) update(pos,zhi,lc[rt],l,mid);
    else         update(pos,zhi,rc[rt],mid+1,r);
}
int query(int s,int t,int &rt,int l,int r){
    if(!rt) return 0;
    if(s<=l&&r<=t) return sum[rt];
    int mid=l+r>>1; int res=0;
    if(s<=mid) res+=query(s,t,lc[rt],l,mid);
    if(t>mid) res+=query(s,t,rc[rt],mid+1,r);
    return res;
}
int main(){
    freopen("runninga.in","r",stdin);
    freopen("runninga.out","w",stdout);
    n=read();m=read();int aa,bb;
    for(int i=1;i<n;i++){
        aa=read();bb=read();
        add(aa,bb);add(bb,aa);
    } dep[1]=1;dfs(1);
    for(int i=1;i<=n;i++) tim[i]=read();
    for(int i=1;i<=m;i++){
        p[i].s=read();p[i].t=read();
        p[i].lca=getlca(p[i].s,p[i].t);
    }
    for(int i=1;i<=m;i++){
        update(in[p[i].s],1,root[dep[p[i].s]],1,n);
        update(in[fa[p[i].lca][0]],-1,root[dep[p[i].s]],1,n);
    }
    for(int i=1;i<=n;i++)
        ans[i]+=query(in[i],out[i],root[dep[i]+tim[i]],1,n);
    cnt=0;mem(sum);mem(root);mem(lc);mem(rc);
    for(int i=1;i<=m;i++){
        int d=(dep[p[i].s]-(dep[p[i].lca]<<1)+500000);
        update(in[p[i].t],1,root[d],1,n);
        update(in[p[i].lca],-1,root[d],1,n);
    }
    for(int i=1;i<=n;i++)
        ans[i]+=query(in[i],out[i],root[tim[i]-dep[i]+500000],1,n);
    for(int i=1;i<=n;i++)
        printf("%d ",ans[i]);
    return 0;
}