#include<iostream>
using namespace std;
constexpr int MAX_N = 500007;
int marka[MAX_N];
bool wziete[MAX_N];
int n, k, rozne;
long long res = 0;
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> n >> k;
for(int i = 0, x;i < n;i++){
cin >> x;
if(k == 0) continue;
if(wziete[x] == false){
wziete[x] = true;
k--;
res += i - rozne++;
}
}
if (k > 0) cout << "-1\n";
else cout << res << '\n';
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | #include<iostream> using namespace std; constexpr int MAX_N = 500007; int marka[MAX_N]; bool wziete[MAX_N]; int n, k, rozne; long long res = 0; int main(){ ios_base::sync_with_stdio(0); cin.tie(0), cout.tie(0); cin >> n >> k; for(int i = 0, x;i < n;i++){ cin >> x; if(k == 0) continue; if(wziete[x] == false){ wziete[x] = true; k--; res += i - rozne++; } } if (k > 0) cout << "-1\n"; else cout << res << '\n'; } |
English