結果

問題 No.3589 Make Ends Meet (Hard)
コンテスト
ユーザー marc2825
提出日時 2026-06-13 16:39:12
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
RE  
実行時間 -
コード長 3,266 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 230 ms
コンパイル使用メモリ 96,236 KB
実行使用メモリ 94,080 KB
最終ジャッジ日時 2026-07-10 21:02:10
合計ジャッジ時間 9,199 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 RE * 1
other AC * 8 RE * 39
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys


def solve():
    input_data = sys.stdin.read().split()
    if not input_data:
        return
    N = int(input_data[0])
    M = int(input_data[1])
    K = int(input_data[2])

    MOD = 998244353
    total_edges = N * (N - 1) // 2
    rem_edges = total_edges - M

    if rem_edges < 0:
        print(0)
        return

    C = [[0] * (N + 1) for _ in range(N + 1)]
    for i in range(N + 1):
        C[i][0] = 1
        for j in range(1, i + 1):
            C[i][j] = (C[i - 1][j - 1] + C[i - 1][j]) % MOD

    curr_states = {(1, 0, 0): 1}

    for d in range(1, K + 1):
        next_states = {}

        for (prev_sz, u, e), ways in curr_states.items():
            rem_v = (N - 2) - u
            min_vars = 0
            max_vars = rem_v
            if d != K:
                min_vars = max(min_vars, 1)

            for k_vars in range(min_vars, max_vars + 1):
                sz = k_vars + (1 if d == K else 0)
                if sz == 0:
                    continue

                v_choices = C[rem_v][k_vars]
                if v_choices == 0:
                    continue

                ways_to_connect = [0] * (prev_sz + 1)
                for j in range(1, prev_sz + 1):
                    ways_to_connect[j] = C[prev_sz][j]

                sz_conn = [0] * (prev_sz * sz + 1)
                sz_conn[0] = 1
                for _ in range(sz):
                    next_sz_conn = [0] * (len(sz_conn))
                    for e1 in range(len(sz_conn)):
                        if sz_conn[e1] == 0:
                            continue
                        for e2 in range(1, prev_sz + 1):
                            if e1 + e2 < len(next_sz_conn):
                                next_sz_conn[e1 + e2] = (
                                    next_sz_conn[e1 + e2]
                                    + sz_conn[e1] * ways_to_connect[e2]
                                ) % MOD
                    sz_conn = next_sz_conn

                inner_max = sz * (sz - 1) // 2
                ways_inner = [C[inner_max][i] for i in range(inner_max + 1)]

                base_ways = (ways * v_choices) % MOD

                for e_conn, w_conn in enumerate(sz_conn):
                    if w_conn == 0:
                        continue
                    for e_inner, w_inner in enumerate(ways_inner):
                        added_edges = e_conn + e_inner
                        if e + added_edges > rem_edges:
                            continue

                        n_u = u + k_vars
                        n_e = e + added_edges
                        n_ways = (base_ways * w_conn % MOD) * w_inner % MOD

                        state_key = (sz, n_u, n_e)
                        next_states[state_key] = (
                            next_states.get(state_key, 0) + n_ways
                        ) % MOD

        curr_states = next_states

    ans = 0
    for (prev_sz, u, e), ways in curr_states.items():
        rem_v_final = (N - 2) - u
        free_edges = (
            rem_v_final * (rem_v_final - 1) // 2 + prev_sz * rem_v_final
        )
        needed = rem_edges - e
        if 0 <= needed <= free_edges:
            ans = (ans + ways * C[free_edges][needed]) % MOD

    print(ans)


if __name__ == "__main__":
    solve()
0