#include <cstdio> #include <cmath> #include <iostream> #include <algorithm> #include <queue> #include <list> #include <vector> #include <map> #include <set> #include <string> #include <cstring> #include <cassert> #include <limits.h> #include <unordered_set> using namespace std; typedef long long LL; typedef long double LD; typedef vector<int> VI; typedef pair<int,int> PII; #define REP(i,n) for(int i=0;i<(n);i++) #define FOR(i,a,b) for(VAR(i,a);i<=(b);++i) #define FORD(i,a,b) for(VAR(i,a);i>=(b);--i) #define FORE(a,b) for(VAR(a,(b).begin());a!=(b).end();++a) #define VAR(a,b) __typeof(b) a=(b) #define SIZE(a) ((int)((a).size())) #define ALL(x) (x).begin(),(x).end() #define CLR(x,a) memset(x,a,sizeof(x)) #define ZERO(x) memset(x,0,sizeof(x)) #define S(t,i) scanf("%" ## t, &i) #define SI(i) scanf("%d", &i) int n, m; unordered_set<int> nums; long long int result; int main() { //ios_base::sync_with_stdio(0); SI(n); SI(m); int num; int num_different = 0; FOR(i, 1, n) { SI(num); if (nums.find(num) == nums.end()) { // cout<<num<<" "<<i - num_different - 1<<"\n"; nums.insert(num); result += i - num_different - 1; num_different++; } if (num_different == m) { break; } } cout << (num_different == m ? result : -1 ) << "\n"; 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | #include <cstdio> #include <cmath> #include <iostream> #include <algorithm> #include <queue> #include <list> #include <vector> #include <map> #include <set> #include <string> #include <cstring> #include <cassert> #include <limits.h> #include <unordered_set> using namespace std; typedef long long LL; typedef long double LD; typedef vector<int> VI; typedef pair<int,int> PII; #define REP(i,n) for(int i=0;i<(n);i++) #define FOR(i,a,b) for(VAR(i,a);i<=(b);++i) #define FORD(i,a,b) for(VAR(i,a);i>=(b);--i) #define FORE(a,b) for(VAR(a,(b).begin());a!=(b).end();++a) #define VAR(a,b) __typeof(b) a=(b) #define SIZE(a) ((int)((a).size())) #define ALL(x) (x).begin(),(x).end() #define CLR(x,a) memset(x,a,sizeof(x)) #define ZERO(x) memset(x,0,sizeof(x)) #define S(t,i) scanf("%" ## t, &i) #define SI(i) scanf("%d", &i) int n, m; unordered_set<int> nums; long long int result; int main() { //ios_base::sync_with_stdio(0); SI(n); SI(m); int num; int num_different = 0; FOR(i, 1, n) { SI(num); if (nums.find(num) == nums.end()) { // cout<<num<<" "<<i - num_different - 1<<"\n"; nums.insert(num); result += i - num_different - 1; num_different++; } if (num_different == m) { break; } } cout << (num_different == m ? result : -1 ) << "\n"; return 0; } |