#684664 (D) No.1631 Sorting Integers (Multiple of K) Easy

提出ソース

結果

問題	No.1631 Sorting Integers (Multiple of K) Easy
ユーザー	daut-dlang
提出日時	2021-07-30 22:29:34
言語	D (dmd 2.106.1)
結果	AC
実行時間	2,348 ms / 3,000 ms
コード長	2,945 bytes
コンパイル時間	768 ms
コンパイル使用メモリ	121,708 KB
実行使用メモリ	244,180 KB
最終ジャッジ日時	2024-06-22 12:02:23
合計ジャッジ時間	16,267 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 6,812 KB
testcase_01	AC	1 ms 6,944 KB
testcase_02	AC	1 ms 6,940 KB
testcase_03	AC	8 ms 6,944 KB
testcase_04	AC	0 ms 6,940 KB
testcase_05	AC	1 ms 6,940 KB
testcase_06	AC	1 ms 6,940 KB
testcase_07	AC	1 ms 6,944 KB
testcase_08	AC	1 ms 6,940 KB
testcase_09	AC	3 ms 6,940 KB
testcase_10	AC	2 ms 6,944 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	1 ms 6,940 KB
testcase_13	AC	1 ms 6,940 KB
testcase_14	AC	1,037 ms 120,060 KB
testcase_15	AC	2,348 ms 240,784 KB
testcase_16	AC	2,300 ms 243,892 KB
testcase_17	AC	2,299 ms 240,764 KB
testcase_18	AC	2,314 ms 244,180 KB
testcase_19	AC	2,333 ms 240,764 KB
testcase_20	AC	4 ms 6,944 KB
testcase_21	AC	6 ms 6,940 KB
testcase_22	AC	8 ms 6,940 KB
testcase_23	AC	42 ms 14,752 KB
testcase_24	AC	16 ms 9,704 KB
testcase_25	AC	12 ms 8,164 KB
testcase_26	AC	4 ms 6,944 KB
testcase_27	AC	680 ms 74,464 KB
testcase_28	AC	315 ms 38,088 KB
testcase_29	AC	85 ms 22,428 KB
testcase_30	AC	351 ms 46,932 KB
testcase_31	AC	465 ms 70,324 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

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[] rotate(double[] vec, double rad) { return [cos(rad)*vec[0] - sin(rad)*vec[1], sin(rad)*vec[0] + cos(rad)*vec[1]]; }

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 K = RD;
	auto c = RDA;

	long toBit(long[] arr, long pos = -1)
	{
		long res;
		foreach (i; 0..9)
		{
			if (i == pos)
				res |= (arr[i]-1) << (i*4);
			else
				res |= arr[i] << (i*4);
		}
		return res;
	}
	long[] toArr(long bit)
	{
		auto res = new long[](9);
		foreach (i; 0..9)
			res[i] = (bit >> (i*4)) & 0b1111;
		return res;
	}

	auto dp = new long[long][][](N+1, K);
	dp[0][0][toBit(c)] = 1;
	long unit = 1;
	foreach (i; 0..N)
	{
		foreach (j; 0..K)
		{
			foreach (key; dp[i][j].keys)
			{
				auto arr = toArr(key);
				//debug writeln("i:", i, " j:", j, " arr:", arr);
				foreach (k; 0..9)
				{
					if (arr[k] == 0) continue;
					auto nj = (j + (k+1) * unit) % K;
					dp[i+1][nj][toBit(arr, k)] += dp[i][j][key];
					//debug writeln("i:", i, " j:", j, " k:", k, " bit:", toBit(arr,k), " arr:", toArr(toBit(arr,k)));
				}
			}
		}
		unit = (unit * 10) % K;
	}

	writeln(dp[N][0].get(0, 0));
	stdout.flush;
	debug readln;
}

yukicoder

結果

テストケース

ソースコード