#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int, int> PII;
typedef pair<ll, ll> PLL;
typedef pair<ld, ld> PLD;
typedef vector<int> VI;
typedef vector<ll> VL;
typedef vector<PII> VII;
typedef vector<VI> VVI;
typedef vector<VL> VVL;
typedef vector<VII> VVII;
typedef vector<PLL> VLL;
typedef vector<VLL> VVLL;
typedef vector<VI> VVI;
typedef vector<VL> VVL;
typedef vector<bool> VB;
template <class T, class G>
ostream &operator<<(ostream &os, const pair<T, G> ¶) {
os << para.first << " " << para.second;
return os;
}
template <class T> ostream &operator<<(ostream &os, const vector<T> &vec) {
for (const T &el : vec)
os << el << " ";
return os;
}
template <class T> ostream &operator<<(ostream &os, const set<T> &vec) {
for (const T &el : vec)
os << el << " ";
return os;
}
#define REP(i, j) for (int i = 0; i < j; i++)
#define REP1(i, j) for (int i = 1; i < j; i++)
#define FOREACH(el, n) for (auto &el : n)
#define LAST(x) (x[(int)x.size() - 1])
#define pb push_back
#define pf push_front
#define st first
#define nd second
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define sp ' '
#define ent '\n'
const int N = 600000;
const int M = 1e6;
const int INF = 1e6 - 5;
bitset<N> used;
struct FenwickTree {
VL bit;
int n;
FenwickTree(int n) {
this->n = n;
bit.assign(n, 0);
}
ll sum(int r) {
ll ret = 0;
for (; r >= 0; r = (r & (r + 1)) - 1)
ret += bit[r];
return ret;
}
void add(int idx, ll delta) {
for (; idx < n; idx = idx | (idx + 1))
bit[idx] += delta;
}
};
FenwickTree tr = FenwickTree(M);
int main() {
ios_base::sync_with_stdio(0);
cin.tie(NULL);
int t = 1;
// cin >> t;
while (t--) {
ll n, k;
cin >> n >> k;
VL v;
ll different = 0;
ll ans = 0;
REP(i, n) {
ll a;
cin >> a;
if (!used[a] && different < k) {
different++;
used[a] = true;
a = different;
} else {
a = INF;
}
v.pb(a);
}
if (different < k) {
cout << -1 << ent;
return 0;
}
for (ll i = 0; i < (ll)v.size(); i++) {
ans += (ll)(i - tr.sum(v[i]));
tr.add(v[i], 1);
}
cout << ans;
}
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 110 111 112 113 114 115 116 117 | #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef unsigned long long ull; typedef long double ld; typedef pair<int, int> PII; typedef pair<ll, ll> PLL; typedef pair<ld, ld> PLD; typedef vector<int> VI; typedef vector<ll> VL; typedef vector<PII> VII; typedef vector<VI> VVI; typedef vector<VL> VVL; typedef vector<VII> VVII; typedef vector<PLL> VLL; typedef vector<VLL> VVLL; typedef vector<VI> VVI; typedef vector<VL> VVL; typedef vector<bool> VB; template <class T, class G> ostream &operator<<(ostream &os, const pair<T, G> ¶) { os << para.first << " " << para.second; return os; } template <class T> ostream &operator<<(ostream &os, const vector<T> &vec) { for (const T &el : vec) os << el << " "; return os; } template <class T> ostream &operator<<(ostream &os, const set<T> &vec) { for (const T &el : vec) os << el << " "; return os; } #define REP(i, j) for (int i = 0; i < j; i++) #define REP1(i, j) for (int i = 1; i < j; i++) #define FOREACH(el, n) for (auto &el : n) #define LAST(x) (x[(int)x.size() - 1]) #define pb push_back #define pf push_front #define st first #define nd second #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() #define sp ' ' #define ent '\n' const int N = 600000; const int M = 1e6; const int INF = 1e6 - 5; bitset<N> used; struct FenwickTree { VL bit; int n; FenwickTree(int n) { this->n = n; bit.assign(n, 0); } ll sum(int r) { ll ret = 0; for (; r >= 0; r = (r & (r + 1)) - 1) ret += bit[r]; return ret; } void add(int idx, ll delta) { for (; idx < n; idx = idx | (idx + 1)) bit[idx] += delta; } }; FenwickTree tr = FenwickTree(M); int main() { ios_base::sync_with_stdio(0); cin.tie(NULL); int t = 1; // cin >> t; while (t--) { ll n, k; cin >> n >> k; VL v; ll different = 0; ll ans = 0; REP(i, n) { ll a; cin >> a; if (!used[a] && different < k) { different++; used[a] = true; a = different; } else { a = INF; } v.pb(a); } if (different < k) { cout << -1 << ent; return 0; } for (ll i = 0; i < (ll)v.size(); i++) { ans += (ll)(i - tr.sum(v[i])); tr.add(v[i], 1); } cout << ans; } return 0; } |
English