#include <cstdio> #include <map> #include <algorithm> #include <vector> #include <queue> #include <list> #include <string> #include <unordered_set> using namespace std; typedef long long LL; typedef vector<int> VI; typedef pair<int,int> PI; #define MP make_pair #define PB push_back #define SD second #define FT first int k,n,m,q; const int N = 100003; const int INF = 20000000; int masa[N]; PI steps[N]; vector<PI> hate; vector<PI> sas[N]; VI zawartosc[N]; LL przelej(int a, int b) { //O(E(a)*b+E(b)*a) unordered_set<int> bb(zawartosc[b].begin(),zawartosc[b].end()); vector<pair<int,PI> > wytracaj; for(int i=0;i<zawartosc[a].size();i++) { int sub = zawartosc[a][i]; for(int i2=0;i2<sas[sub].size();i2++) { PI hater = sas[sub][i2]; if(bb.count(hater.SD)) { wytracaj.PB(MP(hater.FT,MP(sub,hater.SD))); } } } sort(wytracaj.begin(),wytracaj.end()); LL res = 0; for(int i=0;i<wytracaj.size();i++) { PI miedzy = wytracaj[i].SD; LL ile = min(masa[miedzy.FT],masa[miedzy.SD]); res +=ile * 2; masa[miedzy.FT]-=ile; masa[miedzy.SD]-=ile; } for(int i=0;i<zawartosc[a].size();i++) { int sub = zawartosc[a][i]; if(masa[sub] > 0) { zawartosc[b].PB(sub); } } return res; } int main() { scanf("%d%d%d",&n,&m,&k); for(int i=0;i<n;i++) { scanf("%d",&masa[i]); zawartosc[i].PB(i); } for(int i=0;i<m;i++) { int a,b; scanf("%d%d",&a,&b);a--;b--; steps[i] = MP(a,b); } for(int i=0;i<k;i++) { int a,b; scanf("%d%d",&a,&b);a--;b--; sas[a].PB(MP(i,b)); sas[b].PB(MP(i,a)); hate.PB(MP(a,b)); } LL res = 0; for(int i=0;i<m;i++) { res += przelej(steps[i].FT,steps[i].SD); //printf("%lld\n",res); } printf("%lld\n",res); return 0; }
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 | #include <cstdio> #include <map> #include <algorithm> #include <vector> #include <queue> #include <list> #include <string> #include <unordered_set> using namespace std; typedef long long LL; typedef vector<int> VI; typedef pair<int,int> PI; #define MP make_pair #define PB push_back #define SD second #define FT first int k,n,m,q; const int N = 100003; const int INF = 20000000; int masa[N]; PI steps[N]; vector<PI> hate; vector<PI> sas[N]; VI zawartosc[N]; LL przelej(int a, int b) { //O(E(a)*b+E(b)*a) unordered_set<int> bb(zawartosc[b].begin(),zawartosc[b].end()); vector<pair<int,PI> > wytracaj; for(int i=0;i<zawartosc[a].size();i++) { int sub = zawartosc[a][i]; for(int i2=0;i2<sas[sub].size();i2++) { PI hater = sas[sub][i2]; if(bb.count(hater.SD)) { wytracaj.PB(MP(hater.FT,MP(sub,hater.SD))); } } } sort(wytracaj.begin(),wytracaj.end()); LL res = 0; for(int i=0;i<wytracaj.size();i++) { PI miedzy = wytracaj[i].SD; LL ile = min(masa[miedzy.FT],masa[miedzy.SD]); res +=ile * 2; masa[miedzy.FT]-=ile; masa[miedzy.SD]-=ile; } for(int i=0;i<zawartosc[a].size();i++) { int sub = zawartosc[a][i]; if(masa[sub] > 0) { zawartosc[b].PB(sub); } } return res; } int main() { scanf("%d%d%d",&n,&m,&k); for(int i=0;i<n;i++) { scanf("%d",&masa[i]); zawartosc[i].PB(i); } for(int i=0;i<m;i++) { int a,b; scanf("%d%d",&a,&b);a--;b--; steps[i] = MP(a,b); } for(int i=0;i<k;i++) { int a,b; scanf("%d%d",&a,&b);a--;b--; sas[a].PB(MP(i,b)); sas[b].PB(MP(i,a)); hate.PB(MP(a,b)); } LL res = 0; for(int i=0;i<m;i++) { res += przelej(steps[i].FT,steps[i].SD); //printf("%lld\n",res); } printf("%lld\n",res); return 0; } |