#include<stdio.h>
#include<vector>
#include<algorithm>
#define st first
#define nd second
using namespace std;
int z1[500000],z2[500000],lca[500000],pre[200000],pos[200000],t;
vector<int>q[200000],root;
bool wyl[200000];
vector<int>v[200000],koszt[500000];
void pre_pos(int x){
pre[x]=t;
t++;
for(int i=0;i<v[x].size();i++)pre_pos(v[x][i]);
pos[x]=t;
t++;
}
bool ojc(int a,int b){
return (pre[a]<=pre[b]&&pos[a]>=pos[b]);
}
vector<int>kol;
int odpowiedz(int v){
//cout<<"szukam lca "<<v<<" w ";
//for(int i=0;i<kol.size();i++)cout<<kol[i]<<" ";
//cout<<"\n";
if(!ojc(kol[0],v))return -1;
int p=0,k=kol.size()-1,s;
while(p+1<k){
s=(p+k)/2;
if(ojc(kol[s],v))p=s;
else k=s-1;
}
if(ojc(kol[k],v))return kol[k];
else return kol[p];
}
int dfs_lca(int x){
kol.push_back(x);
for(int i=0;i<q[x].size();i++){
lca[q[x][i]]=odpowiedz(z2[q[x][i]]);
}
for(int i=0;i<v[x].size();i++)dfs_lca(v[x][i]);
kol.pop_back();
}
int waga[200000];
main(){
t=0;
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for(int i=0;i<n;i++)scanf("%d",&waga[i]);
for(int i=0;i<m;i++){
int a,b;
scanf("%d%d",&a,&b);
a--;b--;
wyl[a]=1;
v[b].push_back(a);
koszt[b].push_back(i);
}
for(int i=0;i<k;i++){
scanf("%d%d",&z1[i],&z2[i]);
z1[i]--;z2[i]--;
q[z1[i]].push_back(i);
}
for(int i=0;i<n;i++)if(!wyl[i])root.push_back(i);
for(int i=0;i<root.size();i++)pre_pos(root[i]);
for(int i=0;i<root.size();i++)dfs_lca(root[i]);
vector<pair<int,int> >zapyt;
//for(int i=0;i<k;i++)cout<<"lca "<<z1[i]<<" "<<z2[i]<<" to "<<lca[i]<<"\n";
for(int i=0;i<k;i++){
if(lca[i]==-1)continue;
int tura=0;
for(int j=0;j<v[lca[i]].size();j++){
if(ojc(v[lca[i]][j],z1[i]))tura=max(tura,koszt[lca[i]][j]);
if(ojc(v[lca[i]][j],z2[i]))tura=max(tura,koszt[lca[i]][j]);
}
zapyt.push_back(make_pair(tura,i));
}
sort(zapyt.begin(),zapyt.end());
long long wynik=0;
for(int i=0;i<zapyt.size();i++){
int tmp=min(waga[z1[zapyt[i].nd]],waga[z2[zapyt[i].nd]]);
wynik+=tmp*2;
waga[z1[zapyt[i].nd]]-=tmp;
waga[z2[zapyt[i].nd]]-=tmp;
}
printf("%lld\n",wynik);
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | #include<stdio.h> #include<vector> #include<algorithm> #define st first #define nd second using namespace std; int z1[500000],z2[500000],lca[500000],pre[200000],pos[200000],t; vector<int>q[200000],root; bool wyl[200000]; vector<int>v[200000],koszt[500000]; void pre_pos(int x){ pre[x]=t; t++; for(int i=0;i<v[x].size();i++)pre_pos(v[x][i]); pos[x]=t; t++; } bool ojc(int a,int b){ return (pre[a]<=pre[b]&&pos[a]>=pos[b]); } vector<int>kol; int odpowiedz(int v){ //cout<<"szukam lca "<<v<<" w "; //for(int i=0;i<kol.size();i++)cout<<kol[i]<<" "; //cout<<"\n"; if(!ojc(kol[0],v))return -1; int p=0,k=kol.size()-1,s; while(p+1<k){ s=(p+k)/2; if(ojc(kol[s],v))p=s; else k=s-1; } if(ojc(kol[k],v))return kol[k]; else return kol[p]; } int dfs_lca(int x){ kol.push_back(x); for(int i=0;i<q[x].size();i++){ lca[q[x][i]]=odpowiedz(z2[q[x][i]]); } for(int i=0;i<v[x].size();i++)dfs_lca(v[x][i]); kol.pop_back(); } int waga[200000]; main(){ t=0; int n,m,k; scanf("%d%d%d",&n,&m,&k); for(int i=0;i<n;i++)scanf("%d",&waga[i]); for(int i=0;i<m;i++){ int a,b; scanf("%d%d",&a,&b); a--;b--; wyl[a]=1; v[b].push_back(a); koszt[b].push_back(i); } for(int i=0;i<k;i++){ scanf("%d%d",&z1[i],&z2[i]); z1[i]--;z2[i]--; q[z1[i]].push_back(i); } for(int i=0;i<n;i++)if(!wyl[i])root.push_back(i); for(int i=0;i<root.size();i++)pre_pos(root[i]); for(int i=0;i<root.size();i++)dfs_lca(root[i]); vector<pair<int,int> >zapyt; //for(int i=0;i<k;i++)cout<<"lca "<<z1[i]<<" "<<z2[i]<<" to "<<lca[i]<<"\n"; for(int i=0;i<k;i++){ if(lca[i]==-1)continue; int tura=0; for(int j=0;j<v[lca[i]].size();j++){ if(ojc(v[lca[i]][j],z1[i]))tura=max(tura,koszt[lca[i]][j]); if(ojc(v[lca[i]][j],z2[i]))tura=max(tura,koszt[lca[i]][j]); } zapyt.push_back(make_pair(tura,i)); } sort(zapyt.begin(),zapyt.end()); long long wynik=0; for(int i=0;i<zapyt.size();i++){ int tmp=min(waga[z1[zapyt[i].nd]],waga[z2[zapyt[i].nd]]); wynik+=tmp*2; waga[z1[zapyt[i].nd]]-=tmp; waga[z2[zapyt[i].nd]]-=tmp; } printf("%lld\n",wynik); } |
English