智商问题

题目链接

复制代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
inline ll read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

/********************************************************************/

const int maxn = 1e6+5;
int a[maxn];
int last[maxn];
int n, num;

int ask(int x){
    for(int i = 1;i <= num+1;i++){
        if(x <= last[i] && x > last[i-1]){
            int b = (i-1)*num;
            for(int j = 1;j <= num;j++){
                if(a[b+j] >= x){
                    return b+j;
                }
            }
        }
    }
}

int main(){
    n = read();
    for(int i = 1;i <= n;i++)
        a[i] = read();
    sort(a+1, a+1+n);
    num = (int)sqrt(n);
    for(int i = 1;i <= num;i++)
        last[i] = a[i*num];
    last[num+1] = a[n];
    int x;
    while(scanf("%d", &x) != EOF){
        if(x < a[1])
            printf("1\n");
        else if(x > a[n])
            printf("%d\n", n+1);
        else
            printf("%d\n", ask(x));
    }

    return 0;
}