結果

問題 No.2522 Fall in love, Girls!
ユーザー MasKoaTSMasKoaTS
提出日時 2023-08-26 23:58:07
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,209 bytes
コンパイル時間 292 ms
コンパイル使用メモリ 82,448 KB
実行使用メモリ 84,844 KB
最終ジャッジ日時 2024-09-25 13:15:55
合計ジャッジ時間 4,376 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 2 TLE * 1 -- * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
MOD = 998244353

n, m, _ = map(int, input().split())
edges = [[x - 1 for x in map(int, input().split())] for _ in [0] * _]

st = set([])
for edge in edges:
    st |= set(edge)
lis = list(st)
di = { x : i for i, x in enumerate(lis)}
t = len(lis)

graph = [[0] * t for _ in [0] * t]
for x, y in edges:
    graph[di[x]][di[y]] = 1

dp = [-1] * (1 << t)
dp[0] = 1
def calc_dp(s):
    global dp
    if(dp[s] == -1):
        dp[s] = 0
        for i in range(t):
            if not((s >> i) & 1):
                continue
            if all(i == j or not((s >> j) & 1) or graph[j][i] == 0 for j in range(t)):
                dp[s] += calc_dp(s - (1 << i))
        dp[s] %= MOD
    return dp[s]

def nPk(n, k):
    ret = 1
    for i in range(n, n - k, -1):
        ret *= i
        ret %= MOD
    return ret

ans = nPk(n, n - t) * calc_dp((1 << t) - 1) % MOD
for i in range(m):
    if(i in st):
        id = di[i]
        if any(id != j and graph[j][id] == 1 for j in range(t)):
            continue
        ans -= nPk(n - 1, n - t) * calc_dp((1 << t) - 1 - (1 << id)) % MOD
    else:
        ans -= nPk(n - 1, n - t - 1) * calc_dp((1 << t) - 1) % MOD
    ans %= MOD

print(ans)
0