結果
問題 | No.1331 Moving Penguin |
ユーザー |
![]() |
提出日時 | 2021-01-08 22:19:22 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 181 ms / 1,500 ms |
コード長 | 3,878 bytes |
コンパイル時間 | 2,025 ms |
コンパイル使用メモリ | 197,516 KB |
最終ジャッジ日時 | 2025-01-17 12:34:33 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 49 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:152:16: warning: format ‘%lld’ expects argument of type ‘long long int’, but argument 2 has type ‘modint<1000000007>::i64’ {aka ‘long int’} [-Wformat=] 152 | printf("%lld\n", dp[n].a); | ~~~^ ~~~~~~~ | | | | | modint<1000000007>::i64 {aka long int} | long long int | %ld main.cpp:129:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 129 | scanf("%d", &n); | ~~~~~^~~~~~~~~~ main.cpp:131:27: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 131 | For(i, 1, n + 1) scanf("%d", &a[i]); | ~~~~~^~~~~~~~~~~~~
ソースコード
#include <bits/stdc++.h> //#include <atcoder/all> #define For(i, a, b) for (int(i) = (int)(a); (i) < (int)(b); ++(i)) #define rFor(i, a, b) for (int(i) = (int)(a)-1; (i) >= (int)(b); --(i)) #define rep(i, n) For((i), 0, (n)) #define rrep(i, n) rFor((i), (n), 0) #define fi first #define se second using namespace std; typedef long long lint; typedef unsigned long long ulint; typedef pair<int, int> pii; typedef pair<lint, lint> pll; template <class T> bool chmax(T &a, const T &b) { if (a < b) { a = b; return true; } return false; } template <class T> bool chmin(T &a, const T &b) { if (a > b) { a = b; return true; } return false; } template <class T> T div_floor(T a, T b) { if (b < 0) a *= -1, b *= -1; return a >= 0 ? a / b : (a + 1) / b - 1; } template <class T> T div_ceil(T a, T b) { if (b < 0) a *= -1, b *= -1; return a > 0 ? (a - 1) / b + 1 : a / b; } constexpr lint mod = 1000000007; constexpr lint INF = mod * mod; constexpr int MAX = 200010; template <int_fast64_t MOD> struct modint { using i64 = int_fast64_t; i64 a; modint(const i64 a_ = 0) : a(a_) { if (a > MOD) a %= MOD; else if (a < 0) (a %= MOD) += MOD; } modint inv() { i64 t = 1, n = MOD - 2, x = a; while (n) { if (n & 1) (t *= x) %= MOD; (x *= x) %= MOD; n >>= 1; } modint ret(t); return ret; } bool operator==(const modint x) const { return a == x.a; } bool operator!=(const modint x) const { return a != x.a; } modint operator+(const modint x) const { return modint(*this) += x; } modint operator-(const modint x) const { return modint(*this) -= x; } modint operator*(const modint x) const { return modint(*this) *= x; } modint operator/(const modint x) const { return modint(*this) /= x; } modint operator^(const lint x) const { return modint(*this) ^= x; } modint &operator+=(const modint &x) { a += x.a; if (a >= MOD) a -= MOD; return *this; } modint &operator-=(const modint &x) { a -= x.a; if (a < 0) a += MOD; return *this; } modint &operator*=(const modint &x) { (a *= x.a) %= MOD; return *this; } modint &operator/=(modint x) { (a *= x.inv().a) %= MOD; return *this; } modint &operator^=(lint n) { i64 ret = 1; while (n) { if (n & 1) (ret *= a) %= MOD; (a *= a) %= MOD; n >>= 1; } a = ret; return *this; } modint operator-() const { return modint(0) - *this; } modint &operator++() { return *this += 1; } modint &operator--() { return *this -= 1; } bool operator<(const modint x) const { return a < x.a; } }; using mint = modint<1000000007>; vector<mint> fact; vector<mint> revfact; void setfact(int n) { fact.resize(n + 1); revfact.resize(n + 1); fact[0] = 1; rep(i, n) fact[i + 1] = fact[i] * mint(i + 1); revfact[n] = fact[n].inv(); for (int i = n - 1; i >= 0; i--) revfact[i] = revfact[i + 1] * mint(i + 1); } mint getC(int n, int r) { if (n < r) return 0; return fact[n] * revfact[r] * revfact[n - r]; } int main() { int n; scanf("%d", &n); int a[n + 1]; For(i, 1, n + 1) scanf("%d", &a[i]); int b = 400; mint dp[n + 1], sum[b][b]; rep(i, n + 1) dp[i] = 0; dp[1] = 1; rep(i, b) rep(j, b) sum[i][j] = 0; For(i, 1, n) { if (a[i] < b) { sum[a[i]][i % a[i]] += dp[i]; } else { for (int j = i + a[i]; j <= n; j += a[i]) { dp[j] += dp[i]; } } if (a[i] != 1) dp[i + 1] += dp[i]; For(k, 1, b) dp[i + 1] += sum[k][(i + 1) % k]; } printf("%lld\n", dp[n].a); }