Codeforces Round #708 (Div. 2)B. M-arrays

最新推荐文章于 2021-08-15 19:53:27 发布

龙小茶

最新推荐文章于 2021-08-15 19:53:27 发布

阅读量123

点赞数

文章标签：排序算法数据结构

本文链接：https://blog.csdn.net/LONGXC6/article/details/115048133

版权

原题

原题链接
在这里插入图片描述

题意

给一个数组，将其分为多个子集合，且子集和中为恰能被m整除或相邻两数字相加能被整除

思路

取余
对除以m的余数为0的分到一个集合
将余数和为m的两个数分到一个集合，且排序为x，m-x，x，m-x，x

AC代码

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

int main() {
    int t;
    cin >> t;
    while (t--) {
        int n, m;
        cin >> n >> m;
        map<int, int> cnt;
        while (n--) {
            int x;
            cin >> x;
            cnt[x % m]++;
        }
        int ans = 0;
        for (auto &c : cnt) {
            if (c.first == 0) ans++;
            else if (2 * c.first == m) {
                ans++;
            } else if (2 * c.first < m || cnt.find(m — c.first) == cnt.end()) {
                int x = c.second, y = cnt[m — c.first];
                ans += 1 + max(0, abs(x — y) — 1);
            }
        }
        cout << ans << '\n';
    }
    return 0;
}