#include<cstdio> #include<algorithm> #include<vector> #include<set> #include<queue> #include<stack> #include<cmath> #include<map> using namespace std; typedef pair<int,int> PI; typedef long long LL; typedef double D; #define FI first #define SE second #define MP make_pair #define PB push_back #define R(I,N) for(int I=0;I<N;I++) #define F(I,A,B) for(int I=A;I<B;I++) #define FD(I,N) for(int I=N-1;I>=0;I--) #define VAR(v,i) __typeof(i) v=(i) #define FOREACH(i,c) for(VAR(i,(c).begin());i!=(c).end();++i) #define make(A) scanf("%d",&A) #define MAXn 200100 #define MAXk 500100 int a[MAXk],b[MAXk]; vector<int> re[MAXn]; int n,m,k,g[MAXn],ak[MAXn]; int f[MAXn]; LL wyn; //algorytm Tarjana, z biblioteczki Marka Cygana const int N=501000; vector<int> kraw[N],pytania[N]; int kolor[N],fu[N],przodek[N]; int fufind(int x){ return fu[x]<0?x:fu[x]=fufind(fu[x]); } //wywolujemy DfsTarjan(wierzcholek,-1); //zapytania umieszczamy w zmiennej pytania; //zakladam, ze pytamy o 2 rozne wierzcholki void DfsTarjan(int x,int parent){ kolor[x]=1; fu[x]=-1; przodek[x]=x; FOREACH(it,kraw[x]) if (*it!=parent){ DfsTarjan(*it,x); int a=fufind(x),b=fufind(*it); if (fu[a]<fu[b]) swap(a,b); fu[b]+=fu[a]; fu[a]=b; przodek[b]=x; } FOREACH(it,pytania[x]){ int ak = a[*it] + b[*it] - x; if (kolor[ak]==2) re[przodek[fufind(ak)]-n].PB(*it); } kolor[x]=2; } inline void dodaj_krawedz(int a,int b){ kraw[a].PB(b); kraw[b].PB(a); } inline void dodaj_pytanie(int a,int b,int i){ pytania[a].PB(i); pytania[b].PB(i); } //----------------------------- int find(int a){ return f[a]==a?a:f[a]=find(f[a]); } void uni(int a,int b){ f[find(a)]=find(b); } int it; inline void wlej(int a,int b){ a = find(a); b = find(b); dodaj_krawedz(ak[a],it); dodaj_krawedz(ak[b],it); uni(a,b); ak[b] = it; ak[a] = it; it++; } main(){ make(n);make(m);make(k); R(i,n){ make(g[i]); ak[i] = i; f[i] = i; } it = n; R(i,m){ int a,b;make(a);make(b);a--;b--; wlej(a,b); } R(i,n)if(find(1) != find(i))wlej(1,i); R(i,k){ make(a[i]); make(b[i]); a[i]--; b[i]--; dodaj_pytanie(a[i],b[i],i); } DfsTarjan(2*n-2,-1); R(i,m){ sort(re[i].begin(),re[i].end()); R(j,re[i].size()){ int pom = re[i][j]; int ile = min(g[a[pom]],g[b[pom]]); wyn += 2*ile; g[a[pom]] -= ile; g[b[pom]] -= ile; } } printf("%lld\n",wyn); }
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<algorithm> #include<vector> #include<set> #include<queue> #include<stack> #include<cmath> #include<map> using namespace std; typedef pair<int,int> PI; typedef long long LL; typedef double D; #define FI first #define SE second #define MP make_pair #define PB push_back #define R(I,N) for(int I=0;I<N;I++) #define F(I,A,B) for(int I=A;I<B;I++) #define FD(I,N) for(int I=N-1;I>=0;I--) #define VAR(v,i) __typeof(i) v=(i) #define FOREACH(i,c) for(VAR(i,(c).begin());i!=(c).end();++i) #define make(A) scanf("%d",&A) #define MAXn 200100 #define MAXk 500100 int a[MAXk],b[MAXk]; vector<int> re[MAXn]; int n,m,k,g[MAXn],ak[MAXn]; int f[MAXn]; LL wyn; //algorytm Tarjana, z biblioteczki Marka Cygana const int N=501000; vector<int> kraw[N],pytania[N]; int kolor[N],fu[N],przodek[N]; int fufind(int x){ return fu[x]<0?x:fu[x]=fufind(fu[x]); } //wywolujemy DfsTarjan(wierzcholek,-1); //zapytania umieszczamy w zmiennej pytania; //zakladam, ze pytamy o 2 rozne wierzcholki void DfsTarjan(int x,int parent){ kolor[x]=1; fu[x]=-1; przodek[x]=x; FOREACH(it,kraw[x]) if (*it!=parent){ DfsTarjan(*it,x); int a=fufind(x),b=fufind(*it); if (fu[a]<fu[b]) swap(a,b); fu[b]+=fu[a]; fu[a]=b; przodek[b]=x; } FOREACH(it,pytania[x]){ int ak = a[*it] + b[*it] - x; if (kolor[ak]==2) re[przodek[fufind(ak)]-n].PB(*it); } kolor[x]=2; } inline void dodaj_krawedz(int a,int b){ kraw[a].PB(b); kraw[b].PB(a); } inline void dodaj_pytanie(int a,int b,int i){ pytania[a].PB(i); pytania[b].PB(i); } //----------------------------- int find(int a){ return f[a]==a?a:f[a]=find(f[a]); } void uni(int a,int b){ f[find(a)]=find(b); } int it; inline void wlej(int a,int b){ a = find(a); b = find(b); dodaj_krawedz(ak[a],it); dodaj_krawedz(ak[b],it); uni(a,b); ak[b] = it; ak[a] = it; it++; } main(){ make(n);make(m);make(k); R(i,n){ make(g[i]); ak[i] = i; f[i] = i; } it = n; R(i,m){ int a,b;make(a);make(b);a--;b--; wlej(a,b); } R(i,n)if(find(1) != find(i))wlej(1,i); R(i,k){ make(a[i]); make(b[i]); a[i]--; b[i]--; dodaj_pytanie(a[i],b[i],i); } DfsTarjan(2*n-2,-1); R(i,m){ sort(re[i].begin(),re[i].end()); R(j,re[i].size()){ int pom = re[i][j]; int ile = min(g[a[pom]],g[b[pom]]); wyn += 2*ile; g[a[pom]] -= ile; g[b[pom]] -= ile; } } printf("%lld\n",wyn); } |