C - Swaps 2

給定兩個長度為$n$的數組$A,B$，我們可以進行若干次如下操作，使$A$變成$B$，

• 選定$i<n$，將$a_i$減小$1$，將$a_{i+1}$增加$1$。
• 交換$a_i,a_{i+1}$。

問我們進行多少次操作，可以將$A$變成$B$，如果無解則輸出$?1$，否則輸出最小操作次數。

容易發現，不管怎么交換$a_i+i$的值都是不變的，當我們要得到$b_i$的時候，我們一定是去找一個$a_j+j=b_i+i$，且$j$最小的那個點，這樣可以減小操作次數，

所以可以考慮先離散化，然后用$vector$存下$a_i+i$的所有下標。因為每個數的下標都是會變化的，所以我們要知道$A$中每個點的真實下標。

容易發現只有原始數組中的某個數下標大于當前下標，如果這個數向前移動了，這個時候才會使得我們得數組下標變大，所以只要用數組數組存一下即可。

#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; }

總結

以上是生活随笔為你收集整理的C - Swaps 2（树状数组，思维）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。