/* -----------------------
Autor: Tomasz Boguslawski
-------------------------- */
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<fstream>
#include<iomanip>
#include<string>
#include<sstream>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include <fstream>
#include<math.h>
#define LL long long
#define FOR(x, b, e) for(LL x = b; x <= (e); x++)
#define FORD(x, b, e) for(LL x = b; x >= (e); x--)
#define VAR(v, n) __typeof(n) v = (n)
#define ALL(c) (c).begin(), (c).end()
#define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i)
#define DEBUG if (debug)
#define MIN(a,b) ((a>b)?b:a)
#define MAX(a,b) ((a>b)?a:b)
using namespace std;
LL a[501000];
/// MAIN
int main(int argc, char* argv[])
{
// magic formula, which makes streams work faster:
ios_base::sync_with_stdio(0);
LL n; LL k;
cin >> n; cin >> k;
FOR(i,1,n) cin >> a[i];
LL pom;
set<LL> already;
LL cost = 0;
LL s = 0;
FOR(j,1,k)
{
// po wykonaniu petli oranzady na pozycjach 1..j beda rozne
if (already.count(a[j])>0)
{
// element juz byl, trzeba poszukac czegos na wymiane
if (s<j+1) s=j+1;
while ((s<=n)&&(already.count(a[s])>0)) s++;
if (s>n) { cout << "-1\n"; return 0; };
cost+=(s-j);
pom=a[s]; a[s]=a[j]; a[j]=pom;
};
already.insert(a[j]);
};
cout << cost << '\n';
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 | /* ----------------------- Autor: Tomasz Boguslawski -------------------------- */ #include<cstdio> #include<cstdlib> #include<iostream> #include<fstream> #include<iomanip> #include<string> #include<sstream> #include<cstring> #include<map> #include<vector> #include<set> #include<queue> #include<algorithm> #include <fstream> #include<math.h> #define LL long long #define FOR(x, b, e) for(LL x = b; x <= (e); x++) #define FORD(x, b, e) for(LL x = b; x >= (e); x--) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define DEBUG if (debug) #define MIN(a,b) ((a>b)?b:a) #define MAX(a,b) ((a>b)?a:b) using namespace std; LL a[501000]; /// MAIN int main(int argc, char* argv[]) { // magic formula, which makes streams work faster: ios_base::sync_with_stdio(0); LL n; LL k; cin >> n; cin >> k; FOR(i,1,n) cin >> a[i]; LL pom; set<LL> already; LL cost = 0; LL s = 0; FOR(j,1,k) { // po wykonaniu petli oranzady na pozycjach 1..j beda rozne if (already.count(a[j])>0) { // element juz byl, trzeba poszukac czegos na wymiane if (s<j+1) s=j+1; while ((s<=n)&&(already.count(a[s])>0)) s++; if (s>n) { cout << "-1\n"; return 0; }; cost+=(s-j); pom=a[s]; a[s]=a[j]; a[j]=pom; }; already.insert(a[j]); }; cout << cost << '\n'; return 0; }; |
English