#include<bits/stdc++.h> #define pb push_back using namespace std; int n, a, k, r=1; long long wyn; bool bn[500505]; int main() { scanf("%d%d", &n, &k); r=1; for(int i=1; i<=n; i++) { scanf("%d", &a); if(bn[a]==0 && r<=k) { bn[a]=1; wyn+=i-r; r++; } } if(r<=k) { printf("-1"); } else { printf("%d", wyn); } }
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 | #include<bits/stdc++.h> #define pb push_back using namespace std; int n, a, k, r=1; long long wyn; bool bn[500505]; int main() { scanf("%d%d", &n, &k); r=1; for(int i=1; i<=n; i++) { scanf("%d", &a); if(bn[a]==0 && r<=k) { bn[a]=1; wyn+=i-r; r++; } } if(r<=k) { printf("-1"); } else { printf("%d", wyn); } } |