第五届上海市青少年算法竞赛 T5 最大的余数（折半查找）

第五题：T5最大的余数

标签：折半查找
题意：给定$n$个数字$a_1,a_2,\dots ,a_n$，在给定一个整数$m$，请从给定的数字中挑选任意多个数字，使得它们的和模$m$的余数尽量大，输出这个最大的余数。$(1\le n\le 40，1\le a_i\le 10^9，1\le m\le 10^9)$
题解：一看到$n\le 40$，下意识就是折半查找，因为暴力搜索$2^{40}\approx 10^{12}$，肯定会超时。把问题折半一下，分别预处理出各自一半的数的组成情况，时间复杂度是$2\ast 2^{20}\approx 2\ast 10^6$。
比如样例：$110100100010000$
前两个的所有情况之和为$011011$
后三个的所有情况之和为$01001000100001100101001100011100$
前半部分搜索出来的所有情况的值存到$b$数组中，后半部分搜索出来的所以情况存到$c$数组中。
要找出前半部分$+$后半部分对$m$进行取模余数尽可能大的。那我们可以枚举前半部分的所有情况，然后二分查找（$lower\_bound$）一下后半部分中$m-b[i]$的值，找到的是第一个大于等于$m-b[i]$的，所以往前减一位即可。
代码：

#include <bits/stdc++.h>
using namespace std;

int n, m, a[50], ans = 0;
vector<int> b, c;
void dfs1(int p, int sum) {
    if (p == n / 2 + 1) {
        b.push_back(sum);
        return ;
    }
    dfs1(p + 1, (sum + a[p]) % m);
    dfs1(p + 1, sum % m);
}

void dfs2(int p, int sum) {
    if (p == n + 1) {
        c.push_back(sum);
        return ;
    }
    dfs2(p + 1, (sum + a[p]) % m);
    dfs2(p + 1, sum % m);
}

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        a[i] %= m;
    }
    dfs1(1, 0);
    dfs2(n / 2 + 1, 0);
    sort(c.begin(), c.end());
    for (int i = 0; i < b.size(); i++) {
        int p = lower_bound(c.begin(), c.end(), m - b[i]) - c.begin();
        p--;
        ans = max(ans, (b[i] + c[p]) % m);
    }
    cout << ans << endl;
    return 0;
}