結果

問題 No.533 Mysterious Stairs
ユーザー Akihiro Shoji
提出日時 2017-06-25 16:08:08
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 215 ms / 5,000 ms
コード長 612 bytes
コンパイル時間 888 ms
コンパイル使用メモリ 98,140 KB
実行使用メモリ 106,368 KB
最終ジャッジ日時 2024-06-12 20:30:35
合計ジャッジ時間 3,348 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm,
       std.string,
       std.range,
       std.stdio,
       std.conv;

ulong INF = ulong.max;
ulong M = 10L^^9 + 7;
ulong[4][1000001] dp;
ulong N;

ulong solve(ulong i, ulong prev) {
  if (dp[i][prev] != INF) { return dp[i][prev]; }

  if (i == N) { return 1; }
  int ret;

  foreach (d; 3.iota) {
    d++;
    if (prev != d && i + d <= N) {
      ret += solve(i + d, d);
      ret %= M;
    }
  }

  return dp[i][prev] = ret;
}

void main() {
  foreach (i; 1000001.iota) {
    foreach (j; 4.iota) {
      dp[i][j] = INF;
    }
  }

  N = readln.chomp.to!ulong;
  writeln(solve(0, 0));
}
0