cf1061E Politics (费用流)

  1 #include<bits/stdc++.h>
  2 #define CLR(a,x) memset(a,x,sizeof(a))
  3 #define MP make_pair
  4 using namespace std;
  5 typedef long long ll;
  6 typedef unsigned long long ull;
  7 typedef pair<int,int> pa;
  8 const int maxn=505;
  9 
 10 inline ll rd(){
 11     ll x=0;char c=getchar();int neg=1;
 12     while(c<‘0‘||c>‘9‘){if(c==‘-‘) neg=-1;c=getchar();}
 13     while(c>=‘0‘&&c<=‘9‘) x=x*10+c-‘0‘,c=getchar();
 14     return x*neg;
 15 }
 16 
 17 int teg[2][maxn*2][2],tegh[2][maxn],tect[2];
 18 int dem[2][maxn],cfa[2][maxn],rt[2];
 19 inline void adteg(int o,int a,int b){
 20     teg[o][++tect[o]][0]=b,teg[o][tect[o]][1]=tegh[o][a],tegh[o][a]=tect[o];
 21     teg[o][++tect[o]][0]=a,teg[o][tect[o]][1]=tegh[o][b],tegh[o][b]=tect[o];
 22 }
 23 struct Edge{
 24     int b,l,c,ne;
 25 }eg[maxn*6];
 26 int N,egh[maxn*2],ect=1,S=1004,T=1005;
 27 int val[maxn],inl,outl;
 28 
 29 inline void ext(){
 30     printf("-1\n");
 31     exit(0);
 32 }
 33 
 34 inline void adeg(int a,int b,int l,int c){
 35     if(a==S) inl+=l;
 36     if(b==T) outl+=l;
 37     eg[++ect]=(Edge){b,l,c,egh[a]},egh[a]=ect;
 38     eg[++ect]=(Edge){a,0,-c,egh[b]},egh[b]=ect;
 39 }
 40 inline int dfs(int o,int x,int f,int cl){
 41     if(dem[o][x]) cl=x;
 42     cfa[o][x]=cl;
 43     int n=0;
 44     for(int i=tegh[o][x];i;i=teg[o][i][1]){
 45         int b=teg[o][i][0];if(b==f) continue;
 46         n+=dfs(o,b,x,cl);
 47     }
 48     if(dem[o][x]){
 49         if(dem[o][x]<n) ext();
 50         if(o==0) adeg(S,x,dem[o][x]-n,0);
 51         else adeg(x+N,T,dem[o][x]-n,0);
 52         return dem[o][x];
 53     }return n;
 54 }
 55 
 56 queue<int> q;
 57 int dis[maxn*2],ine[maxn*2];
 58 bool flag[maxn*2];
 59 
 60 inline bool spfa(){
 61     CLR(dis,-127);dis[S]=0;
 62     CLR(ine,0);q.push(S);
 63     while(!q.empty()){
 64         int p=q.front();q.pop();
 65         flag[p]=0;
 66         for(int i=egh[p];i;i=eg[i].ne){
 67             int b=eg[i].b;if(!eg[i].l) continue;
 68             if(dis[b]<dis[p]+eg[i].c){
 69                 dis[b]=dis[p]+eg[i].c,ine[b]=i;
 70                 if(!flag[b]) q.push(b),flag[b]=1;
 71             }
 72         }
 73     }return dis[T]>=-1e9;
 74 }
 75 
 76 int main(){
 77     //freopen("","r",stdin);
 78     int i,j,k;
 79     N=rd(),rt[0]=rd(),rt[1]=rd();
 80     for(i=1;i<=N;i++)
 81         val[i]=rd();
 82     for(i=1;i<N;i++) adteg(0,rd(),rd());
 83     for(i=1;i<N;i++) adteg(1,rd(),rd());
 84     for(i=rd();i;i--){
 85         int x=rd(),y=rd();
 86         dem[0][x]=y;
 87     }for(i=rd();i;i--){
 88         int x=rd(),y=rd();
 89         dem[1][x]=y;
 90     }
 91     dfs(0,rt[0],0,0);dfs(1,rt[1],0,0);
 92     for(i=1;i<=N;i++){
 93         adeg(cfa[0][i],cfa[1][i]+N,1,val[i]);
 94     }
 95     if(inl!=outl) ext();
 96     int ans=0;
 97     while(spfa()){
 98         int mi=1e9;
 99         for(i=T;i!=S;i=eg[ine[i]^1].b){
100             mi=min(mi,eg[ine[i]].l);
101         }ans+=mi*dis[T];
102         inl-=mi;
103         for(i=T;i!=S;i=eg[ine[i]^1].b){
104             eg[ine[i]].l-=mi,eg[ine[i]^1].l+=mi;
105         }
106     }
107     if(inl) ext();
108     printf("%d\n",ans);
109     return 0;
110 }