#include <bits/stdc++.h>
using namespace std;
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
long long n, k;
cin >> n >> k;
int marki = 0;
vector <bool> uzyte(500007, false);
long long suma = 0;
int x, pomk = k;
for(int i = 0; i<n && pomk; ++i){
cin >> x;
if(uzyte[x] == false){
suma += i;
uzyte[x] = true;
pomk--;
marki++;
}
}
if(k > marki){
cout << -1 << "\n";
return 0;
}
long long odp = suma - (k-1) - (((k-1)*(k-2))/2);
cout << odp << "\n";
}
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 | #include <bits/stdc++.h> using namespace std; int main(){ ios_base::sync_with_stdio(0); cin.tie(0); long long n, k; cin >> n >> k; int marki = 0; vector <bool> uzyte(500007, false); long long suma = 0; int x, pomk = k; for(int i = 0; i<n && pomk; ++i){ cin >> x; if(uzyte[x] == false){ suma += i; uzyte[x] = true; pomk--; marki++; } } if(k > marki){ cout << -1 << "\n"; return 0; } long long odp = suma - (k-1) - (((k-1)*(k-2))/2); cout << odp << "\n"; } |
English