結果
| 問題 |
No.2727 Tetrahedron Game
|
| コンテスト | |
| ユーザー |
MasKoaTS
|
| 提出日時 | 2023-10-18 15:48:20 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 426 ms / 2,000 ms |
| コード長 | 1,448 bytes |
| コンパイル時間 | 215 ms |
| コンパイル使用メモリ | 81,792 KB |
| 実行使用メモリ | 79,456 KB |
| 最終ジャッジ日時 | 2024-09-18 11:55:44 |
| 合計ジャッジ時間 | 4,568 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 9 |
ソースコード
import sys
input = sys.stdin.readline
MOD = 101
INV2, INV3 = pow(2, MOD - 2, MOD), pow(3, MOD - 2, MOD)
def det(A):
ret = 0
for i in range(3):
ret += A[i][0] * A[(i + 1) % 3][1] * A[(i + 2) % 3][2]
ret -= A[i][2] * A[(i + 1) % 3][1] * A[(i + 2) % 3][0]
return ret
def solve(N, K, P, a, s):
d = abs(det(P))
if(d == 0):
return 'D'
dp = [[-1] * 4 for _ in [0] * MOD]
for j in range(MOD):
dp[j][3] = int(j >= K)
for i in range(N - 1, -1, -1):
ndp = [[-1] * 4 for _ in [0] * MOD]
hands = [1, -1, 0] if(s[i] == 'K') else [0, -1, 1]
val = a[i] + 1
val_div2, val_div3 = (val % 2 == 0), (val % 3 == 0)
for j in range(MOD):
for k in range(4):
results = [dp[j][k], dp[j * val % MOD][k | val_div2 | (val_div3 << 1)]]
for x in hands:
if(x not in results):
continue
ndp[j][k] = x
break
dp = ndp[:]
div2, div3 = (d % 2 == 0), (d % 3 == 0)
stat_init = dp[d % MOD * INV2 * INV3 % MOD][div2 | (div3 << 1)]
if(stat_init == -1):
return 'D'
return 'K' if(stat_init) else 'P'
t = int(input())
for _ in [0] * t:
N, K = map(int, input().split())
P = [list(map(int, input().split())) for _ in [0] * 3]
a = list(map(int, input().split()))
s = input()[:-1]
print(solve(N, K, P, a, s))
MasKoaTS