zoj3635-Cinema in Akiba 树状数组+二分

     n个人去看电影,正好有n个座位,每个人手中的票上的号码为a[i],表示他进场的时候,需要做到当前从1---n中第a[i]个空位,给出m个查询,分别求出查询的人实际的座位位置。

     用树状数组记录一下每个位置当前是否有人,那么对于一个位置i,就可以求出1---i当中有多少个空位,而i位置也就是当前第i-sum(i)个空位了,由于这个i-sum(i)是单调不减的,所以可以二分位置来求出当前的人应该坐到那个位置上,二分的时候注意一下细节,有一个特殊的情况,就是如果位置i上有人,而查询得到i-sum(i)恰好为求得答案时,此时二分区间应该是在左区间而不是右区间。复杂度n*log(n)*log(n)。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn=66000;
int n,m;
int k,x,y;
int ans[maxn];

struct BIT
{
    int dt[maxn];
    void init()
    {
        memset(dt,0,sizeof dt);
    }
    int lowbit(int x)
    {
        return x&(-x);
    }
    void add(int x,int c=1)
    {
        for (int i=x; i<=n; i+=lowbit(i))
        dt[i]+=c;
    }
    int sum(int x)
    {
        int res=0;
        for (int i=x; i>0; i-=lowbit(i))
        res+=dt[i];
        return res;
    }
}bit;

int check(int k,int x)
{
    int tp=bit.sum(k);
    return k-bit.sum(k);
}
bool flag[maxn];
int find(int x)
{
    int l=0,r=n+1;
    int res=0;
    int mid;
    while(l<r)
    {
        mid=(l+r)>>1;
        int ct=check(mid,x);
        if (ct==x)
        {
            if (!flag[mid]) return mid;
            else
            {
                r=mid;
            }
        }
        else if (ct<x) l=mid+1;
        else r=mid;
    }
    return res;
}
int main()
{
//    freopen("in.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        bit.init();
        memset(flag,false,sizeof flag);
        for (int i=1; i<=n; i++)
        {
            scanf("%d",&k);
            ans[i]=find(k);
            bit.add(ans[i]);
            flag[ans[i]]=true;
        }
        scanf("%d",&m);
        scanf("%d",&k);
        cout<<ans[k];
        for (int i=1; i<m; i++)
        {
            scanf("%d",&k);
            cout<<" "<<ans[k];
        }
        cout<<endl;
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值