AtCoder Beginner Contest 205 D - Kth Excluded

[ 只 有 D 题 ]

[ 传 送 门 ]

问题:

给你一个数组 有序的那种

然后给你q次询问 问每次询问的k

即在正整数集合中 第K小的且未出现的数字是什么

思路

先分析 数据范围 很大 需要O(n) 或者 nlogn 的级别

差分求出 每个区间的有多少数 然后？求前缀和 ？ 然后还得标记回去 ？ 不会Emmmm ， 卡死了

正解

我们可以通过用 b[i] = a[i] - i 来表示差分数组 因为是有序的数组

对于每个如果不在序列上的 我们就可以这么求出来有多少个未在序列中(只能说牛逼 我差分错了)

然后进行二分 最后下标即可以返回去通过a数组来求得答案

code:

#include <iostream>
#include <algorithm>
#include <unordered_map>
#define scl(a) scanf("%lld",&a)
#define ptl(a) printf("%lld\n",a)
using namespace std;
typedef long long ll;
const int N  = 1e5+10;
ll a[N],q,b[N],k;
int n;
unordered_map<ll,ll> mp;
void solve()
{
    scanf("%lld%lld",&n,&q);

    for(int i = 1; i<=n; i++)
        scl(a[i]);

    a[n+1] = 1e18+n+1;

    for(int  i =1; i<=n+1; i++) b[i] = a[i]-i; ///表示前面有几个空位置

    while(q -- )
    {
        scl(k);
        int  l = 0,r = n+1;
        while( l < r-1)
        {
            ll m = (l+r)>>1;
            if(k>b[m]) l = m ;
            else r=  m;
        }
        
        k -=(b[r-1]);
        ptl(a[r-1]+k);
    }
}
int main()
{
    solve();
    return 0;
}