POJ 2182 Lost Cows.(线段树）

最新推荐文章于 2022-09-03 21:45:05 发布

原创最新推荐文章于 2022-09-03 21:45:05 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#Lost Cows #POJ2182 #线段树 #N 2 N 8 #000 cows have unique

ACM_线段树&&树状数组专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用线段树解决特定排序问题的方法。该问题要求根据每头牛前面有多少头编号较小的牛来确定它们的原始编号顺序。文章提供了两种实现方式：一种是简单的暴力解法，另一种则是利用线段树进行高效查询和更新。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

~~~~

数据太水，暴力无压力，不过还是用线段树写了下，表现了楼主对线段树的无限热爱之情。

~~~~

题目连接：http://poj.org/problem?id=2182

大致题意;牛牛因XXXXXX原因乱序成一排，现已知每头牛前面有多少头牛比它的编号小（第一头牛前面没有当然就不列粗来了），求每头牛的编号。

思路：从后往前扫描，遇到a,则说明它是剩余序列的第a+1头牛。

~~~~

暴力代码：（157ms）

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 8888
using namespace std;

int q[N],a[N],ans[N];
int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        q[0]=0;a[0]=1;
        for(int i=1;i<n;i++)
        {
            scanf("%d",&q[i]);
            a[i]=i+1;
        }
        for(int i=n-1;i>=0;i--)
        {
            ans[i]=a[q[i]];
            for(int j=q[i];j<n-1;j++)
                a[j]=a[j+1];
        }
        for(int i=0;i<n;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}

~~~~

线段树代码：（32ms）

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 8888
#define lson rt<<1,s,m
#define rson rt<<1|1,m+1,e
using namespace std;

int tre[N<<2];
void build(int rt,int s,int e)
{
    tre[rt]=e-s+1;  //用tre数组存每条线段间有多少个牛
    if(s==e)
        return ;
    int m=(s+e)>>1;
    build(lson);
    build(rson);
}
int query(int val,int rt,int s,int e)
{
    tre[rt]--;  //每次query，牛数必减1，因为下一次询问总是从剩余的序列中。
    if(s==e)
        return s;   //~~
    int m=(s+e)>>1;
    if(val<=tre[rt<<1]) return query(val,lson);
    else return query(val-tre[rt<<1],rson);
}

int q[N],ans[N];
int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        build(1,1,n);
        q[0]=0;     //~~
        for(int i=1;i<n;i++)
            scanf("%d",&q[i]);
        for(int i=n-1;i>=0;i--)
            ans[i]=query(q[i]+1,1,1,n);
        for(int i=0;i<n;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}