AtCoder AGC032E Modulo Pairing (二分、贪心结论)

题目链接

https://atcoder.jp/contests/agc032/tasks/agc032_e

题解

猜结论好题。
结论是: 按\(a_i\)从小到大排序之后，一定存在一种最优解，使得以某个位置为界，两边分别首尾匹配，且满足左边的每一对的和都\(<M\), 右边每一对的和都\(\ge M\).
证明不难，可参考官方题解，此处不再赘述。
然后显然可以枚举这个临界点，然后\(O(n)\)暴力计算答案，时间复杂度\(O(n^2)\).
考虑优化: 在我们配对的时候，当临界点右移，左右两侧的每一对和都会变大。
于是我们只需找到最小的合法临界点，即是最优解。
二分找即可，时间复杂度\(O(n\log n)\).

代码

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cassert>
#include<algorithm>
using namespace std;

const int N = 2e5;
int a[N+3];
int n,m;

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n+n; i++) scanf("%d",&a[i]);
    sort(a+1,a+n+n+1);
    int left = 0,right = n;
    while(left<right)
    {
        int mid = left+((right-left)>>1);
        bool ok = true;
        for(int i=mid*2+1; i<=n+n; i++)
        {
            if(a[i]+a[n+n+mid+mid+1-i]<m) {ok = false; break;}
        }
        if(ok==true) {right = mid;}
        else {left = mid+1;}
    }
    int pos = 2*right;
    int ans = 0;
    for(int i=1; i<=pos; i++) ans = max(ans,a[i]+a[pos+1-i]);
    for(int i=pos+1; i<=n+n; i++) ans = max(ans,(a[i]+a[n+n+pos+1-i])-m);
    printf("%d\n",ans);
    return 0;
}