글 읽기 - 이 대체 왜 이러는지

lkyxbar 1년 전 0

별다른건 없고 답이 불규칙하게 출력됩니다 대체 왜 이러는 거죠?

참고로 마지막줄에 행렬 전체를 출력하는건 테스트의 용도입니다

import sys
def matrix(a, b):
    n, m, o=len(a), len(b), len(b[0])
    asdf=[[0 for _ in range(o)] for _ in range(n)]
    for i in range(n):
        for j in range(o):
            for l in range(m):
                asdf[i][j]+=a[i][l]*b[l][j]
                asdf[i][j]+=1000000007
                asdf[i][j]%=1000000007
    return asdf

def inv(a, K):
    if not K-1:
        return a

    temp=inv(a, K>>1)
    if K%2:
        return matrix(matrix(temp, temp), a)

    elif not K%2:
        return matrix(temp, temp)

N=int(sys.stdin.readline())
if N&1:
    print(0)
else:
    t=[
        [4, -1],
        [1, 0]
    ]
    f=[[3],
       [1]
    ]
    ans=matrix(inv(t, N-1), f)
    print(ans)

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

문제

문제

출처

ICPC

이 대체 왜 이러는지

lkyxbar 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말