結果

問題 No.1608 Yet Another Ants Problem
ユーザー むかで
提出日時 2021-06-08 14:44:28
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 85 ms / 2,000 ms
コード長 1,354 bytes
コンパイル時間 1,181 ms
コンパイル使用メモリ 78,856 KB
最終ジャッジ日時 2025-01-22 04:55:40
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;
using ll = long long;

constexpr int mod = 998244353;

int main()
{
    int N,L; cin>>N>>L;
    vector<int> A(N);
    for(int i=0; i<N; i++) cin>>A[i];

    vector<vector<int>> dp(N+1, vector<int>(N+1));
    dp[1][0] = 1;
    for(int i=2; i<=N; i++){
        for(int j=0; j<=N; j++){
            if (j == 0){
                dp[i][j] = (dp[i-1][j]+1 < mod ? dp[i-1][j]+1 : 0);
            }else{
                dp[i][j] = dp[i-1][j] + dp[i-1][j-1];
                if (dp[i][j] >= mod) dp[i][j] -= mod;
            }
        }
    }

    vector<int> ans(N);
    
    ans[0]++; ans[N-1]++;
    for(int i=0; i<N-1; i++){
        if (A[i] >= L-A[i+1]) ans[i]++;
        else ans[i+1]++;
    }

    int j = N;
    for(int i=0; i<N; i++){
        while(j > 0 && A[j-1] >= L-A[i]) j--;

        for(int k=i+1, l=0; k<j; k++, l++){
            ans[k] += dp[j-i-1][l];
            if (ans[k] >= mod) ans[k] -= mod;
        }

        for(int k=i, l=0; k<N-1; k++, l++){
            ans[k] += dp[N-i-1][l];
            if (ans[k] >= mod) ans[k] -= mod;
        }
        for(int k=i, l=0; k<j-1; k++, l++){
            ans[k] -= dp[j-i-1][l];
            if (ans[k] < 0) ans[k] += mod;
        }
    }

    for(int i=0; i<N; i++){
        cout<<ans[i]<<"\n";
    }

    return 0;
}
0