C - Swaps 2
给定两个长度为nnn的数组A,BA, BA,B,我们可以进行若干次如下操作,使AAA变成BBB,
- 选定i<ni < ni<n,将aia_iai减小111,将ai+1a_{i + 1}ai+1增加111。
- 交换ai,ai+1a_i, a_{i + 1}ai,ai+1。
问我们进行多少次操作,可以将AAA变成BBB,如果无解则输出−1-1−1,否则输出最小操作次数。
容易发现,不管怎么交换ai+ia_i + iai+i的值都是不变的,当我们要得到bib_ibi的时候,我们一定是去找一个aj+j=bi+ia_j + j = b_i + iaj+j=bi+i,且jjj最小的那个点,这样可以减小操作次数,
所以可以考虑先离散化,然后用vectorvectorvector存下ai+ia_i + iai+i的所有下标。因为每个数的下标都是会变化的,所以我们要知道AAA中每个点的真实下标。
容易发现只有原始数组中的某个数下标大于当前下标,如果这个数向前移动了,这个时候才会使得我们得数组下标变大,所以只要用数组数组存一下即可。
#include <bits/stdc++.h>using namespace std;const int N = 2e5 + 10;int a[N], b[N], c[N], sum[N], n, m;vector<int> vt[N];inline int lowbit(int x) {return x & (-x);
}void update(int x, int v) {while (x <= n) {sum[x] += v;x += lowbit(x);}
}int query(int x) {int ans = 0;while (x) {ans += sum[x];x -= lowbit(x);}return ans;
}int query(int l, int r) {return query(r) - query(l - 1);
}int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);scanf("%d", &n);for (int i = 1; i <= n; i++) {scanf("%d", &a[i]);c[i] = a[i] + i;}sort(c + 1, c + 1 + n);m = unique(c + 1, c + 1 + n) - (c + 1);for (int i = 1; i <= n; i++) {scanf("%d", &b[i]);int pos = lower_bound(c + 1, c + 1 + m, b[i] + i) - c;if (c[pos] != b[i] + i) {puts("-1");return 0;}b[i] = pos;vt[lower_bound(c + 1, c + 1 + m, a[i] + i) - c].push_back(i);}long long ans = 0;for (int i = 1; i <= n; i++) {if (!vt[b[i]].size()) {puts("-1");return 0;}int cur = vt[b[i]].back();ans += cur + query(cur + 1, n) - i;vt[b[i]].pop_back();update(cur, 1);}printf("%lld\n", ans);return 0;
}