#include <iostream> #include <algorithm> #include <map> #include <vector> #include <set> using namespace std; bool rek1(int lewy, int prawy, map<int, set<int> > &od_p, map<int, set<int> > &od_l){ //cout << lewy << ' ' << prawy << endl; if(od_p.find(prawy) != od_p.end()){ if(od_p[prawy].find(lewy) != od_p[prawy].end()) return false; } set<int>::iterator it; /*if(od_p.find(prawy) != od_p.end()) for(it=od_p[prawy].begin(); it!=od_p[prawy].end(); it++){ if(!rek1(*it, prawy, od_p, od_l)) return false; } if(od_l.find(lewy) != od_l.end()) for(it=od_l[lewy].begin(); it!=od_l[lewy].end(); it++){ if(!rek1(lewy, *it, od_p, od_l)) return false; }*/ if(od_p.find(lewy - 1) != od_p.end()){ for(it=od_p[lewy - 1].begin(); it!=od_p[lewy - 1].end(); it++){ if(*it != lewy - 1) if(!rek1(*it, prawy, od_p, od_l)) return false; } } if(od_l.find(prawy + 1) != od_l.end()) for(it=od_l[prawy + 1].begin(); it!=od_l[prawy + 1].end(); it++){ if(*it != prawy + 1) if(!rek1(lewy, *it, od_p, od_l)) return false; } return true; } bool rek2(int lewy, int prawy, map<int, set<int> > &od_p, map<int, set<int> > &od_l){ if(od_p.find(prawy) != od_p.end()) if(od_p[prawy].find(lewy) != od_p[prawy].end()) return false; if(od_l.find(prawy + 1) == od_l.end()) return true; set<int>::iterator it; for(it=od_l[prawy + 1].begin(); it!=od_l[prawy + 1].end(); it++){ if(*it != prawy + 1) if(!rek2(lewy, *it, od_p, od_l)) return false; } return true; } bool czy_biore(pair<int, int> para, map<int, set<int> > &od_p, map<int, set<int> > &od_l){ int lewy = para.first; int prawy = para.second; if(!rek1(lewy, prawy, od_p, od_l)){ return false; } //return true; return rek2(lewy, prawy, od_p, od_l); } int main(){ ios_base::sync_with_stdio(0); int n, tmp; cin >> n; map<long long int, vector<pair<int, int> > > m; for(int i=1;i<=n;i++){ for(int j=i;j<=n;j++){ cin >> tmp; pair<int, int> przedzial; przedzial.first = i; przedzial.second = j; m[tmp].push_back(przedzial); } } map<int, set<int> > od_p; map<int, set<int> > od_l; map<long long int, vector<pair<int, int> > >::iterator it; int wzialem = 0; long long int koszt=0; for(it = m.begin(); it != m.end(); it++){ //cout << it->first << ':' << endl; vector<pair<int, int> > v = it->second; vector<pair<int, int> >::iterator it2; for(it2=v.begin(); it2!=v.end(); it2++){ int lewy = it2->first; int prawy = it2->second; //cout << "sprawdzam: " << lewy << ' ' << prawy << endl; if(czy_biore(*it2, od_p, od_l)){ od_p[prawy].insert(lewy); od_l[lewy].insert(prawy); wzialem++; koszt += it->first; if(wzialem == n) goto end; } else{ //cout << "odrzucam: " << lewy << ' ' << prawy << ' ' << it->first << endl; } //cout << it2->first << ' ' << it2->second << endl; } } end: /*map<int, set<int> >::iterator it3; set<int>::iterator it4; cout << "Od lewego:" << endl; for(it3=od_l.begin(); it3!=od_l.end(); it3++){ cout << it3->first << ": "; for(it4=it3->second.begin(); it4!=it3->second.end(); it4++){ cout << *it4 << ' '; } cout << endl; } cout << endl; cout << "Od prawego:" << endl; //map<int, set<int> >::iterator it3; //set<int>::iterator it4; for(it3=od_p.begin(); it3!=od_p.end(); it3++){ cout << it3->first << ": "; for(it4=it3->second.begin(); it4!=it3->second.end(); it4++){ cout << *it4 << ' '; } cout << endl; } cout << endl;*/ cout << koszt << 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 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 | #include <iostream> #include <algorithm> #include <map> #include <vector> #include <set> using namespace std; bool rek1(int lewy, int prawy, map<int, set<int> > &od_p, map<int, set<int> > &od_l){ //cout << lewy << ' ' << prawy << endl; if(od_p.find(prawy) != od_p.end()){ if(od_p[prawy].find(lewy) != od_p[prawy].end()) return false; } set<int>::iterator it; /*if(od_p.find(prawy) != od_p.end()) for(it=od_p[prawy].begin(); it!=od_p[prawy].end(); it++){ if(!rek1(*it, prawy, od_p, od_l)) return false; } if(od_l.find(lewy) != od_l.end()) for(it=od_l[lewy].begin(); it!=od_l[lewy].end(); it++){ if(!rek1(lewy, *it, od_p, od_l)) return false; }*/ if(od_p.find(lewy - 1) != od_p.end()){ for(it=od_p[lewy - 1].begin(); it!=od_p[lewy - 1].end(); it++){ if(*it != lewy - 1) if(!rek1(*it, prawy, od_p, od_l)) return false; } } if(od_l.find(prawy + 1) != od_l.end()) for(it=od_l[prawy + 1].begin(); it!=od_l[prawy + 1].end(); it++){ if(*it != prawy + 1) if(!rek1(lewy, *it, od_p, od_l)) return false; } return true; } bool rek2(int lewy, int prawy, map<int, set<int> > &od_p, map<int, set<int> > &od_l){ if(od_p.find(prawy) != od_p.end()) if(od_p[prawy].find(lewy) != od_p[prawy].end()) return false; if(od_l.find(prawy + 1) == od_l.end()) return true; set<int>::iterator it; for(it=od_l[prawy + 1].begin(); it!=od_l[prawy + 1].end(); it++){ if(*it != prawy + 1) if(!rek2(lewy, *it, od_p, od_l)) return false; } return true; } bool czy_biore(pair<int, int> para, map<int, set<int> > &od_p, map<int, set<int> > &od_l){ int lewy = para.first; int prawy = para.second; if(!rek1(lewy, prawy, od_p, od_l)){ return false; } //return true; return rek2(lewy, prawy, od_p, od_l); } int main(){ ios_base::sync_with_stdio(0); int n, tmp; cin >> n; map<long long int, vector<pair<int, int> > > m; for(int i=1;i<=n;i++){ for(int j=i;j<=n;j++){ cin >> tmp; pair<int, int> przedzial; przedzial.first = i; przedzial.second = j; m[tmp].push_back(przedzial); } } map<int, set<int> > od_p; map<int, set<int> > od_l; map<long long int, vector<pair<int, int> > >::iterator it; int wzialem = 0; long long int koszt=0; for(it = m.begin(); it != m.end(); it++){ //cout << it->first << ':' << endl; vector<pair<int, int> > v = it->second; vector<pair<int, int> >::iterator it2; for(it2=v.begin(); it2!=v.end(); it2++){ int lewy = it2->first; int prawy = it2->second; //cout << "sprawdzam: " << lewy << ' ' << prawy << endl; if(czy_biore(*it2, od_p, od_l)){ od_p[prawy].insert(lewy); od_l[lewy].insert(prawy); wzialem++; koszt += it->first; if(wzialem == n) goto end; } else{ //cout << "odrzucam: " << lewy << ' ' << prawy << ' ' << it->first << endl; } //cout << it2->first << ' ' << it2->second << endl; } } end: /*map<int, set<int> >::iterator it3; set<int>::iterator it4; cout << "Od lewego:" << endl; for(it3=od_l.begin(); it3!=od_l.end(); it3++){ cout << it3->first << ": "; for(it4=it3->second.begin(); it4!=it3->second.end(); it4++){ cout << *it4 << ' '; } cout << endl; } cout << endl; cout << "Od prawego:" << endl; //map<int, set<int> >::iterator it3; //set<int>::iterator it4; for(it3=od_p.begin(); it3!=od_p.end(); it3++){ cout << it3->first << ": "; for(it4=it3->second.begin(); it4!=it3->second.end(); it4++){ cout << *it4 << ' '; } cout << endl; } cout << endl;*/ cout << koszt << endl; return 0; } |