#include <iostream> #include <vector> #include <set> using namespace std; typedef unsigned long long int big; struct krok { big z; big d; }; struct reakcja { big prio; big a; }; struct akcja { big prio; big a; big b; }; struct comp { bool operator()(const reakcja &l, const reakcja &g) { return l.a<g.a; } }; struct acomp { bool operator()(const akcja &l, const akcja &g) const { return l.prio<g.prio; } }; int main(){ big i,n,m,k,c,d,osad; krok kr; reakcja r; akcja a; set<big> s; set<reakcja,comp> ra; set<big>::iterator itz,itd; set<reakcja,comp>::iterator it; set<akcja,acomp>::iterator ita; cin >> n >> m >> k; big masa[n+1]; // 0: masa osadu, 1..n: masy substancji vector<set<big> > stan; // stan fiolek vector<krok> kroki; vector<set<reakcja,comp> > reakcje; set<akcja,acomp> akcje; // reakcje w bierzacym kroku // masa masa[0]=0; for (i=1; i<=n; i++) { cin >> masa[i]; } // stan s.clear(); stan.push_back(s); for (i=1; i<=n; i++) { s.clear(); s.insert(i); stan.push_back(s); } // kroki for (i=0; i<m; i++) { cin >> kr.z >> kr.d; kroki.push_back(kr); } // reakcje ra.clear(); for (i=0; i<=n; i++) { reakcje.push_back(ra); } for (i=0; i<k; i++) { cin >> c >> d; r.prio=i; r.a=d; reakcje[c].insert(r); r.a=c; reakcje[d].insert(r); } for (i=0; i<m; i++) { akcje.clear(); for (itz=stan[kroki[i].z].begin(); itz!=stan[kroki[i].z].end(); ++itz) { for (itd=stan[kroki[i].d].begin(); itd!=stan[kroki[i].d].end(); ++itd) { r.prio=0; r.a=*itd; it=reakcje[*itz].find(r); if (it!=reakcje[*itz].end()) { a.prio=it->prio; a.a=*itz; a.b=*itd; akcje.insert(a); } } } for (ita=akcje.begin(); ita!=akcje.end(); ++ita) { osad =(masa[ita->a]<=masa[ita->b])?masa[ita->a]:masa[ita->b]; masa[ita->a]-=osad; masa[ita->b]-=osad; masa[0]+=2*osad; if (masa[ita->a]==0) { stan[kroki[i].z].erase(ita->a); } if (masa[ita->b]==0) { stan[kroki[i].d].erase(ita->b); } } for (itz=stan[kroki[i].z].begin(); itz!=stan[kroki[i].z].end(); ++itz) { stan[kroki[i].d].insert(*itz); } stan[kroki[i].z].clear(); } cout << masa[0] << 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 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 118 119 120 121 122 | #include <iostream> #include <vector> #include <set> using namespace std; typedef unsigned long long int big; struct krok { big z; big d; }; struct reakcja { big prio; big a; }; struct akcja { big prio; big a; big b; }; struct comp { bool operator()(const reakcja &l, const reakcja &g) { return l.a<g.a; } }; struct acomp { bool operator()(const akcja &l, const akcja &g) const { return l.prio<g.prio; } }; int main(){ big i,n,m,k,c,d,osad; krok kr; reakcja r; akcja a; set<big> s; set<reakcja,comp> ra; set<big>::iterator itz,itd; set<reakcja,comp>::iterator it; set<akcja,acomp>::iterator ita; cin >> n >> m >> k; big masa[n+1]; // 0: masa osadu, 1..n: masy substancji vector<set<big> > stan; // stan fiolek vector<krok> kroki; vector<set<reakcja,comp> > reakcje; set<akcja,acomp> akcje; // reakcje w bierzacym kroku // masa masa[0]=0; for (i=1; i<=n; i++) { cin >> masa[i]; } // stan s.clear(); stan.push_back(s); for (i=1; i<=n; i++) { s.clear(); s.insert(i); stan.push_back(s); } // kroki for (i=0; i<m; i++) { cin >> kr.z >> kr.d; kroki.push_back(kr); } // reakcje ra.clear(); for (i=0; i<=n; i++) { reakcje.push_back(ra); } for (i=0; i<k; i++) { cin >> c >> d; r.prio=i; r.a=d; reakcje[c].insert(r); r.a=c; reakcje[d].insert(r); } for (i=0; i<m; i++) { akcje.clear(); for (itz=stan[kroki[i].z].begin(); itz!=stan[kroki[i].z].end(); ++itz) { for (itd=stan[kroki[i].d].begin(); itd!=stan[kroki[i].d].end(); ++itd) { r.prio=0; r.a=*itd; it=reakcje[*itz].find(r); if (it!=reakcje[*itz].end()) { a.prio=it->prio; a.a=*itz; a.b=*itd; akcje.insert(a); } } } for (ita=akcje.begin(); ita!=akcje.end(); ++ita) { osad =(masa[ita->a]<=masa[ita->b])?masa[ita->a]:masa[ita->b]; masa[ita->a]-=osad; masa[ita->b]-=osad; masa[0]+=2*osad; if (masa[ita->a]==0) { stan[kroki[i].z].erase(ita->a); } if (masa[ita->b]==0) { stan[kroki[i].d].erase(ita->b); } } for (itz=stan[kroki[i].z].begin(); itz!=stan[kroki[i].z].end(); ++itz) { stan[kroki[i].d].insert(*itz); } stan[kroki[i].z].clear(); } cout << masa[0] << endl; return 0; } |