diff --git a/.vscode/settings.json b/.vscode/settings.json index 387f371..c0e8e39 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -1,4 +1,4 @@ { "python.analysis.autoImportCompletions": true, - "python.analysis.typeCheckingMode": "strict" + "python.analysis.typeCheckingMode": "standard" } \ No newline at end of file diff --git a/src/2/P3390.py b/src/2/P3390.py index 1056950..6031a29 100644 --- a/src/2/P3390.py +++ b/src/2/P3390.py @@ -1,7 +1,34 @@ import re import sys +import numpy as np + stdin:str = sys.stdin.read() -lines = re.split(r"\s+",stdin.strip()) +lines = list(map(int,re.split(r"\s+",stdin.strip()))) +def gn(now={ + "num":0 +}): + now["num"]=now["num"]+1 + return lines[now["num"]-1] + + +n = gn() +k= gn() + +ma = np.zeros((n,n)) +for i in range(n): + for j in range(n): + ma[i][j] = gn() + # print(ma[i][j]) + +res = np.identity(n) + +for i in range(k): + res = res@ma + +for i in range(n): + for j in range(n): + print(f'{int(res[i][j]%(1e9+7))} ',end='') + print() \ No newline at end of file