题目链接:D - ±1 Operation 2
题意:给定n个数和q个目标值,每次可以对n个数进行加一或减一操作,输出q个最小操作数(最小操作数为使n个数达到目标值的最小操作次数)(数据范围1≤N,Q≤2×10^5)
如:
5 3
6 11 2 5 5
5
20
0
则 输出 :
10
71
29
分析:先对数组进行排序,然后二分查找目标值大于或小于q的分界位置,最后前后两区间分别计算出总和。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
ll n, q;
cin >> n >> q;
ll a[n+1],b[q+1],sum[n+1]= {0};
for(int i = 1; i <= n; i++) cin >> a[i];
sort(a + 1, a + n + 1);//从1至n+1排序
for(int i = 1; i <= n; i++)
sum[i] = sum[i - 1] + a[i];//记录1~i区间和
ll pos;
for(int i = 1; i <= q; i++)
{
cin >> b[i];
pos = lower_bound(a + 1, a + n + 1, b[i]) - a,pos--;//二分查找不小于目标值的第一个元素
cout << pos * b[i] - sum[pos] + (sum[n] - sum[pos]) - ((n - pos) * b[i]) << endl;//小于x的数字的最小操作数加大于x的最小操作数
}
return 0;
}