//==============
// Zadanie: Oranżada (Dzień 1 PA 2021)
// URL: https://sio2.mimuw.edu.pl/c/pa-2021-1/p/ora/
// Autor: Piotr Wieczorek
//==============
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define REP(i, n) for (int (i) = 0; (i) < (n); (i)++)
#define REPE(i, n) for (int (i) = 0; (i) <= (n); (i)++)
#define FOR(i, z, n) for (int (i) = (z); (i) < (n); (i)++)
#define FORE(i, z, n) for (int (i) = (z); (i) <= (n); (i)++)
typedef long long ll;
// ----------
// SOLUTION
// ----------
#define MAX_N 500000
// ix
vector<int> arr;
int n, k;
void init() {
bitset<MAX_N+1> out;
scanf("%d%d", &n, &k);
REP(i, n) {
int a;
scanf("%d", &a);
if (!out[a]){
out[a] = 1;
arr.pb(i);
}
}
}
int main() {
init();
if (arr.size() < k) {
printf("-1\n");
return 0;
}
ll sum = 0;
int pos = 1;
FOR(i, 1, k) {
sum += arr[i] - pos;
pos++;
}
printf("%lld\n", sum);
}
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 | //============== // Zadanie: Oranżada (Dzień 1 PA 2021) // URL: https://sio2.mimuw.edu.pl/c/pa-2021-1/p/ora/ // Autor: Piotr Wieczorek //============== #include <bits/stdc++.h> using namespace std; #define pb push_back #define REP(i, n) for (int (i) = 0; (i) < (n); (i)++) #define REPE(i, n) for (int (i) = 0; (i) <= (n); (i)++) #define FOR(i, z, n) for (int (i) = (z); (i) < (n); (i)++) #define FORE(i, z, n) for (int (i) = (z); (i) <= (n); (i)++) typedef long long ll; // ---------- // SOLUTION // ---------- #define MAX_N 500000 // ix vector<int> arr; int n, k; void init() { bitset<MAX_N+1> out; scanf("%d%d", &n, &k); REP(i, n) { int a; scanf("%d", &a); if (!out[a]){ out[a] = 1; arr.pb(i); } } } int main() { init(); if (arr.size() < k) { printf("-1\n"); return 0; } ll sum = 0; int pos = 1; FOR(i, 1, k) { sum += arr[i] - pos; pos++; } printf("%lld\n", sum); } |
English