結果

問題 No.2444 一次変換と体積
ユーザー tassei903tassei903
提出日時 2023-08-25 23:14:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 903 ms / 2,000 ms
コード長 2,159 bytes
コンパイル時間 398 ms
コンパイル使用メモリ 82,208 KB
実行使用メモリ 76,168 KB
最終ジャッジ日時 2024-12-24 11:03:11
合計ジャッジ時間 2,738 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################
import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################
n, mod = na()
a = [na() for i in range(3)]
b = list(zip(*a))

z = a[0][0]*a[1][1]*a[2][2] + a[0][1]*a[1][2]*a[2][0] + a[0][2]*a[1][0]*a[2][1] - a[0][2]*a[1][1]*a[2][0] - a[0][1]*a[1][0]*a[2][2] - a[0][0]*a[1][2]*a[2][1]
if z != 0:
    z = abs(z)
    print(pow(z, n, mod))
    exit()
def f(a, b):
    if a[0]*b[1] - a[1]*b[0] == 0 and a[0]*b[2] - a[2]*b[0] == 0:
        return 1
    return 0

def det(a, b):
    return a[1]*b[2]-a[2]*b[1], a[2]*b[0]-a[0]*b[2], a[0]*b[1]-a[1]*b[0]

def mat_mul(A,B):
    assert len(A[0]) == len(B)
    n = len(A)
    m = len(B[0])
    p = len(A[0])
    R = [[0 for j in range(m)]for i in range(n)]
    for i in range(n):
        for j in range(m):
            for k in range(p):
                R[i][j] += A[i][k]*B[k][j] % mod
                R[i][j] %= mod
    return R
e = [[1,0,0],[0,1,0],[0,0,1]]
for i in range(n):
    e = mat_mul(e, a)
a = e

b = list(zip(*a))
if a[0].count(0) + a[1].count(0) + a[2].count(0) == 9:
    print(0)

elif f(b[0], b[1]) and f(b[0], b[2]):
    if (a[0][0] >= 0 and a[1][0] >= 0 and a[2][0] >= 0) or  (a[0][0] <= 0 and a[1][0] <= 0 and a[2][0] <= 0):
        print(0)
    else:
        print("infty")
else:
    if f(b[0], b[1])^1:
        z = det(b[0], b[1])
    elif f(b[0], b[1])^1:
        z = det(b[0], b[2])
    else:
        z = det(b[1], b[1])
    if (z[0] > 0 and z[1] > 0 and z[2] > 0) or (z[0] < 0 and z[1] < 0 and z[2] < 0):
        print("infty")
    else:
        print(0)
0