[arc063e]Integers on a Tree

题目大意

一颗点权树，相邻节点点权差绝对值为1。
现在一些点点权已确定，构造一种方案。

做法

自下而上推出每个节点点权区间范围。
为空则无解，同时如果儿子间对该点奇偶性要求不同也无解。
然后接下来只需自上而下构造，只要和父亲相差1，且点权在区间范围内一定可以合法。

#include<cstdio>#include<algorithm>#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;const int maxn=100000+10;int h[maxn],go[maxn*2],nxt[maxn*2],L[maxn],R[maxn],v[maxn];bool bz[maxn];int i,j,k,l,t,n,m,tot;bool czy;void add(int x,int y){    go[++tot]=y;    nxt[tot]=h[x];    h[x]=tot;}void dfs(int x,int y){    int t=h[x];    while (t){        if (go[t]!=y){            dfs(go[t],x);            if (!czy) return;            bz[x]|=bz[go[t]];        }        t=nxt[t];    }    if (v[x]!=-1) bz[x]=1;    if (!bz[x]) return;    int l,r;    bool gjx=0;    if (v[x]!=-1){        gjx=1;        l=r=v[x];    }    t=h[x];    while (t){        if (go[t]!=y&&bz[go[t]]){            if (!gjx){                gjx=1;                l=L[go[t]]-1;                r=R[go[t]]+1;            }            else{                if (R[go[t]]+1<l||L[go[t]]-1>r||l%2!=(L[go[t]]-1)%2){                    czy=0;                    return;                }                l=max(l,L[go[t]]-1);                r=min(r,R[go[t]]+1);            }        }        t=nxt[t];    }    L[x]=l;R[x]=r;}void dg(int x,int y){    if (x!=1){        if (!bz[x]) v[x]=v[y]+1;        else if (L[x]<=v[y]+1&&v[y]+1<=R[x]) v[x]=v[y]+1;        else v[x]=v[y]-1;    }    int t=h[x];    while (t){        if (go[t]!=y) dg(go[t],x);        t=nxt[t];    }}int main(){    scanf("%d",&n);    fo(i,1,n-1){        scanf("%d%d",&j,&k);        add(j,k);add(k,j);    }    fo(i,1,n) v[i]=-1;    scanf("%d",&m);    fo(i,1,m){        scanf("%d%d",&j,&k);        v[j]=k;    }    czy=1;    dfs(1,0);    if (!czy){        printf("No\n");        return 0;    }    printf("Yes\n");    v[1]=L[1];    dg(1,0);    fo(i,1,n) printf("%d\n",v[i]);}