POJ 2182 Lost Cows （简单题，线段树）

POJ 2182 Lost Cows （线段树（查询））

题目链接： POJ 2182 Lost Cows
题目描述： 有编号1~n的n个数字，2 <= n <= 8000, 乱序排列，顺序是未知的。对于每个位置的数字，知道排在前面比它小的数字有多少个，求这个乱序数列的顺序。
题解思路： 首先用暴力的方法来求解这道题
我们知道pre【】：0 1 2 1 0 和 ans【】：
基本的思路就是从后向前来看这个pre数组，从后向前看，每次只要找到现存有的数字的第pre【i】+ 1的数字就是答案。 暴力解法，每次找到一个答案后要重新排名然后查询，排名的同时可以查询一起做，所以时间复杂度是O(n)有n个这样的操作，所以时间复杂度是O(n * n)。

这道题的数据规模是8000，所以暴力解法也是可以解决的。

接下来我们要考虑的是更高效的解法，使用线段树可以把重新查询的时间复杂度降到O(logn)，使用结构体的线段树每个节点有三个属性，它表示的是一个区间，所以三个属性是左边界， 右边界， 和区间长度。所以我们可以构建一颗这样的线段树，查询只要从根节点向下查找就可以了。因为建树的过程每一次都是把父节点的中间值作为左右子树的边界，所以可以看出线段树的基本思想其实是二分。

代码如下：
暴力解法：

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;

#define ll long long
#define sf scanf
#define pf printf

const int N = 8005;
int pre[N], ans[N], num[N];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n;  cin >> n;
    for(int i = 1; i <= n; i++) num[i] = i;
    pre[1] = 0;
    for(int i = 2; i <= n; i++) cin >> pre[i];
    for(int i = n; i >= 1; i--){
        int k = 0;
        for(int j = 1; j <= n; j++){
            if(num[j] != -1){
                k++;
                if(k == pre[i] + 1){
                    ans[i] = num[j];
                    num[j] = -1;
                    break;
                }
            }
        }
    }

    for(int i = 1; i <= n; i++) cout << ans[i] <<endl;
    return 0;
}

结构体线段树：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;

#define sf scanf
#define pf printf
#define ll long long
const int N = 8005;
int pre[N];
int ans[N];
struct node{
    int l, r, len;
}tree[N * 4];

void build(int left, int right, int u)
{
    tree[u].l = left;
    tree[u].r = right;
    tree[u].len = right - left + 1;
    if(left == right) return;
    build(left, (left + right) >> 1, u << 1);
    build(((left +right) >> 1) + 1, right, (u << 1) + 1);
}
int query(int u, int num)
{
    tree[u].len--;
    if(tree[u].l == tree[u].r) return tree[u].l;
    if(tree[u << 1].len < num) return query(((u << 1) + 1), num - tree[u << 1].len);
    if(tree[u << 1].len >= num) return query(u << 1, num);
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n; cin >> n;
    for(int i = 2; i <= n; i++) cin >> pre[i];
    pre[1] = 0;
    build(1, n, 1);
    for(int i = n; i >= 1; i--) ans[i] = query(1, pre[i] + 1);
    for(int i = 1; i <= n; i++) cout << ans[i] << endl;
    return 0;
}

完全二叉树线段树：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;

#define ll long long
#define sf scanf
#define pf printf

const int N = 8005;
int pre[N];
int ans[N];
int tree[4 * N];

void build(int n, int lleft)
{
    memset(tree, 0, sizeof(tree));
    for(int i = lleft; i < lleft + n; i++) tree[i] = 1;
    while(lleft != 1){
        for(int i = lleft >> 1; i < lleft; i++) tree[i] = tree[i << 1] + tree[(i << 1) + 1];
        lleft = lleft >> 1;
    }
}

int query(int u, int num, int lleft)
{
    tree[u]--;
    if(tree[u] == 0 && u >= lleft) return u;
    if(tree[u << 1] >= num) return query(u << 1, num, lleft);
    if(tree[u << 1] < num) return query((u << 1) + 1, num - tree[u << 1], lleft);
}
int main()
{
    int n; cin >> n;
    for(int i = 2; i <= n; i++) cin >> pre[i];
    pre[1] = 0;
    int lleft = 1 << (int(log(n)/log(2)) + 1);
    build(n, lleft);
    for(int i = n; i >= 1; i--) ans[i] = query(1, pre[i] + 1, lleft) - lleft + 1;
    for(int i = 1; i <= n; i++) cout << ans[i] << endl;
    return 0;
}