結果

問題 No.1521 Playing Musical Chairs Alone
ユーザー tarattata1
提出日時 2021-05-29 00:49:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,521 ms / 2,000 ms
コード長 1,910 bytes
コンパイル時間 791 ms
コンパイル使用メモリ 90,052 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-07 11:59:38
合計ジャッジ時間 20,011 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <string>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <iterator>
#include <cassert>
#include <numeric>
#include <functional>
#include <ctime>
#pragma warning(disable:4996) 

//#define ATCODER
#ifdef ATCODER
#include <atcoder/all>
#endif

typedef long long ll;
typedef unsigned long long ull;
#define LINF  9223300000000000000
#define LINF2 1223300000000000000
#define LINF3 1000000000000
#define INF 2140000000
const long long MOD = 1000000007;
//const long long MOD = 998244353;

using namespace std;
#ifdef ATCODER
using namespace atcoder;
#endif

ll dp[2][1000005];

void solve()
{    
    int n, K, L;
    scanf("%d%d%d", &n, &K, &L);
    int pre = 0, cur = 1;
    dp[pre][0] = 1;
    for (int i = 0; i < K; i++) {
        for (int j = 0; j < n; j++) {
            dp[cur][j] = 0;
        }
        for (int j = 0; j < n; j++) {
            int i0 = (j + 1) % n, i1 = i0 + L;
            if (i1 <= n) {
                dp[cur][i0] = (dp[cur][i0] + dp[pre][j]) % MOD;
                dp[cur][i1] = (dp[cur][i1] - dp[pre][j] +MOD) % MOD;
            }
            else {
                i1 -= n;
                dp[cur][0]  = (dp[cur][0]  + dp[pre][j]) % MOD;
                dp[cur][i1] = (dp[cur][i1] - dp[pre][j] +MOD) % MOD;
                dp[cur][i0] = (dp[cur][i0] + dp[pre][j]) % MOD;
            }
        }
        for (int j = 1; j < n; j++) {
            dp[cur][j] = (dp[cur][j] + dp[cur][j - 1]) % MOD;
        }
        swap(pre, cur);
    }
    for (int i = 0; i < n; i++) {
        printf("%lld\n", dp[pre][i]);
    }

    return;
}

int main()
{
#if 1
    solve();
#else
    int T, t;
    scanf("%d", &T);
    for (t = 0; t < T; t++) {
        //printf("Case #%d: ", t + 1);
        solve();
    }
#endif
    return 0;
}


0