中南林业科技大学第十一届ACM校赛J题

中南林业科技大学第十一届ACM校赛

J.are you ok?

Description

一个长度为n的数组a，数组下标从0开始。现在要求你查询从左到右第一个不小于k的数字a[i], 输出i，并且马上把a[i-1]++；如果你找到的a[i]中的i等于0，那么a[0-1]是非法的，因此只要输出i就行了，不进行a[i-1]++；如果你在数组中找不到一个数字不小于k，则输出”are you ok ”

Input

多组输入，输入直到遇到EOF为止；

第一行输入两个整数n和q，表示数组a中有n个整数，q表示q次查询；

第二行输入n个整数；

第三行到后2+q行，每行输入一个数字k，表示要求你查询从左到右第一个不小于k的数字并马上输出。

注意：1 < n, q <= 1e6, a[i]和k是一个int型的整数

Output

见输入。

Sample Input

3 4
1 2 8
2
2
8
9

Sample Output

1
0
2
are you ok

题意

  中文题。

解题思路

  读题，很容易想到维护一个数组a，a[i]存的是[ 1 , i ]区间内的最大值，然后再利用二分查找下界，找到下标。但是这个题还有一点，就是会更新原数组，原数组改变了，b数组可能会跟着改变。显然用线段树就很容易实现这两个操作，线段树的每个节点存的是区间[ l , r ]的最大值，这样在线段树上进行二分就能解决之。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;

int arr[maxn<<2];
int n,q;

void pushup(int rt)
{
    arr[rt]=max(arr[rt<<1],arr[rt<<1|1]);
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%d",&arr[rt]);
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
int query(int c,int l,int r,int rt)
{
    if(l==r) return arr[rt]>=c?l:l+1;
    int mid=(l+r)>>1;
    if(arr[rt<<1] >= c)
        return query(c,l,mid,rt<<1);
    else
        return query(c,mid+1,r,rt<<1|1);
}
void update(int pos,int l,int r,int rt)
{
    if(l==r)
    {
        arr[rt]++;
        return;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) update(pos,l,mid,rt<<1);
    else update(pos,mid+1,r,rt<<1|1);
    pushup(rt);
}
int main()
{
//    freopen("in.txt","r",stdin);
    int n,q;
    while(~scanf("%d%d",&n,&q))
    {
        build(1,n,1);
        while(q--)
        {
            int c;
            scanf("%d",&c);
            int cur=query(c,1,n,1);
            if(cur==n+1)
            {
                puts("are you ok");
                continue;
            }
            if(cur!=1) update(cur-1,1,n,1);
            printf("%d\n",cur-1);
        }
    }
    return 0;
}

  悄咪咪的在后台加了组非常硬的数据，然后很皮的rejudge了一下，把一个学长的代码卡超时了，皮这一下我很快乐~