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;
}

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值