二分——圣诞节糖果
题目描述
圣诞节临近,彩虹岛的黑心商人 𝑐𝑡𝑟 的糖果店又开始热闹了起来,热心的 𝑠𝑙𝑝 来到 𝑐𝑡𝑟 的店里面帮忙包装糖果。店里面共有 𝑛 堆糖果,其中第 𝑖 堆有 𝑎𝑖 颗糖果,𝑐𝑡𝑟 让 𝑠𝑙𝑝 从中选择两堆糖果,这两堆糖果中每 𝑝 颗包装在一起,如果最后还有剩余就归 𝑠𝑙𝑝 所有了,若两堆不足 𝑝 个则全部归 𝑠𝑙𝑝 所有。作为糖果狂热爱好者,𝑠𝑙𝑝 当然是想拿走尽量多的糖果,因此他想知道自己最多能够拿走多少糖果。
输入描述
输入第一行为一个整数 𝑇(1 ≤ 𝑇 ≤ 10),表示一共有 𝑇 组测试数据。
对于每组测试数据:
第一行有两个整数 𝑛(2 ≤ 𝑛 ≤ 105), 𝑝(1 ≤ 𝑝 ≤ 109),分别表示糖果堆数和包装后每包糖果的数量。
第二行有 𝑛 个整数,其中第 𝑖 个数 𝑎𝑖(1 ≤ 𝑎𝑖 ≤ 109)表示第 𝑖 堆糖果的数量。
输出描述
对于每组测试数据,输出一个整数 𝑥 表示 𝑠𝑙𝑝 能拿走的最多的糖果数目。
示例
输入
2
4 5
1 4 2 3
4 15
12 19 13 20
输出
4
10
说明
对于第一组样例,𝑠𝑙𝑝 选择第一堆和第四堆是最佳选择,会剩余 4 颗糖果。
对于第二组样例,𝑠𝑙𝑝 选择第一堆和第三堆是最佳选择,会剩余 10 颗糖果。
分析
这题可以先将输入的每袋糖果数对 p 取余,即为每袋糖果装袋后剩余的,并将每袋糖果的剩余糖果数从小到大排序,此时 slp 能拿走的糖果数分为两种情况:
- 最后两袋糖果装袋后剩余的糖果数
- 两袋糖果数相加小于 p 的糖果数
因此,我们只要找到上述两种情况的最大值即可,可以先将第一种情况赋给 mx,然后通过二分查找满足第二种情况的糖果数,将其与 mx 进行比较,如果大于 mx,则将该种情况的糖果数赋给 mx,最后将 mx 输出即可。
#include<bits/stdc++.h>
using namespace std;
int t,n,p;
long long a[100005];
int main()
{
cin>>t;
while(t--){
cin>>n>>p;
for(int i=0;i<n;i++){
cin>>a[i];
a[i]=a[i]%p; // 每袋糖果装袋后剩余的
}
sort(a,a+n); // 将每袋糖果的剩余糖果数从小到大排序
long long mx=(a[n-2]+a[n-1])%p; // 最后两袋糖果装袋后剩余的
int l=0,r=n-1;
while(l<r){
while(a[l]+a[r]>=p){ // 如果剩余的糖果数量大于 p,则将 r 减 1
r--;
}
if(l<r){
mx=max(mx,a[l]+a[r]);
l++;
}
}
cout<<mx<<endl;
}
return 0;
}