結果

問題 No.2215 Slide Subset Sum
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2023-02-11 06:19:42
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 1,422 bytes
コンパイル時間 1,972 ms
コンパイル使用メモリ 181,060 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-07 22:19:04
合計ジャッジ時間 9,755 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using Vl = std::vector<ll> ;
ll const m = 998244353;
std::stack<Vl> st1, st2;
stack<int> stt;
int N, M, K;
void Push(int a, int id = 1) {
    stack<Vl> S = (id ? st2 : st1);
    Vl A(K, 0);
    if (S.empty()) {
        A[0] = 1;
        A[a] ++;
    } else {
        auto B = S.top();
        for (int i = 0; i < K; i ++) {
            A[i] = B[i] + B[(i + K - a) % K];
            A[i] %= m;
        }
    }
    S.push(A);
    if (id == 1) {
        stt.push(a);
    }
}
void Pop() {
    if (!st1.empty()) {
        st1.pop();
    } else {
        while (!st2.empty()) {
            auto v = st2.top();
            st2.pop();
            auto a = stt.top();
            stt.pop();
            Push(a, 0);
        }
        st1.pop();
    }
}
void CA() {
    Vl A(K, 0), B(K, 0);
    if (st1.empty()) {
        A[0] = 1;
    } else {
        A = st1.top();
    }
    if (st2.empty()) {
        B[0] = 1;
    } else {
        B = st2.top();
    }
    ll ans = m - 1;
    for (int i = 0; i < K; i ++) {
        ans += (A[i] * B[(K - i) % K]) % m;
        ans %= m;
    }
    cout << ans << endl;
}
int main () {
    cin >> N >> M >> K;
    for (int i = 0; i < M - 1; i ++) {
        int a;
        cin >> a;
        Push(a);
    }
    for (int i = M - 1; i < N; i ++) {
        int a;
        cin >> a;
        Push(a);
        CA();
        Pop();
    }
}
0