#ifdef _MSC_VER #ifndef __GNUC__ #pragma warning(disable: 4996) #endif #define main main0 #endif #include <iostream> using namespace std; typedef long long ll; typedef unsigned long long ull; typedef unsigned int uint; int gatunek[500001]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); ll n, k, i = 0, k1; ll wynik = 0; fill(gatunek, gatunek + 500000, 0); cin >> n >> k; k1 = k; for(i = 0; i < n && k1 > 0; ++i) { int a; cin >> a; if(gatunek[a] == 0) { gatunek[a] = 1; --k1; } else { wynik = -(2 * i + k1 - 1) * k1 / 2; break; } } for(++i ; i < n && k1 > 0; ++i) { int a; cin >> a; if(gatunek[a] == 0) { gatunek[a] = 1; --k1; wynik += i; } } if(wynik >= 0) cout << wynik << endl; else cout << "-1" << 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 | #ifdef _MSC_VER #ifndef __GNUC__ #pragma warning(disable: 4996) #endif #define main main0 #endif #include <iostream> using namespace std; typedef long long ll; typedef unsigned long long ull; typedef unsigned int uint; int gatunek[500001]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); ll n, k, i = 0, k1; ll wynik = 0; fill(gatunek, gatunek + 500000, 0); cin >> n >> k; k1 = k; for(i = 0; i < n && k1 > 0; ++i) { int a; cin >> a; if(gatunek[a] == 0) { gatunek[a] = 1; --k1; } else { wynik = -(2 * i + k1 - 1) * k1 / 2; break; } } for(++i ; i < n && k1 > 0; ++i) { int a; cin >> a; if(gatunek[a] == 0) { gatunek[a] = 1; --k1; wynik += i; } } if(wynik >= 0) cout << wynik << endl; else cout << "-1" << endl; return 0; } |