#include <iostream> #include<algorithm> #include <cstdio> #include <vector> #include <string> #include <cstdlib> #include <cstring> using namespace std; int main () { ios_base::sync_with_stdio(0); int ileB, ileLB, rodz; int ile, rozne; long long sekundy; vector <int> rodzaj1; vector <int> rodzaj2; vector <int> rodzaj3; vector <int> pusty; sekundy=0; cin >> ileB; cin >> ileLB; for (int n=1; n<=500000; n++) pusty.push_back(0); for (int n=1; n<=ileB; n++) { cin >> rodz; rodzaj1.push_back(rodz); rodzaj2.push_back(rodz); rodzaj3.push_back(rodz); } sort (rodzaj1.begin(), rodzaj1.begin()+ileLB-1); ile=1; rozne=rodzaj1[0]; for (int n=1; n<ileLB; n++) { if (rozne!=rodzaj1[n]) { rozne=rodzaj1[n]; ile=ile+1; } } if (ile==ileLB) cout << "0" << endl; else { sort (rodzaj2.begin(), rodzaj2.end()); ile=1; rozne=rodzaj2[0]; for (int n=1; n<ileB; n++) { if (rozne!=rodzaj2[n]) { rozne=rodzaj2[n]; ile=ile+1; } } // cout << ile << endl; // for (int n=0; n<ileB; n++) // cout << rodzaj2[n] << " "; if (ile<ileLB) cout << "-1" << endl; else { pusty[rodzaj3[0]]=1; // cout << rodzaj3[0] << "a" << endl; ile=1; for (int n=1; n<ileB; n++) { if (pusty[rodzaj3[n]]==0) { pusty[rodzaj3[n]]=1; ile=ile+1; if (ile<=ileLB) sekundy=sekundy+n-ile+1; if (ile>ileLB) break; // cout << rodzaj3[n] << " " << ile << " " << n << endl; } } cout << sekundy << 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 | #include <iostream> #include<algorithm> #include <cstdio> #include <vector> #include <string> #include <cstdlib> #include <cstring> using namespace std; int main () { ios_base::sync_with_stdio(0); int ileB, ileLB, rodz; int ile, rozne; long long sekundy; vector <int> rodzaj1; vector <int> rodzaj2; vector <int> rodzaj3; vector <int> pusty; sekundy=0; cin >> ileB; cin >> ileLB; for (int n=1; n<=500000; n++) pusty.push_back(0); for (int n=1; n<=ileB; n++) { cin >> rodz; rodzaj1.push_back(rodz); rodzaj2.push_back(rodz); rodzaj3.push_back(rodz); } sort (rodzaj1.begin(), rodzaj1.begin()+ileLB-1); ile=1; rozne=rodzaj1[0]; for (int n=1; n<ileLB; n++) { if (rozne!=rodzaj1[n]) { rozne=rodzaj1[n]; ile=ile+1; } } if (ile==ileLB) cout << "0" << endl; else { sort (rodzaj2.begin(), rodzaj2.end()); ile=1; rozne=rodzaj2[0]; for (int n=1; n<ileB; n++) { if (rozne!=rodzaj2[n]) { rozne=rodzaj2[n]; ile=ile+1; } } // cout << ile << endl; // for (int n=0; n<ileB; n++) // cout << rodzaj2[n] << " "; if (ile<ileLB) cout << "-1" << endl; else { pusty[rodzaj3[0]]=1; // cout << rodzaj3[0] << "a" << endl; ile=1; for (int n=1; n<ileB; n++) { if (pusty[rodzaj3[n]]==0) { pusty[rodzaj3[n]]=1; ile=ile+1; if (ile<=ileLB) sekundy=sekundy+n-ile+1; if (ile>ileLB) break; // cout << rodzaj3[n] << " " << ile << " " << n << endl; } } cout << sekundy << endl; } } return 0; } |