POJ 2182 线段树

Posted 2020-09-15 Flowersea

链接：

http://poj.org/problem?id=2182

题意：

有N头牛，编号1~N，乱序排成一列，现在已知每头牛前面有多少头牛比它的编号小，

求排队后从前往后数，每头牛的编号

题解：

从后往前扫描，遇到数字a，说明它是剩余序列中的第a+1个数，找到改编号并删除。

代码：

31 int a[MAXN];
32 int tree[MAXN];
33 
34 void pushup(int rt) {
35     tree[rt] = tree[rt << 1] + tree[rt << 1 | 1];
36 }
37 void build(int l, int r, int rt) {
38     if (l == r) {
39         tree[rt] = 1;
40         return;
41     }
42     int m = (l + r) >> 1;
43     build(lson);
44     build(rson);
45     pushup(rt);
46 }
47 
48 int query(int x, int l, int r, int rt) {
49     tree[rt]--;
50     if (l == r) return l;    
51     int m = (l + r) >> 1;
52     if (x <= tree[rt << 1]) return query(x, lson);
53     else return query(x - tree[rt << 1], rson);
54 }
55 
56 int main() {
57     int n;
58     cin >> n;
59     build(1, n, 1);
60     rep(i, 1, n) cin >> a[i];
61     per(i, 0, n) a[i] = query(a[i] + 1, 1, n, 1);
62     rep(i, 0, n) cout << a[i] << endl;
63     return 0;
64 }