#include <cstdio> #include <vector> #include <utility> #include <iostream> #include <cmath> #define PB push_back #define MP make_pair #define ST first #define ND second using namespace std; int ilosc, kroki, rek, a, b, fiolki, pom, licz, duba = 1; long long res, mn; long long Stan[300000]; int Odl[25][300000]; int Nr[300000]; int Size[300000]; int Pod[3000000]; vector <pair<int,int> > Reakcje[300000]; vector <int> Graph[300000]; void DFS(int); void PONUMERUJ(int); int LCA(int,int); bool POTOMEK(int,int); int main() { scanf("%d %d %d", &fiolki, &kroki, &rek); pom = ceil(log2(fiolki)); for(int i = 1; i <= fiolki; i++) { scanf("%lld", &Stan[i]); Odl[0][i] = i; } for(int i = 1; i <= kroki; i++) { scanf("%d %d", &a, &b); Odl[0][a] = b; Graph[b].PB(a); } for(int i = 1; i <= pom; i++) for(int j = 1; j <= fiolki; j++) Odl[i][j] = Odl[i-1][Odl[i-1][j]]; for(int i = 1; i <= fiolki; i++) if(Odl[0][i] == i) { PONUMERUJ(i); duba++; } for(int i = 1; i <= rek; i++) { scanf("%d %d", &a, &b); if(Pod[a] == Pod[b]) Reakcje[LCA(a,b)].PB(MP(a,b)); } for(int i = 1; i <= fiolki; i++) if(Odl[0][i] == i) DFS(i); printf("%lld\n", res); } void DFS(int v) { for(int i = 0; i < Graph[v].size(); i++) DFS(Graph[v][i]); for(int i = 0; i < Reakcje[v].size(); i++) { a = Reakcje[v][i].ST; b = Reakcje[v][i].ND; mn = min(Stan[a],Stan[b]); res += 2*mn; Stan[a] -= mn; Stan[b] -= mn; } } bool POTOMEK(int u, int v) { if((Nr[u] >= Nr[v] && Nr[u] < Nr[v] + Size[v])) return true; else return false; } int LCA(int u, int v) { if(POTOMEK(u,v)) return v; if(POTOMEK(v,u)) return u; int ii = u; int jj = pom; while(jj >= 0) { if(POTOMEK(v,Odl[jj][ii])) jj--; else ii = Odl[jj][ii]; } return Odl[0][ii]; } void PONUMERUJ(int v) { Nr[v] = ++licz; Pod[v] = duba; for(int i = 0; i < Graph[v].size(); i++) { PONUMERUJ(Graph[v][i]); Size[v] += Size[Graph[v][i]]; } Size[v]++; }
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 105 106 107 108 109 110 111 112 113 114 115 116 117 | #include <cstdio> #include <vector> #include <utility> #include <iostream> #include <cmath> #define PB push_back #define MP make_pair #define ST first #define ND second using namespace std; int ilosc, kroki, rek, a, b, fiolki, pom, licz, duba = 1; long long res, mn; long long Stan[300000]; int Odl[25][300000]; int Nr[300000]; int Size[300000]; int Pod[3000000]; vector <pair<int,int> > Reakcje[300000]; vector <int> Graph[300000]; void DFS(int); void PONUMERUJ(int); int LCA(int,int); bool POTOMEK(int,int); int main() { scanf("%d %d %d", &fiolki, &kroki, &rek); pom = ceil(log2(fiolki)); for(int i = 1; i <= fiolki; i++) { scanf("%lld", &Stan[i]); Odl[0][i] = i; } for(int i = 1; i <= kroki; i++) { scanf("%d %d", &a, &b); Odl[0][a] = b; Graph[b].PB(a); } for(int i = 1; i <= pom; i++) for(int j = 1; j <= fiolki; j++) Odl[i][j] = Odl[i-1][Odl[i-1][j]]; for(int i = 1; i <= fiolki; i++) if(Odl[0][i] == i) { PONUMERUJ(i); duba++; } for(int i = 1; i <= rek; i++) { scanf("%d %d", &a, &b); if(Pod[a] == Pod[b]) Reakcje[LCA(a,b)].PB(MP(a,b)); } for(int i = 1; i <= fiolki; i++) if(Odl[0][i] == i) DFS(i); printf("%lld\n", res); } void DFS(int v) { for(int i = 0; i < Graph[v].size(); i++) DFS(Graph[v][i]); for(int i = 0; i < Reakcje[v].size(); i++) { a = Reakcje[v][i].ST; b = Reakcje[v][i].ND; mn = min(Stan[a],Stan[b]); res += 2*mn; Stan[a] -= mn; Stan[b] -= mn; } } bool POTOMEK(int u, int v) { if((Nr[u] >= Nr[v] && Nr[u] < Nr[v] + Size[v])) return true; else return false; } int LCA(int u, int v) { if(POTOMEK(u,v)) return v; if(POTOMEK(v,u)) return u; int ii = u; int jj = pom; while(jj >= 0) { if(POTOMEK(v,Odl[jj][ii])) jj--; else ii = Odl[jj][ii]; } return Odl[0][ii]; } void PONUMERUJ(int v) { Nr[v] = ++licz; Pod[v] = duba; for(int i = 0; i < Graph[v].size(); i++) { PONUMERUJ(Graph[v][i]); Size[v] += Size[Graph[v][i]]; } Size[v]++; } |