#include <bits/stdc++.h> using namespace std; bool tab[500001]; int n, k, a, res, last=1; vector<int>vec; int main(){ scanf("%d%d", &n, &k); for(int i=0; i<n; i++){ scanf("%d", &a); if(tab[a]==0){ tab[a]=1; vec.push_back(i+1); } } if(vec.size()<k){ printf("-1"); return 0; } for(int i=0; i<k; i++){ res+=(vec[i]-last); last++; } printf("%d", res); return 0; }
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 | #include <bits/stdc++.h> using namespace std; bool tab[500001]; int n, k, a, res, last=1; vector<int>vec; int main(){ scanf("%d%d", &n, &k); for(int i=0; i<n; i++){ scanf("%d", &a); if(tab[a]==0){ tab[a]=1; vec.push_back(i+1); } } if(vec.size()<k){ printf("-1"); return 0; } for(int i=0; i<k; i++){ res+=(vec[i]-last); last++; } printf("%d", res); return 0; } |