#include <bits/stdc++.h> using namespace std; bool tab[1000000]; int main() { int n,k,w=0,r=0; cin>>n>>k; for(int i = 1; i<=n ;i ++) { int x; cin>>x; if(!tab[x]&&r<k) { w+=i; r++; tab[x]=1; } } if(r<k)cout<<-1; else cout<<w-(k+1)*k/2; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | #include <bits/stdc++.h> using namespace std; bool tab[1000000]; int main() { int n,k,w=0,r=0; cin>>n>>k; for(int i = 1; i<=n ;i ++) { int x; cin>>x; if(!tab[x]&&r<k) { w+=i; r++; tab[x]=1; } } if(r<k)cout<<-1; else cout<<w-(k+1)*k/2; } |