#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; } |
polski