待改 洛谷1600

45分,垃圾cgold;

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int MAXN=600010;
int n,m,tot,tim;
int fst[MAXN],nxt[MAXN],cdis[MAXN],cnt[MAXN],csum[MAXN];
int deep[MAXN],fa[MAXN],top[MAXN],son[MAXN],siz[MAXN];
int ll[MAXN],rr[MAXN];//一条链;
int dfn[MAXN],tmp[MAXN];//差分; 
struct hh
{
    int from,to;
}ma[MAXN];
struct sh
{
    int from,to,lca;
}ss[MAXN];
void build(int f,int t)
{
    ma[++tot]=(hh){f,t};
    nxt[tot]=fst[f];
    fst[f]=tot;
    return;
}

void dfs(int x,int f)
{
    siz[x]=1,fa[x]=f;
    deep[x]=deep[f]+1;
    dfn[++tim]=x;
    for(int i=fst[x];i;i=nxt[i])
    {
        int v=ma[i].to;
        if(v==f) continue;
        csum[v]=csum[x]+1;
        dfs(v,x),siz[x]+=siz[v];
        if(!son[x] || siz[son[x]]<siz[v]) son[x]=v;
    }
    return;
}

void rdfs(int x,int st)
{
    top[x]=st;
    if(!son[x]) return;
    rdfs(son[x],st);
    for(int i=fst[x];i;i=nxt[i])
    {
        int v=ma[i].to;
        if(v==fa[x] || v==son[x]) continue;
        rdfs(v,v);
    }
    return;
}

int get_lca(int x,int y)
{
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(deep[fx]<deep[fy]) swap(x,y),swap(fx,fy);
        x=fa[fx],fx=top[x];
    }
    return deep[x]>deep[y]?y:x;
}


void calc(int x,int y)//n^n;
{
    int sum=0,ccnt=0,lc=get_lca(x,y);
    int l=csum[x]+csum[y]-2*csum[lc];

    if(x==lc)//lca->y;
    {
        if(cdis[y]==l) cnt[y]++;
        while(y!=lc)
        {
            sum++,y=fa[y];
            if(cdis[y]==l-sum) cnt[y]++;
        }
        return;
    }
    else if(y==lc)//x->lca;
    {
        if(!cdis[x]) cnt[x]++;
        while(x!=lc)
        {
            sum++,x=fa[x];
            if(cdis[x]==sum) cnt[x]++;
        }
        return;
    }    
    else//x->lca,lca->y;
    {
        if(!cdis[x]) cnt[x]++;    
        while(x!=lc)//x->lca
        {
            x=fa[x],sum++;
            if(cdis[x]==sum) cnt[x]++;
        }

        if(cdis[y]==l) cnt[y]++;
        while(y!=lc) //lca->y;
        {
            y=fa[y],ccnt++;
            if(cdis[y]==l-ccnt) cnt[y]++;
        }
    }
    return;
}

void solve_lian()
{
    int ccans=0;

    for(int i=1;i<=n;i++)
    {
        if(i-cdis[i]>=1) ll[i]=i-cdis[i];
        if(i+cdis[i]<=n) rr[i]=i+cdis[i];
    }

    for(int i=1;i<=m;i++)
        tmp[ss[i].from]++,tmp[ss[i].to+1]--;

    for(int i=1;i<=n;i++) tmp[i]+=tmp[i-1];

    for(int i=1;i<=n;i++)
    {
        int cccnt=0;
        if(ll[i] && !cdis[ll[i]])
        {
            ccans=min(tmp[ll[i]],tmp[i]);
            cnt[i]+=ccans;
            cccnt=tmp[i]-ccans;
        }
        else if(rr[i] && !cdis[rr[i]])
        {
            ccans=min(tmp[rr[i]],cccnt);
            cnt[i]+=ccans;
        }
    }

    for(int i=1;i<=n;i++) printf("%d ",cnt[i]);

    return;
}

void solve_sta()
{
    dfs(1,0);

    for(int i=1;i<=m;i++)
    {
        tmp[ss[i].from]++;
        tmp[ss[i].to]++;
        tmp[get_lca(ss[i].from,ss[i].to)]-=2;
    }

    for(int i=n;i>=1;i--) 
        tmp[fa[dfn[i]]]+=tmp[dfn[i]];//差分记录路径;

    tmp[1]=m;//注意; 

    for(int i=1;i<=n;i++)
        if(csum[i]==cdis[i]) 
            cnt[i]+=tmp[i];

    for(int i=1;i<=n;i++) printf("%d ",cnt[i]);

    return;
}
/*
void solve_end()
{
    dfs(1,0);

    for(int i=1;i<=m;i++)
    {
        tmp[ss[i].from]++;
        tmp[ss[i].to]++;
        tmp[get_lca(ss[i].from,ss[i].to)]-=2;
    }

    for(int i=n;i>=1;i--) 
        tmp[fa[dfn[i]]]+=tmp[dfn[i]];//差分记录路径;


}

*/
void solve()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        build(x,y),build(y,x);
    }

    for(int i=1;i<=n;i++)
        scanf("%d",&cdis[i]);

    for(int i=1;i<=m;i++) scanf("%d%d",&ss[i].from,&ss[i].to);

    if(n<=1000 && m<=1000)
    {
        dfs(1,0),rdfs(1,1);
        for(int i=1;i<=m;i++)
            calc(ss[i].from,ss[i].to);
        for(int i=1;i<=n;i++) printf("%d ",cnt[i]);
    }

    else if(n==99994 && m==99994) solve_lian();
    else if(n==99995 && m==99995) solve_sta();
//    else if(n==99996 && m==99996) solve_end();
    return;
}

int main()
{
    solve();
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值