結果

問題 No.770 Median Sequence
ユーザー Pachicobue
提出日時 2018-12-19 16:11:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 657 ms / 2,000 ms
コード長 1,161 bytes
コンパイル時間 916 ms
コンパイル使用メモリ 79,084 KB
最終ジャッジ日時 2025-01-06 19:39:57
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using ll = long long;
constexpr ll MOD = 1000000007;
template <typename T>
std::vector<T> Vec(const std::size_t n, T v) { return std::vector<T>(n, v); }
template <class... Args>
auto Vec(const std::size_t n, Args... args) { return std::vector<decltype(Vec(args...))>(n, Vec(args...)); }
int main()
{
    int N;
    std::cin >> N;
    auto dp = Vec(2, N + 1, N + 1, 0LL);  // 現在の坂成分で下がったか;Aiまで決定;Ai = h;
    auto sum = Vec(2, N + 1, N + 1, 0LL);
    dp[0][0][0] = 1LL;
    std::fill(sum[0][0].begin(), sum[0][0].end(), 1LL);
    for (int i = 1; i <= N; i++) {
        for (int j = 1; j <= N; j++) {
            (dp[0][i][j] += sum[0][i - 1][j] + sum[1][i - 1][j - 1]) %= MOD;
            (dp[1][i][j] += dp[1][i - 1][j] + (j == N ? 0LL : dp[1][i - 1][j + 1])) %= MOD;
            if (j < N and i + j >= N) { (dp[1][i][j] += dp[0][i + j - N][j + 1]) %= MOD; }
            sum[0][i][j] = (sum[0][i][j - 1] + dp[0][i][j]) % MOD;
            sum[1][i][j] = (sum[1][i][j - 1] + dp[1][i][j]) % MOD;
        }
    }
    std::cout << (sum[0][N][N] + sum[1][N][N]) % MOD << std::endl;
    return 0;
}
0