#include <bits/stdc++.h>
template <typename... Ts>
void __print(Ts &&...ts) {}
#ifdef DEBUG
#include "print.hpp"
#endif // DEBUG
using namespace std;
template <class T>
using pq_max = priority_queue<T>;
template <class T>
using pq_min = priority_queue<T, vector<T>, greater<T>>;
#define FOR(i, a, b) for (int i = a; i < (b); ++i)
#define F0R(i, a) for (int i = 0; i < (a); ++i)
#define FORd(i, a, b) for (int i = (b)-1; i >= a; --i)
#define F0Rd(i, a) for (int i = (a)-1; i >= 0; --i)
#define uid(a, b) uniform_int_distribution<int>(a, b)(rng)
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
#define all(x) x.begin(), x.end()
#define GET_FOR(_1, _2, _3, NAME, ...) NAME
#define loop(...) GET_FOR(__VA_ARGS__, FOR, F0R)(__VA_ARGS__)
#define GET_FORd(_1, _2, _3, NAME, ...) NAME
#define loopr(...) GET_FOR(__VA_ARGS__, FORd, F0Rd)(__VA_ARGS__)
#define sz(x) ((int)(x).size())
using ll = long long;
template <typename T>
using vec = vector<T>;
template <typename T, size_t N>
using arr = array<T, N>;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using vvl = vector<vl>;
using pi = pair<int, int>;
using pl = pair<ll, ll>;
///////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////
class Solution {};
// #define N 10
#define N (5 * int(1e5) + 2)
int tab[N];
int remap[N];
int result(int n, int k) {
memset(remap, 0xFF, sizeof(int) * n);
loop(i, n) {
int a;
cin >> a;
tab[i] = a - 1;
}
int idx = 0;
for (int a : tab) {
if (remap[a] == -1) {
remap[a] = idx++;
}
if (idx == k) {
break;
}
}
if (idx < k) {
return -1;
}
idx = 0;
int res = 0;
loop(i, n) {
int a = tab[i];
a = remap[a];
if (a == idx) {
res += i - a;
++idx;
}
}
return res;
}
void solution() {
int n, k;
cin >> n >> k;
cout << result(n, k);
}
int main(void) {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int qs = 1;
// cin >> qs;
while (qs--) {
solution();
}
// Solution solution;
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 107 108 109 | #include <bits/stdc++.h> template <typename... Ts> void __print(Ts &&...ts) {} #ifdef DEBUG #include "print.hpp" #endif // DEBUG using namespace std; template <class T> using pq_max = priority_queue<T>; template <class T> using pq_min = priority_queue<T, vector<T>, greater<T>>; #define FOR(i, a, b) for (int i = a; i < (b); ++i) #define F0R(i, a) for (int i = 0; i < (a); ++i) #define FORd(i, a, b) for (int i = (b)-1; i >= a; --i) #define F0Rd(i, a) for (int i = (a)-1; i >= 0; --i) #define uid(a, b) uniform_int_distribution<int>(a, b)(rng) mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); #define all(x) x.begin(), x.end() #define GET_FOR(_1, _2, _3, NAME, ...) NAME #define loop(...) GET_FOR(__VA_ARGS__, FOR, F0R)(__VA_ARGS__) #define GET_FORd(_1, _2, _3, NAME, ...) NAME #define loopr(...) GET_FOR(__VA_ARGS__, FORd, F0Rd)(__VA_ARGS__) #define sz(x) ((int)(x).size()) using ll = long long; template <typename T> using vec = vector<T>; template <typename T, size_t N> using arr = array<T, N>; using vi = vector<int>; using vvi = vector<vi>; using vl = vector<ll>; using vvl = vector<vl>; using pi = pair<int, int>; using pl = pair<ll, ll>; /////////////////////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////////////////// class Solution {}; // #define N 10 #define N (5 * int(1e5) + 2) int tab[N]; int remap[N]; int result(int n, int k) { memset(remap, 0xFF, sizeof(int) * n); loop(i, n) { int a; cin >> a; tab[i] = a - 1; } int idx = 0; for (int a : tab) { if (remap[a] == -1) { remap[a] = idx++; } if (idx == k) { break; } } if (idx < k) { return -1; } idx = 0; int res = 0; loop(i, n) { int a = tab[i]; a = remap[a]; if (a == idx) { res += i - a; ++idx; } } return res; } void solution() { int n, k; cin >> n >> k; cout << result(n, k); } int main(void) { ios_base::sync_with_stdio(false); cin.tie(NULL); int qs = 1; // cin >> qs; while (qs--) { solution(); } // Solution solution; return 0; } |
English