倍数问题_数论

P1366 - [蓝桥杯2018初赛]倍数问题 - New Online Judgehttp://oj.ecustacm.cn/problem.php?id=1366

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 100004;
int a[N] = {0};
vector<int>s[1024];
int main() {
    cin.tie(0);
    int n,K;
    cin >> n >> K;
    for(int i=0; i<n; i++) {
        cin >> a[i];
    }
    sort(a, a+n);
    for(int i=0; i<n; i++) {
        s[a[i]%K].push_back(a[i]);
    }
    int ans = 0;
    for(int i=0; i<K; i++) {
        if (s[i].empty()) {
            continue;
        }
        for(int j=0; j<K; j++) {
            if (s[j].empty()) {
                continue;
            }
            int t = (K - i + K - j)%K;
            if (s[t].empty()) {
                continue;
            }
            int a = s[i].back(), b = s[j].back(), c = s[t].back();
            if(j==i) {
                if (s[i].size()>1) {
                    b = s[i][s[i].size()-2];
                }
                if(t==i) {
                    if (s[i].size()>2) {
                        c = s[i][s[i].size()-3];
                    }
                }
            } else if(t==i) {
                if (s[i].size()>1) {
                    c = s[i][s[i].size()-2];
                }
            } else if(j==t) {
                if (s[j].size()>1) {
                    c = s[j][s[j].size()-2];
                }
            }
            ans = max(ans, a+b+c);
        }
    }
    cout << ans;
    return 0;
}