結果
問題 | No.1331 Moving Penguin |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
#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;}