#include <iostream> #include <set> #include <unordered_set> #include <map> #include <unordered_map> #include <vector> #include <tuple> using namespace std; typedef long long LL; typedef long double LLD; typedef unsigned long long ULL; typedef vector<int> VI; typedef vector<pair<int, int>> VPII; typedef pair<int, int> PII; #define FOR(x, b, e) for(int x = b; x <= (e); ++x) #define FORD(x, b, e) for(int x = b; x >= (e); --x) #define REP(x, n) for(int x = 0; x < (n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define ST first #define ND second #define INF 1000000001 #define MP make_pair #ifndef ONLINE_JUDGE #define debug(x) cerr << #x <<" "; _print(x); cerr << endl; #else #define debug(x) #endif void _print(LL t) {cerr << t;} void _print(int t) {cerr << t;} void _print(string t) {cerr << t;} void _print(char t) {cerr << t;} void _print(LLD t) {cerr << t;} void _print(double t) {cerr << t;} void _print(ULL t) {cerr << t;} template <class T, class V> void _print(pair <T, V> p); template <class T> void _print(vector <T> v); template <class T> void _print(set <T> v); template <class T, class V> void _print(map <T, V> v); template <class T> void _print(multiset <T> v); template <class T, class V> void _print(pair <T, V> p) {cerr << "{"; _print(p.ST); cerr << ","; _print(p.ND); cerr << "}";} template <class T> void _print(vector <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";} template <class T> void _print(set <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";} template <class T> void _print(multiset <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";} template <class T, class V> void _print(map <T, V> v) {cerr << "[ "; for (auto i : v) {_print(i); cerr << " ";} cerr << "]";} int main() { ios::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, a; cin >> n >> k; vector<int> inpt(n); vector<int> there_is(n + 1, false); vector<bool> taken_in_new_arr(n + 1, false); int counter = 0; int max_index = -1; bool brake = false; for(int i = 0; i < n; ++i){ cin >> a; if(!brake){ inpt[i] = a; if(there_is[a] == 0) counter++; there_is[a]++; if(counter == k){ max_index = i; brake = true; } } } if(max_index == -1){ cout << (-1) << endl; return 0; } LL moves = 0; LL moved_nums = 0; for(int i = max_index; i >= 0; --i){ if(taken_in_new_arr[inpt[i]] == false && there_is[inpt[i]] == 1){ taken_in_new_arr[inpt[i]] = true; } else { moves += (max_index - i) - moved_nums; moved_nums++; there_is[inpt[i]]--; } } cout << moves << endl; 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 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 | #include <iostream> #include <set> #include <unordered_set> #include <map> #include <unordered_map> #include <vector> #include <tuple> using namespace std; typedef long long LL; typedef long double LLD; typedef unsigned long long ULL; typedef vector<int> VI; typedef vector<pair<int, int>> VPII; typedef pair<int, int> PII; #define FOR(x, b, e) for(int x = b; x <= (e); ++x) #define FORD(x, b, e) for(int x = b; x >= (e); --x) #define REP(x, n) for(int x = 0; x < (n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define ST first #define ND second #define INF 1000000001 #define MP make_pair #ifndef ONLINE_JUDGE #define debug(x) cerr << #x <<" "; _print(x); cerr << endl; #else #define debug(x) #endif void _print(LL t) {cerr << t;} void _print(int t) {cerr << t;} void _print(string t) {cerr << t;} void _print(char t) {cerr << t;} void _print(LLD t) {cerr << t;} void _print(double t) {cerr << t;} void _print(ULL t) {cerr << t;} template <class T, class V> void _print(pair <T, V> p); template <class T> void _print(vector <T> v); template <class T> void _print(set <T> v); template <class T, class V> void _print(map <T, V> v); template <class T> void _print(multiset <T> v); template <class T, class V> void _print(pair <T, V> p) {cerr << "{"; _print(p.ST); cerr << ","; _print(p.ND); cerr << "}";} template <class T> void _print(vector <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";} template <class T> void _print(set <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";} template <class T> void _print(multiset <T> v) {cerr << "[ "; for (T i : v) {_print(i); cerr << " ";} cerr << "]";} template <class T, class V> void _print(map <T, V> v) {cerr << "[ "; for (auto i : v) {_print(i); cerr << " ";} cerr << "]";} int main() { ios::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k, a; cin >> n >> k; vector<int> inpt(n); vector<int> there_is(n + 1, false); vector<bool> taken_in_new_arr(n + 1, false); int counter = 0; int max_index = -1; bool brake = false; for(int i = 0; i < n; ++i){ cin >> a; if(!brake){ inpt[i] = a; if(there_is[a] == 0) counter++; there_is[a]++; if(counter == k){ max_index = i; brake = true; } } } if(max_index == -1){ cout << (-1) << endl; return 0; } LL moves = 0; LL moved_nums = 0; for(int i = max_index; i >= 0; --i){ if(taken_in_new_arr[inpt[i]] == false && there_is[inpt[i]] == 1){ taken_in_new_arr[inpt[i]] = true; } else { moves += (max_index - i) - moved_nums; moved_nums++; there_is[inpt[i]]--; } } cout << moves << endl; return 0; } |