#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]++; } |
English