[蓝桥杯]倍数问题

标题：倍数问题

【题目描述】
众所周知，小葱同学擅长计算，尤其擅长计算一个数是否是另外一个数的倍数。但小葱只擅长两个数的情况，当有很多个数之后就会比较苦恼。现在小葱给了你 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。

求解思路：
核心思路是要求三个变量和为k的倍数，循环遍历两个变量，第三个按照倍数等约束条件解出，然后查找。
将输入的n个数num分成k组，分组的依据是num%k的余数，范围是0~k-1。
因为k不超过1000，所以k可以考虑用数组，每个数组表示模k余数相同的数。因为每个数组的元素长度不定，所以数组的每个元素是vector容器(int类型)，将输入的数存起来。
对vector里每个数组中的每个数从大到小进行排序。
第一重循环：i
数组为空的不用管,continue
对于第一组，当这些数组内数据至少有3个及以上时，把max更新为这个数组的前三个之和。这是因为第一组是余数为0的 一组，都是k的倍数，所以任意三个数之和都满足是k的倍数。排序后取前三个为最大。
第二重循环：j
数组为空的不用管，continue
★根据前两重循环变量计算第三个变量：z = k - (i+j)%k 计算思路：既然要找三个数是k的倍数，前两个数模k求余得i和j，i+j再模k得还差多少可以整除k。 k-(i+j)%k得z模k的值。
找到z后对z进行判断：
1.z为负，舍去。（理论上不会，不过写上防止出错）
2.z，i，j三个数都相等时，这种情况在第一重循环中考虑过了，直接去这个数组的前三个就可。
3.如果i和j不相等，但z和i，j中的一个相等。即只有两个数组用上，一个数组取两个数，另一个数组取一个数。
这是如果z数组中至少有两个数，另一个至少有一个（肯定至少有一个，因为为0的情况已经continue了）数的情况下，
max取值为i数组的第一个数，j数组的第一个数和z数组的第二个数。（这里比较巧妙，虽然知道了z和i，j其中一个相等，但i，j数组的第一个肯定用上，z的第二个也肯定用上。虽然z和i，j一个相等，但这里不用区别开。将两种情况合并。）
4.如果i和j相等，且i数组至少有两个数，z不和i，j相等，且z至少有一个数的情况下，max取值为i数组的第一个数，j数组的第二个数和z数组的第一个数。（i，j顺序可变）
5.如果i，j和z三个互不相等，且各自都至少有一个数（其实i和j肯定都至少有一个，因为之前为空的都判断过了，这里主要是对z数组的判断）的情况下，max取值为i数组的第一个数，j数组的第一个数和z数组的第一个数。

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

const int MAX_K = 1e3+100;

bool compare(int a, int b)
{
    return (a > b);
}

int main()
{

    int n, k, i, j,z, num, Max;
    cin >> n >> k;
    vector<int> p[MAX_K];//the type of array is vector
    for(i = 0; i < n; i++){
        cin >> num;
        p[num%k].push_back(num);
    }
    for(i = 0; i < k; i++){
        sort(p[i].begin(), p[i].end(), compare);
    }
    Max = 0;
    for(i = 0; i < k; i++){
        if(!p[i].size())
            continue;
        if(i == 0 && p[i].size() >= 3)
            Max = max(Max, p[i][0] + p[i][1] + p[i][2]);
        for(j = 0; j < k; j++){
            if(!p[j].size())
                continue;
            z = k - (i+j)%k;
            if(z == i && i == j)
                continue;
            else if( (i == j && z != i ) && p[i].size()>=2 && p[z].size()>=1 )
                Max = max(Max, p[i][0] + p[j][1] + p[z][0]);
            else if( (i != j && (z == i || z == j ) ) && p[z].size()>=2 )
                Max = max(Max, p[i][0] + p[j][0] + p[z][1]);
            else if((i != j && z != i && z != j) && p[z].size()>=1 )
                Max = max(Max, p[i][0] + p[j][0] + p[z][0]);
        }
    }
    cout  << Max << endl;
    return 0;
}