#include <bits/stdc++.h>
#define FOR(i,n) for(int i = 0; i<n; ++i)
#define FOREACH(v, a) for(auto & v : (a))
#define X first
#define Y second
#define PR std::pair
#define MPR std::make_pair
typedef long long ll;
typedef std::vector<int> VI;
using namespace std;
#pragma region Ready functions
int fastPower(int a, int b){
int res=1, pop_a = a;
while(b){
if(b&1) res *= pop_a;
pop_a *= pop_a;
b>>=1;
}
return res;
}
#pragma endregion
int closestPos[500001];
priority_queue<pair<ll,int>> marki;
int main(){
ios_base::sync_with_stdio(0); cin.tie(0);
int n,k;
cin>>n>>k;
//cout<<n<<endl;
FOR(i, n + 1) closestPos[i] = 9999999;
for(int i = 0; i<n; i++){
int x;
cin>>x;
closestPos[x] = min(closestPos[x], i);
}
for(int i = 1; i<=n; i++){
if(closestPos[i] == 9999999) continue;
marki.push({-1*closestPos[i], -1*i});
}
ll moves = 0;
ll rozne = 0;
//cout<<marki.size()<<endl;
while(marki.size() && rozne < k){
moves += -1*marki.top().first - rozne;
rozne++;
marki.pop();
}
if(rozne < k){
cout<<-1;
}
else{
cout<<moves;
}
}
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 | #include <bits/stdc++.h> #define FOR(i,n) for(int i = 0; i<n; ++i) #define FOREACH(v, a) for(auto & v : (a)) #define X first #define Y second #define PR std::pair #define MPR std::make_pair typedef long long ll; typedef std::vector<int> VI; using namespace std; #pragma region Ready functions int fastPower(int a, int b){ int res=1, pop_a = a; while(b){ if(b&1) res *= pop_a; pop_a *= pop_a; b>>=1; } return res; } #pragma endregion int closestPos[500001]; priority_queue<pair<ll,int>> marki; int main(){ ios_base::sync_with_stdio(0); cin.tie(0); int n,k; cin>>n>>k; //cout<<n<<endl; FOR(i, n + 1) closestPos[i] = 9999999; for(int i = 0; i<n; i++){ int x; cin>>x; closestPos[x] = min(closestPos[x], i); } for(int i = 1; i<=n; i++){ if(closestPos[i] == 9999999) continue; marki.push({-1*closestPos[i], -1*i}); } ll moves = 0; ll rozne = 0; //cout<<marki.size()<<endl; while(marki.size() && rozne < k){ moves += -1*marki.top().first - rozne; rozne++; marki.pop(); } if(rozne < k){ cout<<-1; } else{ cout<<moves; } } |
English