#771072 (C++14) No.1972 Modulo Set

• 問題: No.1972 Modulo Set
• ユーザー: startcpp
• 提出日時: 2022-06-14 11:42:09
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 220 ms / 2,000 ms
• コード長: 986 bytes
• コンパイル時間: 1,193 ms
• コンパイル使用メモリ: 81,092 KB
• 実行使用メモリ: 10,368 KB
• 最終ジャッジ日時: 2024-04-15 09:00:16
• 合計ジャッジ時間: 5,508 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	18 ms; 5,376 KB
testcase_04	AC	27 ms; 5,376 KB
testcase_05	AC	29 ms; 5,376 KB
testcase_06	AC	54 ms; 5,376 KB
testcase_07	AC	75 ms; 5,376 KB
testcase_08	AC	33 ms; 5,376 KB
testcase_09	AC	6 ms; 5,376 KB
testcase_10	AC	25 ms; 5,376 KB
testcase_11	AC	75 ms; 5,376 KB
testcase_12	AC	13 ms; 5,376 KB
testcase_13	AC	8 ms; 5,376 KB
testcase_14	AC	9 ms; 5,376 KB
testcase_15	AC	112 ms; 6,272 KB
testcase_16	AC	126 ms; 6,912 KB
testcase_17	AC	32 ms; 5,376 KB
testcase_18	AC	3 ms; 5,376 KB
testcase_19	AC	97 ms; 6,528 KB
testcase_20	AC	119 ms; 6,912 KB
testcase_21	AC	41 ms; 5,376 KB
testcase_22	AC	87 ms; 6,144 KB
testcase_23	AC	166 ms; 9,344 KB
testcase_24	AC	59 ms; 5,760 KB
testcase_25	AC	101 ms; 7,040 KB
testcase_26	AC	161 ms; 6,784 KB
testcase_27	AC	200 ms; 9,600 KB
testcase_28	AC	66 ms; 5,888 KB
testcase_29	AC	106 ms; 6,944 KB
testcase_30	AC	141 ms; 8,192 KB
testcase_31	AC	7 ms; 5,376 KB
testcase_32	AC	122 ms; 7,936 KB
testcase_33	AC	213 ms; 10,368 KB
testcase_34	AC	198 ms; 10,240 KB
testcase_35	AC	218 ms; 10,368 KB
testcase_36	AC	220 ms; 10,368 KB

ソースコード

//各aについて、mod M = a, M - aのうち片方しか取れない
//0, M / 2 は それぞれ1個まで。
#include <iostream>
#include <algorithm>
#include <map>
#define int long long
using namespace std;

int gcd(int a, int b) {
	if (b == 0) return a;
	return gcd(b, a % b);
}

int n, m;
int a[100000];
map<int, int> mp;

signed main() {
	int i;

	cin >> n >> m;
	for (i = 0; i < n; i++) {
		cin >> a[i];
		a[i] %= m;
		if (mp.find(a[i]) == mp.end()) mp[a[i]] = 0;
		mp[a[i]]++;
	}

	for (map<int, int>::iterator it = mp.begin(); it != mp.end(); it++) {
		int mod = it->first;
		if (mod == 0 || mod * 2 == m) {
			mp[mod] = min(mp[mod], 1LL);
		}
	}

	int pair_cnt = 0, single_cnt = 0;
	for (map<int, int>::iterator it = mp.begin(); it != mp.end(); it++) {
		int mod = it->first;
		if (mod * 2 != m && mp.find(m - mod) != mp.end()) {
			pair_cnt += max(mp[mod], mp[m - mod]);
		}
		else {
			single_cnt += mp[mod];
		}
	}

	cout << pair_cnt / 2 + single_cnt << endl;
	return 0;
}