結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー chocono2230
提出日時 2021-01-08 23:59:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 479 ms / 1,500 ms
コード長 1,350 bytes
コンパイル時間 5,417 ms
コンパイル使用メモリ 253,844 KB
最終ジャッジ日時 2025-01-17 14:46:54
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(ri,n) for(int ri = (int)(n-1); ri >= 0; ri--)
#define rep2(i,x,n) for(int i = (int)(x); i < (int)(n); i++)
#define rrep2(ri,x,n) for(int ri = (int)(n-1); ri >= (int)(x); ri--)
#define repit(itr,x) for(auto itr = x.begin(); itr != x.end(); itr++)
#define rrepit(ritr,x) for(auto ritr = x.rbegin(); ritr != x.rend(); ritr++)
#define ALL(x) x.begin(), x.end()
using ll = long long;
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint1000000007;

const int sz = 500;

int main(){
  int n;
  cin >> n;
  vector<int> a(n);
  rep(i, n) cin >> a.at(i);
  auto v = vector(sz, vector(sz, mint(0)));
  vector<mint> dp(n+1, 0);
  dp.front() = 1;
  rep2(i, 1, n+1){
    if(i == 1 || a.at(i-2) != 1) dp.at(i) += dp.at(i-1);
    rep2(j, 1, sz){
      dp.at(i) += v.at(j).at(i%j);
    }
    int na = a.at(i-1);
    if(na < sz){
      v.at(na).at(i%na) += dp.at(i);
    }else{
      for(int j = i + na; j < dp.size(); j += na){
        dp.at(j) += dp.at(i);
      }
    }
  }
  // for(auto i : dp){
  //   cerr << i.val() << " ";
  // }
  // cerr << endl;
  // rep(i, 10){
  //   rep(j, 10){
  //     cerr << v.at(i).at(j).val() << " ";
  //   }
  //   cerr << endl;
  // }
  // cerr << endl;
  cout << dp.back().val() << endl;
  return 0;
}
0