程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> HDU 1394 Minimum Inversion Number（樹狀數組||線段樹）

HDU 1394 Minimum Inversion Number（樹狀數組||線段樹）

編輯：關於C++

題目鏈接：點擊打開鏈接

對於求逆序數的問題，通常用線段樹或者樹狀數組來維護，樹狀數組代碼短，好寫，還是盡量寫樹狀數組吧。

首先求出原始排列的逆序數，那麼對於每一次操作，因為都是將當前排列的第一個數拿到最後一個位置，所以答案就增加了所有比他大的數字個數，減小了所有比他小的數字個數。

細節參見代碼：

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 1000000000;
const int maxn = 5000+10;
int T,n,m,bit[maxn],a[maxn];
int sum(int x) {
    int ans = 0;
    while(x > 0) {
        ans += bit[x];
        x -= x & -x;
    }
    return ans;
}
void add(int x, int d) {
    while(x <= n) {
        bit[x] += d;
        x += x & -x;
    }
}
int main() {
    while(~scanf("%d",&n)) {
        memset(bit, 0, sizeof(bit));
        int ans = 0;
        for(int i=1;i<=n;i++) {
            scanf("%d",&a[i]);
            a[i]++;
            ans += sum(n) - sum(a[i]);
            add(a[i], 1);
        }
        int cur = ans;
        for(int i=1;i