C++求和（算法编程题解）

题目描述
给你一个长度为 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;
}