結果

問題 No.681 Fractal Gravity Glue
ユーザー nebukuro09nebukuro09
提出日時 2018-04-28 00:52:17
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 1 ms / 2,000 ms
コード長 1,975 bytes
コンパイル時間 691 ms
コンパイル使用メモリ 115,000 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-13 00:57:18
合計ジャッジ時間 1,339 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import std.stdio, std.array, std.string, std.conv, std.algorithm;
import std.typecons, std.range, std.random, std.math, std.container;
import std.numeric, std.bigint, core.bitop, std.bitmanip;
immutable long MOD = 10^^9 + 7;
void main() {
auto N = readln.chomp.to!long;
auto s = readln.split.map!(to!long);
auto n = s[0];
auto k = s[1];
long ans = G(n, k);
long dfs(long n, long k, long rest) {
if (rest == 0) return 0;
if (n == 1) return rest;
long cnt = 0;
long ret = 0;
foreach (i; 0..k*2+1) {
if (i % 2 == 0) {
long tmp = 1;
foreach (_; 0..n-1) {
tmp *= k+1;
if (tmp - 1 > rest) return dfs(n-1, k, rest);
}
tmp -= 1;
if (cnt + tmp >= rest) return (ret + dfs(n-1, k, rest - cnt)) % MOD;
cnt += tmp;
ret = (ret + G(n-1, k)) % MOD;
} else {
if (cnt + 1 == rest) return (ret + n) % MOD;
ret = (ret + n) % MOD;
cnt += 1;
}
}
return ret;
}
n = 10000;
outer: while (true) {
if (n < 5) break;
long tmp = 1;
foreach (_; 0..n) {
tmp *= k+1;
if (tmp > N) {
n -= 1;
continue outer;
}
}
if (tmp <= N) {
n += 10;
break;
}
}
ans = (ans - dfs(n, k, N)) % MOD;
ans = (ans + MOD) % MOD;
ans.writeln;
}
long G(long n, long k) {
long ans;
ans = powmod(k+1, n+1, MOD);
ans = (ans - k - 1) % MOD;
ans = ans * powmod(k, MOD-2, MOD) % MOD;
ans = (ans - n) % MOD;
ans = (ans + MOD) % MOD;
return ans;
}
long powmod(long a, long x, long m) {
long ret = 1;
while (x) {
if (x % 2) ret = ret * a % m;
a = a * a % m;
x /= 2;
}
return ret;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0