cf1061E Politics (费用流)

看到数据范围，考虑网络流..但考的时候完全不知道怎么建图

考虑流量表示选的点个数，费用表示选点的收益，跑最大费用最大流

那么我用一个点x表示某树中的询问点x，刨去它子孙询问点的子树后的子树

对于树1，连边S->x，流量为x的限定数-孩子询问的限定数，费用为0

对于树2，连边x->T，流量为x的限定数-孩子询问的限定数，费用为0

以限制数量

之后对于每个点，给它在树1中最近的询问祖先到树2中最近的询问祖先 连边，流量为1，费用为点权

表示选这个点

如果S的出流量和不等于T的入流量和，或者最后没跑满，则说明无解

 #include<bits/stdc++.h>
 #define CLR(a,x) memset(a,x,sizeof(a))
 #define MP make_pair
 using namespace std;
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pa;
 const int maxn=;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 int teg[][maxn*][],tegh[][maxn],tect[];
 int dem[][maxn],cfa[][maxn],rt[];
 inline void adteg(int o,int a,int b){
     teg[o][++tect[o]][]=b,teg[o][tect[o]][]=tegh[o][a],tegh[o][a]=tect[o];
     teg[o][++tect[o]][]=a,teg[o][tect[o]][]=tegh[o][b],tegh[o][b]=tect[o];
 }
 struct Edge{
     int b,l,c,ne;
 }eg[maxn*];
 int N,egh[maxn*],ect=,S=,T=;
 int val[maxn],inl,outl;

 inline void ext(){
     printf("-1\n");
     exit();
 }

 inline void adeg(int a,int b,int l,int c){
     if(a==S) inl+=l;
     if(b==T) outl+=l;
     eg[++ect]=(Edge){b,l,c,egh[a]},egh[a]=ect;
     eg[++ect]=(Edge){a,,-c,egh[b]},egh[b]=ect;
 }
 inline int dfs(int o,int x,int f,int cl){
     if(dem[o][x]) cl=x;
     cfa[o][x]=cl;
     int n=;
     for(int i=tegh[o][x];i;i=teg[o][i][]){
         int b=teg[o][i][];if(b==f) continue;
         n+=dfs(o,b,x,cl);
     }
     if(dem[o][x]){
         if(dem[o][x]<n) ext();
         if(o==) adeg(S,x,dem[o][x]-n,);
         else adeg(x+N,T,dem[o][x]-n,);
         return dem[o][x];
     }return n;
 }

 queue<int> q;
 int dis[maxn*],ine[maxn*];
 bool flag[maxn*];

 inline bool spfa(){
     CLR(dis,-);dis[S]=;
     CLR(ine,);q.push(S);
     while(!q.empty()){
         int p=q.front();q.pop();
         flag[p]=;
         for(int i=egh[p];i;i=eg[i].ne){
             int b=eg[i].b;if(!eg[i].l) continue;
             if(dis[b]<dis[p]+eg[i].c){
                 dis[b]=dis[p]+eg[i].c,ine[b]=i;
                 if(!flag[b]) q.push(b),flag[b]=;
             }
         }
     }return dis[T]>=-1e9;
 }

 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     N=rd(),rt[]=rd(),rt[]=rd();
     for(i=;i<=N;i++)
         val[i]=rd();
     for(i=;i<N;i++) adteg(,rd(),rd());
     for(i=;i<N;i++) adteg(,rd(),rd());
     for(i=rd();i;i--){
         int x=rd(),y=rd();
         dem[][x]=y;
     }for(i=rd();i;i--){
         int x=rd(),y=rd();
         dem[][x]=y;
     }
     dfs(,rt[],,);dfs(,rt[],,);
     for(i=;i<=N;i++){
         adeg(cfa[][i],cfa[][i]+N,,val[i]);
     }
     if(inl!=outl) ext();
     int ans=;
     while(spfa()){
         int mi=1e9;
         for(i=T;i!=S;i=eg[ine[i]^].b){
             mi=min(mi,eg[ine[i]].l);
         }ans+=mi*dis[T];
         inl-=mi;
         for(i=T;i!=S;i=eg[ine[i]^].b){
             eg[ine[i]].l-=mi,eg[ine[i]^].l+=mi;
         }
     }
     if(inl) ext();
     printf("%d\n",ans);
     return ;
 }