#include<cstdio> #include<iostream> #include<algorithm> #include<vector> #include<tr1/unordered_map> #include<queue> #include<cstdlib> #include<list> #include<set> #include<map> #include<cmath> #define MP make_pair #define PB push_back #define s second #define f first #define PII pair<int,int> #define VPII vector <pair<int,int> > #define VI vector <int> #define abs(a) max((a),-(a)) #define LL long long #define LD long double #define ALL(x) x.begin(),x.end() #define PU putchar_unlocked #define GU getchar_unlocked #define DBG(x) cerr<<#x<<" "<<(x)<<endl; using namespace std; int a,b,c,d,e,f,n,m,mx,l,r,k; LL wynik; char ch; int INF=1e9+1; const int N=2e5+4; int t[N]; VI fio[N]; int ile[N]; PII kroki[N]; VPII V[N]; void scal(int a,int b) { a=t[a]; b=t[b]; vector< pair<int, PII > > X; PII y; if(fio[a].size()>fio[b].size())swap(a,b); for(int i=0;i<fio[a].size();i++) { for(int j=0;j<V[fio[a][i]].size();j++) { y=V[fio[a][i]][j]; if(t[y.s]==b)X.PB(MP(y.f,MP(fio[a][i],y.s))); } } for(int i=0;i<fio[a].size();i++) { t[fio[a][i]]=b; fio[b].PB(fio[a][i]); } fio[a].clear(); //fio[a].swap(VI()); sort(ALL(X)); for(int i=0;i<X.size();i++) { a=X[i].s.f; b=X[i].s.s; c=min(ile[b],ile[a]); ile[a]-=c; ile[b]-=c; // printf("%d %d %d\n",a,b,c); wynik+=c*2; } } void solve() { scanf("%d%d%d",&n,&m,&k); for(int i=1;i<=n;i++) { t[i]=i; fio[i].PB(i); scanf("%d",&ile[i]); } for(int i=1;i<=m;i++) { scanf("%d%d",&kroki[i].f,&kroki[i].s); } for(int i=1;i<=k;i++) { scanf("%d%d",&a,&b); V[a].PB(MP(i,b)); V[b].PB(MP(i,a)); } for(int i=1;i<=m;i++) { scal(kroki[i].f,kroki[i].s); } printf("%lld\n",wynik); } main() { solve(); }
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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | #include<cstdio> #include<iostream> #include<algorithm> #include<vector> #include<tr1/unordered_map> #include<queue> #include<cstdlib> #include<list> #include<set> #include<map> #include<cmath> #define MP make_pair #define PB push_back #define s second #define f first #define PII pair<int,int> #define VPII vector <pair<int,int> > #define VI vector <int> #define abs(a) max((a),-(a)) #define LL long long #define LD long double #define ALL(x) x.begin(),x.end() #define PU putchar_unlocked #define GU getchar_unlocked #define DBG(x) cerr<<#x<<" "<<(x)<<endl; using namespace std; int a,b,c,d,e,f,n,m,mx,l,r,k; LL wynik; char ch; int INF=1e9+1; const int N=2e5+4; int t[N]; VI fio[N]; int ile[N]; PII kroki[N]; VPII V[N]; void scal(int a,int b) { a=t[a]; b=t[b]; vector< pair<int, PII > > X; PII y; if(fio[a].size()>fio[b].size())swap(a,b); for(int i=0;i<fio[a].size();i++) { for(int j=0;j<V[fio[a][i]].size();j++) { y=V[fio[a][i]][j]; if(t[y.s]==b)X.PB(MP(y.f,MP(fio[a][i],y.s))); } } for(int i=0;i<fio[a].size();i++) { t[fio[a][i]]=b; fio[b].PB(fio[a][i]); } fio[a].clear(); //fio[a].swap(VI()); sort(ALL(X)); for(int i=0;i<X.size();i++) { a=X[i].s.f; b=X[i].s.s; c=min(ile[b],ile[a]); ile[a]-=c; ile[b]-=c; // printf("%d %d %d\n",a,b,c); wynik+=c*2; } } void solve() { scanf("%d%d%d",&n,&m,&k); for(int i=1;i<=n;i++) { t[i]=i; fio[i].PB(i); scanf("%d",&ile[i]); } for(int i=1;i<=m;i++) { scanf("%d%d",&kroki[i].f,&kroki[i].s); } for(int i=1;i<=k;i++) { scanf("%d%d",&a,&b); V[a].PB(MP(i,b)); V[b].PB(MP(i,a)); } for(int i=1;i<=m;i++) { scal(kroki[i].f,kroki[i].s); } printf("%lld\n",wynik); } main() { solve(); } |