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