結果

問題 No.1972 Modulo Set
ユーザー daut-dlangdaut-dlang
提出日時 2022-06-10 21:59:01
言語 D
(dmd 2.106.1)
結果
WA  
実行時間 -
コード長 3,175 bytes
コンパイル時間 867 ms
コンパイル使用メモリ 113,964 KB
実行使用メモリ 14,880 KB
最終ジャッジ日時 2023-09-04 17:23:30
合計ジャッジ時間 4,268 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 AC 4 ms
4,384 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 AC 6 ms
4,380 KB
testcase_14 AC 5 ms
4,380 KB
testcase_15 WA -
testcase_16 AC 73 ms
10,796 KB
testcase_17 AC 27 ms
6,836 KB
testcase_18 AC 1 ms
4,376 KB
testcase_19 AC 60 ms
12,348 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 AC 50 ms
11,348 KB
testcase_23 AC 84 ms
11,232 KB
testcase_24 AC 35 ms
10,256 KB
testcase_25 AC 51 ms
11,724 KB
testcase_26 WA -
testcase_27 AC 88 ms
11,180 KB
testcase_28 AC 34 ms
11,256 KB
testcase_29 AC 50 ms
11,528 KB
testcase_30 AC 65 ms
12,800 KB
testcase_31 AC 5 ms
4,380 KB
testcase_32 AC 61 ms
12,740 KB
testcase_33 AC 111 ms
12,164 KB
testcase_34 AC 112 ms
13,756 KB
testcase_35 AC 114 ms
14,880 KB
testcase_36 AC 108 ms
13,936 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio, std.conv, std.functional, std.string;
import std.algorithm, std.array, std.container, std.range, std.typecons;
import std.bigint, std.numeric, std.math, std.random;
import core.bitop;

string FMT_F = "%.10f";

static File _f;
void file_io(string fn) { _f = File(fn, "r"); }
static string[] s_rd;
T _RD(T = long)() { while(!s_rd.length) s_rd = readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T _RD(T = long)(File f) { while(!s_rd.length) s_rd = f.readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T[] _RDA(T = long)(T fix = 0) { auto r = readln.chomp.split.to!(T[]); r[] += fix; return r; }
T[] _RDA(T = long)(File f, T fix = 0) { auto r = f.readln.chomp.split.to!(T[]); r[] += fix; return r; }
T RD(T = long)() { if (_f.isOpen) return _RD!T(_f); else return _RD!T; }
T[] RDA(T = long)(T fix = 0) { if (_f.isOpen)  return _RDA!T(_f, fix); else return _RDA!T(fix); }

size_t[] MAKE_IDX(alias less = "a < b", Range)(Range range) { auto idx = new size_t[](range.length); makeIndex!(less)(range, idx); return idx;}
size_t MIN_POS(alias less = "a < b", Range)(Range range) { auto r = minPos!(less)(range); return range.length - r.length; }

void chmin(T)(ref T x, T y) { x = min(x, y); } void chmax(T)(ref T x, T y) { x = max(x, y); }
bool inside(T)(T x, T b, T e) { return x >= b && x < e; }
T lcm(T)(T x, T y) { return x * (y / gcd(x, y)); }
double euDist(T)(T[] a, T[] b) { auto c = a.dup; c[] -= b[]; c[] *= c[]; return sqrt(cast(double)c.sum); }
double[] rotate(double[] vec, double rad) { return [cos(rad)*vec[0] - sin(rad)*vec[1], sin(rad)*vec[0] + cos(rad)*vec[1]]; }
double norm(double[] vec) { return sqrt(reduce!((a,b)=>a+b*b)(0.0, vec)); }
double dotProd(double[] a, double[] b) { auto r = a.dup; r[] *= b[]; return r.sum; }

//long mod = 10^^9 + 7;
long mod = 998_244_353;
//long mod = 1_000_003;
void moda(ref long x, long y) { x = (x + y) % mod; }
void mods(ref long x, long y) { x = ((x + mod) - (y % mod)) % mod; }
void modm(ref long x, long y) { x = (x * y) % mod; }
void modpow(ref long x, long y) { if (!y) { x = 1; return; } auto t = x; x.modpow(y>>1); x.modm(x); if (y&1) x.modm(t); }
void modd(ref long x, long y) { y.modpow(mod - 2); x.modm(y); }

void main()
{
	auto N = RD;
	auto M = RD;
	auto A = RDA;

	long[][long] list;
	foreach (i; 0..N)
		list[A[i] % M] ~= i;

	long ans;
	foreach (key; list.keys.sort)
	{
		if (key*2 == M)
		{
			long cnt;
			foreach (i; list[M/2])
			{
				if (A[i] <= M/2)
					++cnt;
			}
			ans += max(1, cnt);
		}
		else
		{
			auto a = list.get(M-key, []);
			if (a.empty)
				ans += list[key].length;
			else if (key*2 > M) continue;
			else
			{
				auto b = list[key];
				long[] arr1, arr2;
				foreach (i; a)
					arr1 ~= A[i];
				foreach (i; b)
					arr2 ~= A[i];

				arr1.sort;
				arr2.sort;
				debug writeln(arr1);
				debug writeln(arr2);
				long cnt = arr2.length;
				long j;
				foreach_reverse (i; 0..arr1.length)
				{
					while (j < arr2.length)
					{
						if (arr1[i]+arr2[j] < M)
							++j;
						else
							break;
					}
					cnt.chmax(i+1+j);
				}
				ans += cnt;
			}
		}
	}

	writeln(ans);
	stdout.flush;
	debug readln;
}
0