#include <iostream> #include <sstream> #include <fstream> #include <string> #include <vector> #include <queue> #include <algorithm> #include <map> #include <set> using namespace std; typedef long long LL; #define MAX(a, b) ((a) > (b) ? (a) : (b)) #define MIN(a, b) ((a) < (b) ? (a) : (b)) #define ABS(val) ((val) < 0 ? -(val) : (val)) #define FOR(x, b, e) for(int x = b; x <= (e); ++x) #define FORD(x, b, e) for(int x = b; x >= (e); --x) #define REP(x, n) for(int x = 0; x < (n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define ST first #define ND second /* Brute force! Please do not expect much :)*/ set<int> regs[200200]; int quant[200200]; int main() { int n, m, k; scanf("%d %d %d", &n, &m, &k); REP(i, n) { scanf("%d", &(quant[i+1])); regs[i+1].insert(i+1); } vector<pair<int, int> > pours; REP(i, m) { int from, to; scanf("%d %d", &from, &to); pours.PB(make_pair(from, to)); } vector<pair<int, int> > reactions; REP(i, k) { int r1, r2; scanf("%d %d", &r1, &r2); reactions.PB(make_pair(r1, r2)); } LL sum = 0LL; FOREACH(p, pours) { int from = p->first; int to = p->second; /* printf("pour: %d %d\n", from, to); printf("sizes: %d %d\n", regs[from].size(), regs[to].size()); */ FOREACH(r, reactions) { int r1 = r->first; int r2 = r->second; bool f1in1 = regs[from].find(r1) != regs[from].end(); bool f2in1 = regs[from].find(r2) != regs[from].end(); bool f1in2 = regs[to].find(r1) != regs[to].end(); bool f2in2 = regs[to].find(r2) != regs[to].end(); //printf("%d %d %d %d\n", f1in1, f2in1, f1in2, f2in2); if((f1in1 && f2in2) || (f2in1 && f1in2)) { // printf("AAL0, %d %d; %d %d\n", r1, r2, quant[r1], quant[r2]); int sub = MIN(quant[r1], quant[r2]); sum += (LL)(2 * sub); quant[r1] -= sub; quant[r2] -= sub; } } regs[to].insert(ALL(regs[from])); } printf("%lld\n", sum); 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 95 96 97 | #include <iostream> #include <sstream> #include <fstream> #include <string> #include <vector> #include <queue> #include <algorithm> #include <map> #include <set> using namespace std; typedef long long LL; #define MAX(a, b) ((a) > (b) ? (a) : (b)) #define MIN(a, b) ((a) < (b) ? (a) : (b)) #define ABS(val) ((val) < 0 ? -(val) : (val)) #define FOR(x, b, e) for(int x = b; x <= (e); ++x) #define FORD(x, b, e) for(int x = b; x >= (e); --x) #define REP(x, n) for(int x = 0; x < (n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define ST first #define ND second /* Brute force! Please do not expect much :)*/ set<int> regs[200200]; int quant[200200]; int main() { int n, m, k; scanf("%d %d %d", &n, &m, &k); REP(i, n) { scanf("%d", &(quant[i+1])); regs[i+1].insert(i+1); } vector<pair<int, int> > pours; REP(i, m) { int from, to; scanf("%d %d", &from, &to); pours.PB(make_pair(from, to)); } vector<pair<int, int> > reactions; REP(i, k) { int r1, r2; scanf("%d %d", &r1, &r2); reactions.PB(make_pair(r1, r2)); } LL sum = 0LL; FOREACH(p, pours) { int from = p->first; int to = p->second; /* printf("pour: %d %d\n", from, to); printf("sizes: %d %d\n", regs[from].size(), regs[to].size()); */ FOREACH(r, reactions) { int r1 = r->first; int r2 = r->second; bool f1in1 = regs[from].find(r1) != regs[from].end(); bool f2in1 = regs[from].find(r2) != regs[from].end(); bool f1in2 = regs[to].find(r1) != regs[to].end(); bool f2in2 = regs[to].find(r2) != regs[to].end(); //printf("%d %d %d %d\n", f1in1, f2in1, f1in2, f2in2); if((f1in1 && f2in2) || (f2in1 && f1in2)) { // printf("AAL0, %d %d; %d %d\n", r1, r2, quant[r1], quant[r2]); int sub = MIN(quant[r1], quant[r2]); sum += (LL)(2 * sub); quant[r1] -= sub; quant[r2] -= sub; } } regs[to].insert(ALL(regs[from])); } printf("%lld\n", sum); return 0; } |