AtCoderD - Kth Excluded 差分+思维

Posted 2021-11-22 karshey

题
参考

用s来记录差分前缀和情况：
如：
a 1 3 5 8 9 10 14
b 0 1 1 2 0 0 3
s 0 1 2 4 4 4 7
即，b[2]代表a[2]-a[1]-1,即a[2]到a[1]有一个数字可以放。
s[2]表示从头到a[2]有s[2]个数字可以放。

也就是说，如果我们想知道第k个数字是几，我们要知道第k-1个数字在哪个a的范围内，然后再加上差值。
如，我们想知道第3个数字在哪，我们找到下标temp=2，此时a[2]=5,且由s数组我们知道从头a[1]到a[2]可以放s[2]=2个数字，我们直接让5-2=3，相当于把可以放的数字填平，于是相当于3前面没有数字可以放，3就是要求的第0个最小的数。
所求就是3+k；

//参考：https://blog.csdn.net/skl4869/article/details/117886656 
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define fir(i,a,n) for(int i=a;i<=n;i++)
//======================
const int N=1e5+10;
ll a[N],b[N],s[N];
int n,q;
int main()
{
	cin>>n>>q;
	fir(i,1,n) scanf("%lld",&a[i]);
	sort(a+1,a+1+n);
	fir(i,1,n)
	{
		b[i]=a[i]-a[i-1]-1;
	}
	fir(i,1,n)
	{
		s[i]=b[i]+s[i-1];
	}
	while(q--)
	{
		ll t;scanf("%lld",&t);
		int temp=lower_bound(s,s+1+n,t)-s;
		cout<<t+a[temp-1]-s[temp-1]<<endl;
	}
	return 0; 
}