【JZOJ5262】【GDOI2018模拟8.12】树(DP，性质题)

Description

Solution

首先我们可以知道两个性质：1、路径u-v和路径v-w可以合并为路径u-w；2、路径u1-v1加路径u2-v2和路径u1-v2加路径u2-v1是等价的(就是起始点和终点可以互换)
那么知道这些性质之后就很好做了。我们只用知道每个点多少次做起点和多少次做终点。
我们设f[i]表示满足i子树的需求i上的值要是多少。
那么枚举i的所有儿子，判断a[i]-f[i]，看看当前是需要增大还是需要变小，然后再看看哪一个的编号大，判断路径方向，然后更新in和out(表示出度和入度)
然后起点终点去个min减一下就知道每个点做起点或终点多少次了，然后把所有的起点和终点排一次序，两两连接就好了。

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
#define rep(i,a) for(i=first[a];i;i=next[i])
using namespace std;
const int maxn=1e6+7;
int i,j,k,l,t,n,m,ans,head,tail,x,y,z,fu;
int first[maxn*2],last[maxn*2],next[maxn*2],num,a[maxn];
int f[maxn],d[maxn],fa[maxn],in[maxn],out[maxn];
int b[maxn],c[maxn];
void add(int x,int y){
    last[++num]=y,next[num]=first[x],first[x]=num;
}
int main(){
//  freopen("fan.in","r",stdin);
//  freopen("fan.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n)scanf("%d",&a[i]);
    fo(i,1,n-1)scanf("%d%d",&k,&l),add(k,l),add(l,k);
    head=0,d[tail=1]=1;
    while(head<tail){
        rep(i,d[++head]){
            if(last[i]==fa[d[head]])continue;
            fa[last[i]]=d[head];d[++tail]=last[i];
        }
    }
    fod(k,n,1){
        x=d[k];
        rep(i,x){
            y=last[i];if(y==fa[x])continue;
            z=a[y]-f[y];
            if(z>0){
                if(x>y)in[x]+=z,out[y]+=z,f[x]+=z;
                else out[x]+=z,in[y]+=z,f[x]+=z;
            }
            else{
                z=-z;
                if(x>y)in[y]+=z,out[x]+=z,f[x]-=z;
                else out[y]+=z,in[x]+=z,f[x]-=z;
            }
        }
    }
    fo(i,1,n){
        j=min(in[i],out[i]),in[i]-=j,out[i]-=j;
        while(in[i])in[i]--,c[++c[0]]=i;
        while(out[i])out[i]--,b[++b[0]]=i;
    }
    printf("%d\n",b[0]);
    fo(i,1,b[0])printf("%d %d\n",b[i],c[i]);
}