其他分享
首页 > 其他分享> > F - Shift and Inversions

F - Shift and Inversions

作者:互联网

F - Shift and Inversions
题目大意
给你n个0到n-1全排列的数组,然后这个数组会循环左移n-1
问你这个过程中数组的逆序数对是多少
逆序数对,满足i<j, ai>aj 的对<i,j>的个数
2≤N≤3×1e5
思路
初始的逆序对ans用树状数组或者归并排序做,这里就用树状数组的了,
然后比如{2, 0, 1, 3}这个对循环左移变成了{0, 1, 3, 2},就是2往后移了,那么现在,比2的数移到了左边,逆序对就少了a1对,比2大的数也移到了左边,逆序对就多了a2对
由于他是全排列,那么比2小的数a1就有2对,比2大的数a2就有4 - 1 - 2对,也就是说只需要ans - a1 + a2就行了。
逆序对最大值是n*(n - 1) / 2所以会爆int,需要开long long
ac代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define lowbit(x) x & -x
const int maxn = 3e5 + 5;
int c[maxn], n;
void add(int x){
    for(int i = x; i <= n; i += lowbit(i))
        c[i] ++;
}
int query(int x){
    int ans = 0;
    for(int i = x; i > 0; i -= lowbit(i))
        ans += c[i];
    return ans;
}
int a[maxn];
int main(){
    ll ans = 0; cin >> n;
    for(int i = 1; i <= n; i ++){
        cin >> a[i]; a[i] ++;
        //由于树状数组不能加0,所以a[i]需要加1,那么下面的d1就要对应的+1, d2要对应的-1
        add(a[i]);
        //query求的是小于等于a[i]的数的个数,i是现在有的个数
        //那么现在要求的是逆序对,也就是要求前面比他大的数的个数,那么就是i-query
        ans += i - query(a[i]); 
    }
    int d1 = 0, d2 = 0;
    for(int i = 1; i <= n; i ++){
        cout << ans << endl;
        d1 = n - a[i], d2 = a[i] - 1;
        ans += d1 - d2;
    }
    return 0;
}

标签:int,Shift,long,数组,ans,Inversions,query,逆序
来源: https://blog.csdn.net/weixin_44941429/article/details/113470365