#966735 (D) No.2709 1975 Powers

• 問題: No.2709 1975 Powers
• ユーザー: k82b
• 提出日時: 2024-03-31 13:59:35
• 言語: D (dmd 2.106.1)
• 結果: AC
• 実行時間: 119 ms / 2,000 ms
• コード長: 1,337 bytes
• コンパイル時間: 5,093 ms
• コンパイル使用メモリ: 212,864 KB
• 実行使用メモリ: 34,688 KB
• 最終ジャッジ日時: 2024-09-30 18:46:07
• 合計ジャッジ時間: 5,757 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	99 ms; 34,176 KB
testcase_01	AC	103 ms; 34,048 KB
testcase_02	AC	101 ms; 34,560 KB
testcase_03	AC	106 ms; 34,560 KB
testcase_04	AC	114 ms; 34,560 KB
testcase_05	AC	119 ms; 34,688 KB
testcase_06	AC	107 ms; 34,304 KB
testcase_07	AC	103 ms; 34,688 KB
testcase_08	AC	106 ms; 34,432 KB
testcase_09	AC	108 ms; 34,688 KB
testcase_10	AC	104 ms; 34,688 KB
testcase_11	AC	99 ms; 34,304 KB
testcase_12	AC	100 ms; 34,560 KB
testcase_13	AC	110 ms; 34,560 KB
testcase_14	AC	105 ms; 34,432 KB
testcase_15	AC	103 ms; 34,688 KB
testcase_16	AC	108 ms; 34,432 KB
testcase_17	AC	102 ms; 34,432 KB
testcase_18	AC	101 ms; 34,560 KB
testcase_19	AC	111 ms; 34,560 KB
testcase_20	AC	102 ms; 34,560 KB
testcase_21	AC	103 ms; 34,560 KB
testcase_22	AC	109 ms; 34,560 KB
testcase_23	AC	111 ms; 34,688 KB
testcase_24	AC	111 ms; 34,560 KB
testcase_25	AC	111 ms; 34,432 KB
testcase_26	AC	109 ms; 34,688 KB

ソースコード

import std, core.bitop;
bool chmin(T)(ref T SE, T MI) { if (SE > MI) { SE = MI; return true; } else { return false; } }
bool chmax(T)(ref T SE, T MI) { if (SE < MI) { SE = MI; return true; } else { return false; } }
int lowerBound(T)(T[] SE, T MI) { int lo = -1, hi = cast(int)(SE.length); while (hi - lo > 1) { int mid = lo + hi >> 1; (SE[mid] < MI ? lo : hi) = mid; } return hi; }
int upperBound(T)(T[] SE, T MI) { int lo = -1, hi = cast(int)(SE.length); while (hi - lo > 1) { int mid = lo + hi >> 1; (SE[mid] > MI ? hi : lo) = mid; } return hi; }
void main() {
	int N, P, Q;
	readf("%s %s %s\n", N, P, Q);
	auto A = readln.chomp.split.to!(int[]);
	A.sort;
	enum semi = 2020202;
	auto p10 = new int[semi];
	p10[0] = 1;
	foreach (i; 1 .. semi) {
		p10[i] = p10[i - 1] * 10 % P;
	}
	auto p9 = new int[semi];
	p9[0] = 1;
	foreach (i; 1 .. semi) {
		p9[i] = p9[i - 1] * 9 % P;
	}
	auto p7 = new int[semi];
	p7[0] = 1;
	foreach (i; 1 .. semi) {
		p7[i] = p7[i - 1] * 7 % P;
	}
	auto p5 = new int[semi];
	p5[0] = 1;
	foreach (i; 1 .. semi) {
		p5[i] = p5[i - 1] * 5 % P;
	}
	auto cnt = new int[P];
	foreach (i; 0 .. N) {
		++cnt[p5[A[i]]];
	}
	int ans;
	foreach (i; 0 .. N) {
		--cnt[p5[A[i]]];
		foreach (j; 0 .. i) {
			foreach (k; 0 .. j) {
				ans += cnt[(Q - (p10[A[k]] + p9[A[j]] + p7[A[i]]) % P + P) % P];
			}
		}
	}
	ans.writeln;
}