#include <bits/stdc++.h> using namespace std; #define int long long main() { int n,k; cin>>n>>k; vector<bool> used(n+1, 0); vector<int> kol; for(int i=0; i<n; i++) { int x; cin>>x; if(used[x]==0 && kol.size()<k) { used[x]=1; kol.push_back(i); } } int w=0; if(kol.size()!=k) cout<<-1; else { for(int i=0; i<k; i++) w+=kol[i]-i; cout<<w; } }
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; #define int long long main() { int n,k; cin>>n>>k; vector<bool> used(n+1, 0); vector<int> kol; for(int i=0; i<n; i++) { int x; cin>>x; if(used[x]==0 && kol.size()<k) { used[x]=1; kol.push_back(i); } } int w=0; if(kol.size()!=k) cout<<-1; else { for(int i=0; i<k; i++) w+=kol[i]-i; cout<<w; } } |