글 읽기 - 반례 부탁드립니다.

rudtn0403 2년 전 0

80%에서 틀리고 있습니다. 어떤 경우에 문제가 생기나요?

import sys

N, K = map(int, sys.stdin.readline().split())
A = [list(map(int, sys.stdin.readline().split())) for _ in range(N)]

def multiply(n, A, B):
    H = [[0 for _ in range(N)] for _ in range(N)]
    for i in range(n):
        for j in range(n):
            for r in range(n):
                H[i][j] = (H[i][j] + A[i][r]*B[r][j])%1000
    return H
def power(n, k, H):
    if k==1:
        return H
    elif k==2:
        return multiply(n, H, H)
    else:
        tmp = power(n, k//2, H)
        if k%2==0:
            return multiply(n, tmp, tmp)
        else:
            return multiply(n, multiply(n, tmp, tmp), H)

ans = power(N,K,A)
for arr in ans:
    print(*arr)

ai4youej 2년 전 0

https://www.acmicpc.net/board/...

rudtn0403 2년 전 0

감사합니다

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

반례 부탁드립니다.

rudtn0403 2년 전 0

ai4youej 2년 전 0

rudtn0403 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말