結果

問題 No.1340 おーじ君をさがせ
コンテスト
ユーザー lam6er
提出日時 2025-04-16 15:38:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 122 ms / 2,000 ms
コード長 1,281 bytes
コンパイル時間 529 ms
コンパイル使用メモリ 81,780 KB
実行使用メモリ 76,452 KB
最終ジャッジ日時 2025-04-16 15:43:56
合計ジャッジ時間 5,263 ms
ジャッジサーバーID
（参考情報） 		judge4 / 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)

if t == 0:
    print(1)
else:
    # Construct the transition matrix masks
    mask = []
    for i in range(n):
        if not edges[i]:
            mask.append(0)
        else:
            m_val = 0
            for j in edges[i]:
                m_val |= 1 << j
            mask.append(m_val)

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

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

    mt = matrix_power(mask, t)
    res_mask = mt[0]
    print(bin(res_mask).count('1'))
0