题目描述
给你一个长度为 N 的正整数序列 A=(A1,A2,…,An),接下来你会收到 M次询问,每次询问给你一个正整数 K,你需要回答:在 1∼K 之间,未出现在序列 A 中的整数之和。
输入
第一行2个正整数N和M;
第二行N个正整数,为序列A;
接下来M行,每行一个正整数,其中第i个数表示第i次询问中对应的K值。
输出
输出M行,每行一个整数,依次表示M次询问的答案。
样例
输入
4 1
1 6 3 1
5
输出
11
输入
8 4
123 1234 666 249794 666 2627034 24979445 930302156
10000
158260522
1000000000
10
输出
50002977
12523196463128207
499999999541839548
55
提示
参考代码
#include<bits/stdc++.h>
using namespace std;
long long N, M, K;
long long a[100005];
long long b[100005];
long long s[100005];
long long ans;
int main(){
cin >> N >> M;
for(int i=1;i<=N;i++){
cin >> a[i];
}
sort(a+1, a+N+1);
int p=1;
for(int i=1;i<=N;i++){
if(a[i]==a[i-1]){
continue;
}
b[p++]=a[i];
}
N=p-1;
for(int i=1;i<=N;i++){
s[i]=s[i-1]+b[i];
}
long long t=0;
for(int i=1;i<=M;i++){
cin >> K;
t=K*(1+K)/2;
int l=1, r=N, mid;
while(l<=r){
mid=(l+r)/2;
if(b[mid]>K){
r=mid-1;
}
else{
l=mid+1;
}
}
t-=s[r];
cout << t << endl;
}
return 0;
}