#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<tr1/unordered_map>
#include<queue>
#include<cstdlib>
#include<list>
#include<set>
#include<map>
#include<cmath>
#define MP make_pair
#define PB push_back
#define s second
#define f first
#define PII pair<int,int>
#define VPII vector <pair<int,int> >
#define VI vector <int>
#define abs(a) max((a),-(a))
#define LL long long
#define LD long double
#define ALL(x) x.begin(),x.end()
#define PU putchar_unlocked
#define GU getchar_unlocked
#define DBG(x) cerr<<#x<<" "<<(x)<<endl;
using namespace std;
int a,b,c,d,e,f,n,m,mx,l,r,k;
LL wynik;
char ch;
int INF=1e9+1;
const int N=2e5+4;
int t[N];
VI fio[N];
int ile[N];
PII kroki[N];
VPII V[N];
void scal(int a,int b)
{
a=t[a];
b=t[b];
vector< pair<int, PII > > X;
PII y;
if(fio[a].size()>fio[b].size())swap(a,b);
for(int i=0;i<fio[a].size();i++)
{
for(int j=0;j<V[fio[a][i]].size();j++)
{
y=V[fio[a][i]][j];
if(t[y.s]==b)X.PB(MP(y.f,MP(fio[a][i],y.s)));
}
}
for(int i=0;i<fio[a].size();i++)
{
t[fio[a][i]]=b;
fio[b].PB(fio[a][i]);
}
fio[a].clear();
//fio[a].swap(VI());
sort(ALL(X));
for(int i=0;i<X.size();i++)
{
a=X[i].s.f;
b=X[i].s.s;
c=min(ile[b],ile[a]);
ile[a]-=c;
ile[b]-=c;
// printf("%d %d %d\n",a,b,c);
wynik+=c*2;
}
}
void solve()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
{
t[i]=i;
fio[i].PB(i);
scanf("%d",&ile[i]);
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&kroki[i].f,&kroki[i].s);
}
for(int i=1;i<=k;i++)
{
scanf("%d%d",&a,&b);
V[a].PB(MP(i,b));
V[b].PB(MP(i,a));
}
for(int i=1;i<=m;i++)
{
scal(kroki[i].f,kroki[i].s);
}
printf("%lld\n",wynik);
}
main()
{
solve();
}
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 | #include<cstdio> #include<iostream> #include<algorithm> #include<vector> #include<tr1/unordered_map> #include<queue> #include<cstdlib> #include<list> #include<set> #include<map> #include<cmath> #define MP make_pair #define PB push_back #define s second #define f first #define PII pair<int,int> #define VPII vector <pair<int,int> > #define VI vector <int> #define abs(a) max((a),-(a)) #define LL long long #define LD long double #define ALL(x) x.begin(),x.end() #define PU putchar_unlocked #define GU getchar_unlocked #define DBG(x) cerr<<#x<<" "<<(x)<<endl; using namespace std; int a,b,c,d,e,f,n,m,mx,l,r,k; LL wynik; char ch; int INF=1e9+1; const int N=2e5+4; int t[N]; VI fio[N]; int ile[N]; PII kroki[N]; VPII V[N]; void scal(int a,int b) { a=t[a]; b=t[b]; vector< pair<int, PII > > X; PII y; if(fio[a].size()>fio[b].size())swap(a,b); for(int i=0;i<fio[a].size();i++) { for(int j=0;j<V[fio[a][i]].size();j++) { y=V[fio[a][i]][j]; if(t[y.s]==b)X.PB(MP(y.f,MP(fio[a][i],y.s))); } } for(int i=0;i<fio[a].size();i++) { t[fio[a][i]]=b; fio[b].PB(fio[a][i]); } fio[a].clear(); //fio[a].swap(VI()); sort(ALL(X)); for(int i=0;i<X.size();i++) { a=X[i].s.f; b=X[i].s.s; c=min(ile[b],ile[a]); ile[a]-=c; ile[b]-=c; // printf("%d %d %d\n",a,b,c); wynik+=c*2; } } void solve() { scanf("%d%d%d",&n,&m,&k); for(int i=1;i<=n;i++) { t[i]=i; fio[i].PB(i); scanf("%d",&ile[i]); } for(int i=1;i<=m;i++) { scanf("%d%d",&kroki[i].f,&kroki[i].s); } for(int i=1;i<=k;i++) { scanf("%d%d",&a,&b); V[a].PB(MP(i,b)); V[b].PB(MP(i,a)); } for(int i=1;i<=m;i++) { scal(kroki[i].f,kroki[i].s); } printf("%lld\n",wynik); } main() { solve(); } |
English