#474844 (D) No.1043 直列大学

提出ソース

結果

問題	No.1043 直列大学
ユーザー	daut-dlang
提出日時	2020-05-01 22:55:44
言語	D (dmd 2.107.1)
結果	AC
実行時間	98 ms / 2,000 ms
コード長	3,202 bytes
コンパイル時間	727 ms
コンパイル使用メモリ	106,752 KB
実行使用メモリ	6,460 KB
最終ジャッジ日時	2023-09-04 07:37:37
合計ジャッジ時間	3,316 ms
ジャッジサーバーID （参考情報）	judge14 / judge13

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	7 ms 4,888 KB
testcase_01	AC	7 ms 6,460 KB
testcase_02	AC	8 ms 5,936 KB
testcase_03	AC	6 ms 4,828 KB
testcase_04	AC	7 ms 4,884 KB
testcase_05	AC	7 ms 5,640 KB
testcase_06	AC	7 ms 5,636 KB
testcase_07	AC	7 ms 5,420 KB
testcase_08	AC	7 ms 5,400 KB
testcase_09	AC	20 ms 5,168 KB
testcase_10	AC	23 ms 5,472 KB
testcase_11	AC	30 ms 4,904 KB
testcase_12	AC	98 ms 5,148 KB
testcase_13	AC	70 ms 4,824 KB
testcase_14	AC	73 ms 4,896 KB
testcase_15	AC	85 ms 4,912 KB
testcase_16	AC	73 ms 5,928 KB
testcase_17	AC	46 ms 4,852 KB
testcase_18	AC	45 ms 5,400 KB
testcase_19	AC	69 ms 5,144 KB
testcase_20	AC	44 ms 5,108 KB
testcase_21	AC	63 ms 5,144 KB
testcase_22	AC	66 ms 5,884 KB
testcase_23	AC	90 ms 6,168 KB
testcase_24	AC	52 ms 4,880 KB
testcase_25	AC	69 ms 5,412 KB
testcase_26	AC	75 ms 6,160 KB
testcase_27	AC	29 ms 5,928 KB
testcase_28	AC	55 ms 5,680 KB
testcase_29	AC	16 ms 5,736 KB
testcase_30	AC	66 ms 5,736 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)); }

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

T binarySearch(alias pred, T)(T ok, T ng)
{ 
	while (abs(ok-ng) > 1)
	{
		auto mid = (ok+ng)/2;
		if (unaryFun!pred(mid)) ok = mid;
		else ng = mid;
	}
	return ok;
}

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

	auto dp1 = new long[](10^^5+1);
	dp1[0] = 1;
	foreach (i; 0..N)
	{
		foreach_reverse (j; 0..dp1.length)
		{
			if (dp1[j] == 0) continue;
			dp1[j+V[i]].moda(dp1[j]);
		}
	}
	/*auto arr1 = new long[](dp1.length+1);
	foreach (i; 0..dp1.length)
	{
		arr1[i+1] = arr1[i] + dp1[i];
	}*/

	auto dp2 = new long[](10^^5+1);
	dp2[0] = 1;
	foreach (i; 0..M)
	{
		foreach_reverse (j; 0..dp2.length)
		{
			if (dp2[j] == 0) continue;
			dp2[j+R[i]].moda(dp2[j]);
		}
	}
	dp2[0] = 0;
	auto arr2 = new long[](dp2.length+1);
	foreach (i; 0..dp2.length)
	{
		auto tmp = arr2[i];
		tmp.moda(dp2[i]);
		arr2[i+1].moda(tmp);
	}

	long ans;
	foreach (i; 1..dp1.length)
	{
		if (dp1[i] == 0) continue;

		bool f(long x)
		{
			return i > B * x;
		}
		bool g(long x)
		{
			return i >= A * x;
		}

		auto r = binarySearch!(f)(0, cast(long)dp2.length);
		auto l = binarySearch!(g)(0, cast(long)dp2.length);
		debug writeln("l:", l, " r:", r);
		auto tmp = dp1[i];
		auto tmp2 = arr2[l+1];
		tmp2.mods(arr2[r+1]);
		tmp.modm(tmp2);
		ans.moda(tmp);
		debug writeln("ans:", ans);
	}

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

yukicoder

結果

テストケース

ソースコード