POJ Problem 2182 - Lost Cows
题目类型:树状数组、二分、标记
题意
有编号是 1 ~ n 1~n 1~n 的 n n n 个数字, 2 ≤ n ≤ 8000 2 \leq n \leq 8000 2≤n≤8000, 乱序排列,顺序是未知的。对于每个位置的数字,知道排在它前面比它小的数字有多少个。求这个乱序数列的顺序。
分析
观察发现最后一位数可以直接确定,然后将这个数从 1 − n 1 - n 1−n 的集合中划去,然后再看倒数第二个数,发现这个数应该是剩下的数中第 a [ i ] + 1 a[i] + 1 a[i]+1 小的数,然后再将这个数划去。从后往前重复次操作便可求出所有的结果。可以通过树状数组来维护这个集合,然后每次二分去找第 a[i] + 1 小的数。时间复杂度 O ( n log n log n ) O(n\logn \log n) O(nlognlogn)。
代码
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