結果

問題 No.1340 おーじ君をさがせ
コンテスト
ユーザー lam6er
提出日時 2025-04-16 15:38:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 136 ms / 2,000 ms
コード長 1,001 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 81,664 KB
実行使用メモリ 76,728 KB
最終ジャッジ日時 2025-04-16 15:43:35
合計ジャッジ時間 5,684 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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