POJ Problem 2182 - Lost Cows

POJ Problem 2182 - Lost Cows

原题地址

题目类型：树状数组、二分、标记

题意

有编号是 $1～n$ 的 $n$ 个数字， $2\le n\le 8000$, 乱序排列，顺序是未知的。对于每个位置的数字，知道排在它前面比它小的数字有多少个。求这个乱序数列的顺序。

分析

观察发现最后一位数可以直接确定，然后将这个数从 $1-n$ 的集合中划去，然后再看倒数第二个数，发现这个数应该是剩下的数中第 $a[i]+1$ 小的数，然后再将这个数划去。从后往前重复次操作便可求出所有的结果。可以通过树状数组来维护这个集合，然后每次二分去找第 a[i] + 1 小的数。时间复杂度 $O(n\log n\log n)$。

代码

static int[] a;
static int[] C;

public static void solve() throws IOException {
    int n = nextInt();
    
    init(n);
    
    for (int i = 2; i <= n; i++) a[i] = nextInt();
    for (int i = 1; i <= n; i++) add(i, 1, n);
    
    int[] ans = new int[n + 1];
    for (int i = n; i > 0; i--) {
        int aim = a[i] + 1;
        int l = 1, r = n;
        while (l < r) {
            int mid = l + r >> 1;
            if (getSum(mid) >= aim) r = mid;
            else l = mid + 1;
        }
        
        ans[i] = r;
        add(r, -1, n);
    }
    
    for (int i = 1; i <= n; i++) pw.println(ans[i]);
} 

public static void add(int x, int v, int n) {
    while (x <= n) {
        C[x] += v;
        x += lowbit(x);
    }
}

public static int getSum(int x) {
    int re = 0;
    while (x > 0) {
        re += C[x];
        x -= lowbit(x);
    }
    return re;
}

public static int lowbit(int x) { return x & (-x); }

public static void init(int n) {
    a = new int[n + 1];
    C = new int[n + 1];
}

/*******************************************************************************************/


/*******************************************************************************************/

// Fast I/O