結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー spihill
提出日時 2021-01-09 00:19:47
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 190 ms / 1,500 ms
コード長 643 bytes
コンパイル時間 1,787 ms
コンパイル使用メモリ 196,344 KB
最終ジャッジ日時 2025-01-17 14:49:13
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

constexpr int mod = 1e9 + 7;
constexpr int S = 320;

int main() {
    int N; cin >> N;
    vector<int> A(N), dp1(N);
    for (auto& x : A) cin >> x;
    dp1[0] = 1;
    vector<vector<int>> dp2(S);
    for (int i = 0; i < S; i++) dp2[i] = vector<int>(i);
    for (int i = 0; i < N; i++) {
        for (int j = 1; j < S; j++) (dp1[i] += dp2[j][i % j]) %= mod; 
        if (A[i] < S) (dp2[A[i]][i % A[i]] += dp1[i]) %= mod;
        else for (int j = i + A[i]; j < N; j += A[i]) (dp1[j] += dp1[i]) %= mod;
		if (A[i] != 1 && i != N-1) (dp1[i+1] += dp1[i]) %= mod;
    }
    cout << dp1[N-1] << endl;
}
0