結果

問題 No.2584 The University of Tree
ユーザー hitonanodehitonanode
提出日時 2023-12-12 00:59:58
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 363 ms / 3,000 ms
コード長 2,634 bytes
コンパイル時間 1,356 ms
コンパイル使用メモリ 102,672 KB
実行使用メモリ 58,752 KB
最終ジャッジ日時 2024-09-27 04:44:25
合計ジャッジ時間 10,938 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 3 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 146 ms
30,976 KB
testcase_23 AC 135 ms
14,664 KB
testcase_24 AC 77 ms
8,832 KB
testcase_25 AC 85 ms
9,728 KB
testcase_26 AC 62 ms
10,368 KB
testcase_27 AC 131 ms
17,792 KB
testcase_28 AC 67 ms
14,208 KB
testcase_29 AC 34 ms
8,064 KB
testcase_30 AC 277 ms
26,404 KB
testcase_31 AC 313 ms
40,320 KB
testcase_32 AC 56 ms
15,488 KB
testcase_33 AC 186 ms
57,600 KB
testcase_34 AC 40 ms
7,552 KB
testcase_35 AC 319 ms
39,164 KB
testcase_36 AC 243 ms
21,840 KB
testcase_37 AC 244 ms
18,048 KB
testcase_38 AC 241 ms
17,972 KB
testcase_39 AC 282 ms
27,308 KB
testcase_40 AC 305 ms
40,188 KB
testcase_41 AC 316 ms
34,304 KB
testcase_42 AC 308 ms
35,680 KB
testcase_43 AC 283 ms
30,464 KB
testcase_44 AC 344 ms
47,232 KB
testcase_45 AC 363 ms
47,232 KB
testcase_46 AC 190 ms
58,752 KB
testcase_47 AC 232 ms
21,828 KB
testcase_48 AC 82 ms
9,088 KB
testcase_49 AC 235 ms
16,864 KB
testcase_50 AC 188 ms
14,592 KB
testcase_51 AC 97 ms
9,728 KB
testcase_52 AC 41 ms
6,400 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <utility>
#include <vector>
using namespace std;
#define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i<i##_end_;i++)
#define REP(i, n) FOR(i,0,n)

#include <atcoder/modint>
using mint = atcoder::modint998244353;

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);

    int N;
    cin >> N;
    vector<vector<int>> to_edges(N);
    vector<int> edgexor(N - 1);
    REP(i, N) {
        int C;
        cin >> C;
        while (C--) {
            int e;
            cin >> e;
            --e;
            edgexor.at(e) ^= i;
            to_edges.at(i).push_back(e);
        }
    }

    vector<mint> dpchild(N), dppar(N);
    vector<mint> ret(N);
    vector<int> curs(N, -1);

    auto rec1 = [&](auto &&self, int now, int prvvid, int prveid) -> void {

        if (prveid >= 0) {
            int cur = 0;
            while (to_edges.at(now).at(cur) != prveid) ++cur;
            curs.at(now) = cur;
        }

        const int D = to_edges.at(now).size();

        for (int eid : to_edges.at(now)) {
            if (eid != prveid) self(self, edgexor.at(eid) ^ now, now, eid);
        }

        vector<mint> tmp;
        tmp.reserve((1 + D) * 2);
        tmp.push_back(1);
        for (int eid : to_edges.at(now)) tmp.push_back(dpchild.at(edgexor.at(eid) ^ now));

        REP(i, 1 + D) tmp.push_back(tmp.at(i));

        vector<mint> tmp_cs(tmp.size() + 1);
        const mint w = mint(now + 1).inv();
        const mint winvpow = mint(now + 1).pow(D);

        REP(i, tmp.size()) tmp_cs.at(i + 1) = tmp_cs.at(i) * w + tmp.at(i);

        auto get_sum = [&](int l) -> mint {
            return tmp_cs.at(l + D) * winvpow - tmp_cs.at(l);
        };

        ret.at(now) += get_sum(1);

        if (curs.at(now) >= 0) dpchild.at(now) += get_sum(1 + curs.at(now) + 1);

        REP(d, D) {
            int nxt = edgexor.at(to_edges.at(now).at(d)) ^ now;
            if (nxt != prvvid) dppar.at(nxt) += get_sum(1 + d + 1);
        }
    };

    rec1(rec1, 0, -1, -1);

    auto rec2 = [&](auto &&self, int now, int prveid) -> void {
        const int D = to_edges.at(now).size();

        ret.at(now) += dppar.at(now) * mint(now + 1).pow(1 + curs.at(now));
        REP(d, D) {
            const int eid = to_edges.at(now).at(d);
            if (eid == prveid) continue;
            int nxt = edgexor.at(eid) ^ now;

            if (prveid >= 0) {
                dppar.at(nxt) += dppar.at(now) * mint(now + 1).pow((curs.at(now) + (D + 1) - d) % (D + 1));
            }

            self(self, nxt, eid);
        }
    };
    rec2(rec2, 0, -1);

    for (auto x : ret) cout << x.val() << '\n';
}
0