結果

問題 No.1340 おーじ君をさがせ
ユーザー lam6er
提出日時 2025-04-15 21:47:04
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 1,001 bytes
コンパイル時間 352 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 76,732 KB
最終ジャッジ日時 2025-04-15 21:48:31
合計ジャッジ時間 6,346 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 59
権限があれば一括ダウンロードができます

ソースコード

diff #

n, m, t = map(int, input().split())
edges = [[] for _ in range(n)]
for _ in range(m):
    a, b = map(int, input().split())
    edges[a].append(b)

# Preprocess adjacency matrix with bitmasks
adj = [0] * n
for u in range(n):
    if len(edges[u]) > 0:
        mask = 0
        for v in edges[u]:
            mask |= 1 << v
        adj[u] = mask

def multiply(a, b):
    n = len(a)
    result = [0] * n
    for i in range(n):
        row = a[i]
        res = 0
        for k in range(n):
            if (row >> k) & 1:
                res |= b[k]
        result[i] = res
    return result

def matrix_power(mat, power):
    n = len(mat)
    # Initialize result as identity matrix
    result = [0] * n
    for i in range(n):
        result[i] = 1 << i
    while power > 0:
        if power % 2 == 1:
            result = multiply(result, mat)
        mat = multiply(mat, mat)
        power //= 2
    return result

powered = matrix_power(adj, t)
row0 = powered[0]
count = bin(row0).count('1')
print(count)
0