#include <iostream> #include <algorithm> #include <vector> using namespace std; struct Przedzial { int poczatek, koniec; int koszt; }; bool operatorSortowaniaKosztow(const Przedzial &p1, const Przedzial &p2) { if (p1.koszt == p2.koszt) return p1.koniec-p1.poczatek < p2.koniec-p2.poczatek; else return p1.koszt < p2.koszt; } int main() { int n; scanf("%d", &n); vector<bool> sprawdzone(n+1, false); int liczbaSprawdzonych = 0; int liczbaPrzedzialow = n*(n+1)/2; int pierwszyPrzedzial = 0; long long int calkowityKoszt = 0; vector<Przedzial> przedzialy(liczbaPrzedzialow); vector<bool> nieuzytePrzedzialy(liczbaPrzedzialow, true); for (int i=1; i<=n; ++i) { for (int j=i; j<=n; ++j) { scanf("%d", &przedzialy[pierwszyPrzedzial].koszt); przedzialy[pierwszyPrzedzial].poczatek = i; przedzialy[pierwszyPrzedzial].koniec = j; pierwszyPrzedzial++; } } sort(przedzialy.begin(), przedzialy.end(), operatorSortowaniaKosztow); pierwszyPrzedzial = 0; int znalezionyPrzedzial = -1; int liczbaNiesprawdzonych = 0; int niesprawdzony = -1; while (liczbaSprawdzonych < n) { for (int i=pierwszyPrzedzial; i<liczbaPrzedzialow; ++i) { liczbaNiesprawdzonych = 0; niesprawdzony = -1; if (nieuzytePrzedzialy[i]) { for (int j=przedzialy[i].poczatek; j<=przedzialy[i].koniec; ++j) { if (!sprawdzone[j]) { if (liczbaNiesprawdzonych == 0) niesprawdzony = j; liczbaNiesprawdzonych++; } } if (liczbaNiesprawdzonych == 0) { nieuzytePrzedzialy[i] = false; if (i>0 && (pierwszyPrzedzial == i-1) && nieuzytePrzedzialy[i-1] == false) pierwszyPrzedzial++; } else if (liczbaNiesprawdzonych == 1) { znalezionyPrzedzial = i; calkowityKoszt += przedzialy[znalezionyPrzedzial].koszt; sprawdzone[niesprawdzony] = true; liczbaSprawdzonych++; break; } } else { if (i>0 && (pierwszyPrzedzial == i-1) && nieuzytePrzedzialy[i-1] == false) pierwszyPrzedzial++; } } } printf("%lld\n", calkowityKoszt); 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 | #include <iostream> #include <algorithm> #include <vector> using namespace std; struct Przedzial { int poczatek, koniec; int koszt; }; bool operatorSortowaniaKosztow(const Przedzial &p1, const Przedzial &p2) { if (p1.koszt == p2.koszt) return p1.koniec-p1.poczatek < p2.koniec-p2.poczatek; else return p1.koszt < p2.koszt; } int main() { int n; scanf("%d", &n); vector<bool> sprawdzone(n+1, false); int liczbaSprawdzonych = 0; int liczbaPrzedzialow = n*(n+1)/2; int pierwszyPrzedzial = 0; long long int calkowityKoszt = 0; vector<Przedzial> przedzialy(liczbaPrzedzialow); vector<bool> nieuzytePrzedzialy(liczbaPrzedzialow, true); for (int i=1; i<=n; ++i) { for (int j=i; j<=n; ++j) { scanf("%d", &przedzialy[pierwszyPrzedzial].koszt); przedzialy[pierwszyPrzedzial].poczatek = i; przedzialy[pierwszyPrzedzial].koniec = j; pierwszyPrzedzial++; } } sort(przedzialy.begin(), przedzialy.end(), operatorSortowaniaKosztow); pierwszyPrzedzial = 0; int znalezionyPrzedzial = -1; int liczbaNiesprawdzonych = 0; int niesprawdzony = -1; while (liczbaSprawdzonych < n) { for (int i=pierwszyPrzedzial; i<liczbaPrzedzialow; ++i) { liczbaNiesprawdzonych = 0; niesprawdzony = -1; if (nieuzytePrzedzialy[i]) { for (int j=przedzialy[i].poczatek; j<=przedzialy[i].koniec; ++j) { if (!sprawdzone[j]) { if (liczbaNiesprawdzonych == 0) niesprawdzony = j; liczbaNiesprawdzonych++; } } if (liczbaNiesprawdzonych == 0) { nieuzytePrzedzialy[i] = false; if (i>0 && (pierwszyPrzedzial == i-1) && nieuzytePrzedzialy[i-1] == false) pierwszyPrzedzial++; } else if (liczbaNiesprawdzonych == 1) { znalezionyPrzedzial = i; calkowityKoszt += przedzialy[znalezionyPrzedzial].koszt; sprawdzone[niesprawdzony] = true; liczbaSprawdzonych++; break; } } else { if (i>0 && (pierwszyPrzedzial == i-1) && nieuzytePrzedzialy[i-1] == false) pierwszyPrzedzial++; } } } printf("%lld\n", calkowityKoszt); return 0; } |