#include <iostream>
#include <algorithm>
#include <set>
#include <list>
#define MAXN 200002
#define MAXK 500002
using namespace std;
struct Pair {
int x;
int y;
};
Pair Sb[MAXN];
Pair St[MAXN];
Pair R[MAXK];
set<int> CR[MAXN];
list<int> FS[MAXN];
typedef set<int>::iterator Sit;
typedef list<int>::iterator Lit;
int m,n,k;
int main()
{
ios_base::sync_with_stdio(0);
cin >> n >> m >> k;
long long g = 0;
int r;
for(int i=1; i<=n; ++i)
{
cin >> Sb[i].x;
Sb[i].y = i;
FS[i].push_back(i);
}
for(int i=0; i<m; ++i)
{
cin >> St[i].x >> St[i].y;
}
for(int i=0; i<k; ++i)
{
cin >> R[i].x >> R[i].y;
CR[R[i].x].insert(i);
}
for(int i=0; i<m; ++i)
{
CR[St[i].y].insert(CR[St[i].x].begin(), CR[St[i].x].end());
for(Lit it=FS[St[i].x].begin(); it!=FS[St[i].x].end(); ++it)
{
if(Sb[*it].x>0)
{
Sb[*it].y = St[i].y;
FS[St[i].y].push_back(*it);
}
}
for(Sit it=CR[St[i].y].begin(); it!=CR[St[i].y].end();)
{
if (Sb[R[*it].x].y==St[i].y && Sb[R[*it].y].y==St[i].y)
{
r = min(Sb[R[*it].x].x, Sb[R[*it].y].x);
g += 2*r;
Sb[R[*it].x].x -= r;
Sb[R[*it].y].x -= r;
it = CR[St[i].y].erase(it);
}
else if(Sb[R[*it].x].x==0 || Sb[R[*it].y].x==0)
{
it = CR[St[i].y].erase(it);
}
else
{
++it;
}
}
}
cout << g << endl;
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 | #include <iostream> #include <algorithm> #include <set> #include <list> #define MAXN 200002 #define MAXK 500002 using namespace std; struct Pair { int x; int y; }; Pair Sb[MAXN]; Pair St[MAXN]; Pair R[MAXK]; set<int> CR[MAXN]; list<int> FS[MAXN]; typedef set<int>::iterator Sit; typedef list<int>::iterator Lit; int m,n,k; int main() { ios_base::sync_with_stdio(0); cin >> n >> m >> k; long long g = 0; int r; for(int i=1; i<=n; ++i) { cin >> Sb[i].x; Sb[i].y = i; FS[i].push_back(i); } for(int i=0; i<m; ++i) { cin >> St[i].x >> St[i].y; } for(int i=0; i<k; ++i) { cin >> R[i].x >> R[i].y; CR[R[i].x].insert(i); } for(int i=0; i<m; ++i) { CR[St[i].y].insert(CR[St[i].x].begin(), CR[St[i].x].end()); for(Lit it=FS[St[i].x].begin(); it!=FS[St[i].x].end(); ++it) { if(Sb[*it].x>0) { Sb[*it].y = St[i].y; FS[St[i].y].push_back(*it); } } for(Sit it=CR[St[i].y].begin(); it!=CR[St[i].y].end();) { if (Sb[R[*it].x].y==St[i].y && Sb[R[*it].y].y==St[i].y) { r = min(Sb[R[*it].x].x, Sb[R[*it].y].x); g += 2*r; Sb[R[*it].x].x -= r; Sb[R[*it].y].x -= r; it = CR[St[i].y].erase(it); } else if(Sb[R[*it].x].x==0 || Sb[R[*it].y].x==0) { it = CR[St[i].y].erase(it); } else { ++it; } } } cout << g << endl; return 0; } |
English