poj 2182 -Lost Cows -线段树-还原逆序数组

题意:给出一个数组每个元素的逆序数,还原这个数组(序号为1-n)



思路:还是从逆序开始处理,对于每个位置i,如果逆序数是a[i],并且我们不考虑i之后的位置,则第i个位置的序号本应该是a[i]+1,但是如果i之后的位置还存在k个小于等于a[i]+1的数,则位置i需要改为 a[i]+1+k。

因此,我们从后面往前处理,每次处理只需要看之前的操作(对应操作编号i到n)是否确定了小于等于原本应该填的数(a[i]+1), 如果是则i位置应填的数X增加一 ,变为a[i]+1+1...以此类推

线段树的节点维护当前区间已被确定的节点数,我们只需要判断 sum[左区间]+a[i]+1<= mid,便可知道进左区间还是右了,



#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;
#define ptf(ar1,ar2)  printf("ar1:%d\n",ar2);
const long long maxn = 8005;
struct tree
{
    int sum[maxn*4];
    void init()
    {
        memset(sum,0,sizeof(sum));
    }
    void pushup(int rt)
    {
        sum[rt]= (sum[rt<<1]+sum[rt<<1|1]);
    }
    void update(int l,int r,int num,int rt,int val)
    {
        if (l==r&&l==num )
        {
            sum[rt]=val;
            return ;
        }
        int mid=(l+r)/2;
        if (num<=mid)
            update(l,mid,num,rt<<1,val);
        else
            update(mid+1,r,num,rt<<1|1,val);

        pushup(rt);
    }
    int query(int num,int l,int r,int rt) //查询大于等于val的第一个位置
    {
        if (l==r)
        {
            sum[rt]=1;
            return l;
        }
        int mid=(l+r)>>1;
        int ret=-1;
        if (sum[rt<<1]+num<=mid)    //应该插入的位置为sum[rt<<1]+num
                ret=query(num,l,mid,rt<<1);
        else
               ret=query(num+sum[rt<<1],mid+1,r,rt<<1|1);//如果左区间不满足,则应该插入sum[rt]+num
        pushup(rt);
        return ret;
    }
};
tree tp;
struct node
{
    int l,r,id;
    int num;
};
int tm[8005];
int ans[8005];
int main(  )
{
    int i;
    int x;
    int n,h,w;
    int l,r;
        int cun=0;
     scanf("%d",&n);
        for (i=2; i<=n; i++)
        {
            scanf("%d",&tm[i]);
        }
        for (i=n;i>=2;i--)
        {
            int ret=tp.query( tm[i]+1,1,n,1);
        //ptf(1,2);
            ans[i]=ret;
        }
            int sum=0;
            for (i=2; i<=n; i++)
            if (ans[i]) sum+=ans[i];
            ans[1]=n*(n+1)/2-sum;

         for (i=1; i<=n; i++)
        {
            printf("%d\n",ans[i]);
        }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值