#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define mid (l+r)/2
#define PI acos((ld)-1)
#define BOOST ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0)
typedef long long ll;
typedef long double ld;
typedef pair<int, int> ii;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<ii> vii;
typedef vector<vi> vvi;
const int N = 5e5+5;
int in[N];
bool vis[N];
int main(){
BOOST;
int n, k, cnt = 0;
ll ans = 0;
cin >> n >> k;
for(int i=1; i<=n; i++){
cin >> in[i];
if(i <= k && vis[in[i]])
ans += k-i, cnt++;
if(cnt > 0 && i > k && !vis[in[i]])
ans += i-k, cnt--;
vis[in[i]] = 1;
}
if(cnt > 0)
cout << -1;
else
cout << ans;
}
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 | #include <bits/stdc++.h> using namespace std; #define fi first #define se second #define pb push_back #define all(x) (x).begin(), (x).end() #define mid (l+r)/2 #define PI acos((ld)-1) #define BOOST ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0) typedef long long ll; typedef long double ld; typedef pair<int, int> ii; typedef vector<int> vi; typedef vector<ll> vll; typedef vector<ii> vii; typedef vector<vi> vvi; const int N = 5e5+5; int in[N]; bool vis[N]; int main(){ BOOST; int n, k, cnt = 0; ll ans = 0; cin >> n >> k; for(int i=1; i<=n; i++){ cin >> in[i]; if(i <= k && vis[in[i]]) ans += k-i, cnt++; if(cnt > 0 && i > k && !vis[in[i]]) ans += i-k, cnt--; vis[in[i]] = 1; } if(cnt > 0) cout << -1; else cout << ans; } |
English