【蓝桥杯】倍数问题

倍数问题

众所周知，小葱同学擅长计算，尤其擅长计算一个数是否是另外一个数的倍数。但小葱只擅长两个数的情况，当有很多个数之后就会比较苦恼。现在小葱给了你 n 个数，希望你从这 n 个数中找到三个数，使得这三个数的和是 K 的倍数，且这个和最大。数据保证一定有解。
输入格式
　　从标准输入读入数据。

第一行包括 2 个正整数 n, K。
　　第二行 n 个正整数，代表给定的 n 个数。
输出格式
　　输出到标准输出。
　　输出一行一个整数代表所求的和。
样例入

4 3
1 2 3 4

样例输出

9

样例说明
　　选择2、3、4。
数据约定
　　对于 30% 的数据，$n<=100$。
　　对于 60% 的数据，$n<=1000$。
　　对于另外 20% 的数据，$K<=10$。
　　对于 100% 的数据，$1<=n<=10^5,1<=K<=10^3$，给定的 n 个数均不超过 $10^8$。

思路

• 暴力+优化
• 纯粹的三重循环肯定是过不了的，所以需要优化，不过还是暴力的板子
• 优化暴力法的几个方面：减少循环层数；二分；缩小搜索范围；做一些预处理，以空间换时间。
• 这道题中可以先做一些预处理，用空间换时间，然后以此来减少循环次数，做到优化，其实从后往前看的话，这道题在做一方面优化的同时，也一起做了其他方面的优化，比如还缩小了搜索范围。
• 这道题是求三个数的和是某个数的倍数，而不是等于那个数，那就要用到取模方面的知识了
• 根据题意，我们要找 $a,b,c$ 使得：$$(a+b+c)modk=0$$
• 而由模运算的知识可知，上面的式子等价于：$$(amodk+bmodk+cmodk)modk=0$$
• 这样便可以减少一重循环：利用 $a$ 和 $b$ 将 $c$ 确定下来，就可以从三重循环降到二重循环
• 做一个预处理，把输入的 $n$ 个数按照余数分类，余数为0的放在一个集合中，为1的放在一个集合中……，这个过程可以用一个二维数组实现，而且还可以加一个限制条件，即我们只需要3个数的和，所以每个余数的集合里面维护三个数即可，并且要这些和最大，那么就维护最大的三个数
• 令 $r_1=amodk$；$r_2=bmodk$；$r_3=cmodk$，如果 $r_1+r_2+r_3$ 是 $k$ 的倍数的话，那么就是符合题意的
• 我们知道：$r_1,r_2,r_3<k$，由此便可以求出 $r_3$ 来：$$r_3=(k-r_1+k-r_2)modk$$
• 我们需要 $r_1+r_2+r_3$ 是 $k$ 的倍数，其中 $r_1、r_2$ 是可以算出的，而且 $k$ 是已知的，要求的就只有 $r_3$，由于 $r_1,r_2<k$，那么就可以先求出 $k-r_1,k-r_2$ 的值，看看还差多少凑够 $k$ 的两倍(至于为什么是两倍，因为两倍是"最近"的一个 $k$ 的倍数)，那么就让 $r_3=k-r_1+k-r_2$，而且要考虑 $r_3$ 的值是否大于等于 $k$，但不管是否大于等于，只要再对 $k$ 取个模即可，即 $r_3=(k-r_1+k-r_2)modk$
• 找到了 $r_1,r_2,r_3$ 之后，事情还没有结束，我们要找的是 $a,b,c$ 而不是它们模 $k$ 之后的余数，所以还要进一步求 $a,b,c$
• 我们做了预处理之后，会得到一个关于余数的二维数组，我们需要对这个二维数组进行二重循环，找到 $r_1,r_2$，并确定 $r_3$，找到之后，做以下分类：
  • 如果 $r_1=r_2=r_3$，说明 $a,b,c$ 同余，则答案更新为余数为 $r_1$ 的那三个数(直接访问预处理数组即可)
  • 如果 $r_1=r_2\ne r_3$，说明 $a,b$ 同余，则答案更新为余数为 $r_1$ 的前两个数与余数为 $r_3$ 的第一个数之和
  • 如果 $r_1=r_3\ne r_2$，说明 $a,c$ 同余，则答案更新为余数为 $r_1$ 的前两个数与余数为 $r_2$ 的第一个数之和
  • 如果 $r_2=r_3\ne r_1$，说明 $c,b$ 同余，则答案更新为余数为 $r_2$ 的前两个数与余数为 $r_1$ 的第一个数之和
  • 如果 $r_1\ne r_2\ne r_3$，说明 $a,b,c$ 两两不同余，则答案更新为余数为 $r_1$ 的第一个数与余数为 $r_2$ 的第一个数与余数为 $r_3$ 的第一个数之和
• 然后取最大值，输出即可
• 这样就只对 $k$ 规模的数组做了二重循环，由题可知 $k<10^3$，所以方案是可行的，预处理时维护前三大的数可以用朴素的“插入排序”即可，由于只有3个数进行排序，是 $O(1)$ 复杂度，所以预处理的总复杂度是 $O(n)$ 的，所以最后的复杂度是 $O(k^2)$，而 $k<10^3$，所以不会超时

代码如下

C++用多了，复习一下C语言

#include <stdio.h>

int max(int a, int b) { return a > b ? a : b; }

int n, k;
int a[1001][4];

void solve() {
    scanf("%d%d", &n, &k);
    int i = 0, j = 0, t = 0, d = 0;
    while (n--) {
        scanf("%d", &t);
        d = t % k;
        //一个朴素的插入排序
        for (i = 1; i <= 3; i++) {
            if (t > a[d][i]) {
                for (j = 3; j > i; j--) {
                    a[d][j] = a[d][j - 1];
                }
                a[d][i] = t;
                a[d][0]++;  //利用第一个位置存数的个数
                break;
            }
        }
    }

    int ans = -1;  //最终答案
    int tmp = 0;   //中间答案

    for (i = 0; i < k; i++) {
        for (j = 0; j < k; j++) {
            d = (2 * k - i - j) % k;
            //五种情况，注意如果没有那么多个数的话就不要更新，直接跳过，以免出错
            if (i == j && j == d && a[i][0] >= 3) {
                tmp = a[i][1] + a[i][2] + a[i][3];
            } else if (i != j && i != d && j != d && a[i][0] >= 1 &&
                       a[j][0] >= 1 && a[d][0] >= 1) {
                tmp = a[i][1] + a[j][1] + a[d][1];
            } else if (i == j && i != d && a[i][0] >= 2 && a[d][0] >= 1) {
                tmp = a[i][1] + a[i][2] + a[d][1];
            } else if (i == d && i != j && a[i][0] >= 2 && a[j][0] >= 1) {
                tmp = a[i][1] + a[i][2] + a[j][1];
            } else if (j == d && i != j && a[j][0] >= 2 && a[i][0] >= 1) {
                tmp = a[j][1] + a[j][2] + a[i][1];
            }
            ans = max(ans, tmp);
        }
    }
    printf("%d\n", ans);
}

int main(void) {
    solve();
    return 0;
}