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