結果

問題 No.1629 Sorting Integers (SUM of M)
コンテスト
ユーザー daut-dlang
提出日時 2021-07-30 21:13:50
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 220 ms / 2,000 ms
コード長 3,551 bytes
コンパイル時間 775 ms
コンパイル使用メモリ 122,164 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-22 11:58:42
合計ジャッジ時間 2,633 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

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[] 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); }

struct Cmb
{
	void init(size_t n)
	{
		long powmod(long a, long p)
		{
			long ans = 1;
			long mul = a;
			while (p > 0)
			{
				if ((p & 1) == 1)
					ans.modm(mul);
				p >>= 1;
				mul.modm(mul);
			}
			return ans;
		}

		++n;
		fact = new long[2][](n);
		fact[0][0] = 1;
		foreach (i; 1..n)
		{
			fact[i][0] = fact[i-1][0];
			fact[i][0].modm(i);
		}
		fact[n-1][1] = powmod(fact[n-1][0], mod - 2);
		foreach_reverse (i; 0..n-1)
		{
			fact[i][1] = fact[i+1][1];
			fact[i][1].modm(i+1);
		}
	}

	long get(size_t n, size_t r)
	{
		long res = fact[n][0];
		res.modm(fact[r][1]);
		res.modm(fact[n-r][1]);
		return res;
	}

	long[2][] fact;
}

void main()
{
	auto N = RD;
	auto c = RDA;

	Cmb cmb;
	cmb.init(N);
	
	auto pat1 = new long[](9);
	auto pat2 = new long[](9);
	auto rem = N;
	long tot = 1;
	foreach (i; 0..9)
	{
		if (c[i] == 0) continue;
		pat1[i] = cmb.get(rem, c[i]);
		pat2[i] = cmb.get(N-1, c[i]-1);
		tot.modm(pat1[i]);
		rem -= c[i];
	}
	auto pat3 = new long[](9);
	foreach (i; 0..9)
	{
		if (c[i] == 0) continue;
		pat3[i] = tot;
		pat3[i].modd(cmb.get(N, c[i]));
	}
	debug writeln("pat1:", pat1);
	debug writeln("pat2:", pat2);
	debug writeln("pat3:", pat3);

	long ans;
	foreach (i; 0..N)
	{
		long unit = 10;
		unit.modpow(i);
		debug writeln("unit:", unit);
		foreach (j; 0..9)
		{
			if (c[j] == 0) continue;
			long tmp = j+1;
			tmp.modm(unit);
			tmp.modm(pat3[j]);
			tmp.modm(pat2[j]);
			ans.moda(tmp);
			debug writeln("i:", i, " j:", j, " pat3:", pat3[i], " pat2:", pat2[i], " tmp:", tmp);
		}
	}

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