結果

問題 No.2367 Painting Gascket
ユーザー shobonvipshobonvip
提出日時 2023-06-30 01:54:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 68 ms / 2,000 ms
コード長 1,959 bytes
コンパイル時間 222 ms
コンパイル使用メモリ 82,168 KB
実行使用メモリ 72,564 KB
最終ジャッジ日時 2024-07-07 06:20:28
合計ジャッジ時間 3,130 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
52,348 KB
testcase_01 AC 35 ms
52,836 KB
testcase_02 AC 42 ms
56,300 KB
testcase_03 AC 37 ms
52,696 KB
testcase_04 AC 36 ms
52,384 KB
testcase_05 AC 36 ms
52,952 KB
testcase_06 AC 37 ms
52,672 KB
testcase_07 AC 35 ms
52,264 KB
testcase_08 AC 65 ms
72,564 KB
testcase_09 AC 61 ms
70,140 KB
testcase_10 AC 52 ms
67,656 KB
testcase_11 AC 64 ms
70,112 KB
testcase_12 AC 54 ms
68,144 KB
testcase_13 AC 54 ms
67,848 KB
testcase_14 AC 52 ms
66,684 KB
testcase_15 AC 53 ms
67,392 KB
testcase_16 AC 57 ms
68,060 KB
testcase_17 AC 62 ms
71,276 KB
testcase_18 AC 63 ms
69,864 KB
testcase_19 AC 60 ms
69,104 KB
testcase_20 AC 65 ms
70,724 KB
testcase_21 AC 60 ms
69,656 KB
testcase_22 AC 64 ms
71,148 KB
testcase_23 AC 62 ms
69,696 KB
testcase_24 AC 55 ms
68,228 KB
testcase_25 AC 57 ms
68,340 KB
testcase_26 AC 54 ms
67,128 KB
testcase_27 AC 61 ms
69,944 KB
testcase_28 AC 62 ms
70,056 KB
testcase_29 AC 35 ms
52,612 KB
testcase_30 AC 35 ms
53,560 KB
testcase_31 AC 34 ms
52,408 KB
testcase_32 AC 35 ms
53,564 KB
testcase_33 AC 65 ms
72,204 KB
testcase_34 AC 68 ms
71,096 KB
testcase_35 AC 67 ms
70,664 KB
testcase_36 AC 66 ms
71,080 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 難しい

# 部分的なグラフを 6 種類程度考え, それは DP になるだろう.
# dp1[i]
# = (A) dp1[i-1] ** 3
# + (B) (n-1) * dp2[i-1] ** 3
# dp2[i]
# = (A) dp2[i-1] ** 3
# + (B) dp1[i-1] * dp2[i-1] ** 2
# + (C) (n-2) * dp2[i-1] * dp3[i-1] ** 2
# dp3[i]
# = (A, B, C) 3 * dp2[i-1] ** 2 * dp3[i-1]
# + (D) (n-3) * dp3[i-1] ** 3
# dp4[i]
# = (A) dp1[i-1] * dp4[i-1] ** 2
# + (B) (n-1) * dp2[i-1] * dp5[i-1] ** 2
# dp5[i]
# = (A, B) 2 * dp2[i-1] * dp4[i-1] * dp5[i-1]
# = (C) (n-2) * dp1[i-1] * dp5[i-1] ** 2
# dp6[i]
# = (A) dp4[i-1] ** 2 * dp6[i-1]
# + (B) (n-1) * dp5[i-1] ** 2 * dp6[i-1]

# 答えは dp6[k-1] ** 3 * n

# 初期値
# dp1[0] = n-1
# dp2[0] = n-2
# dp3[0] = n-3
# dp4[0] = n-1
# dp5[0] = n-2
# dp6[0] = n-1

mod = 10 ** 9 + 7

k, n = map(int,input().split())

if k == 0:
	print(n % mod)
	exit()

dp1 = [0] * k
dp2 = [0] * k
dp3 = [0] * k
dp4 = [0] * k
dp5 = [0] * k
dp6 = [0] * k

dp1[0] = n-1
dp2[0] = n-2
dp3[0] = n-3
dp4[0] = n-1
dp5[0] = n-2
dp6[0] = n-1

for i in range(1, k):
	dp1[i] = dp1[i-1] * dp1[i-1] % mod * dp1[i-1] % mod\
		   + dp2[i-1] * dp2[i-1] % mod * dp2[i-1] % mod * (n-1) % mod
	dp2[i] = dp2[i-1] * dp2[i-1] % mod * dp2[i-1] % mod\
		   + dp1[i-1] * dp2[i-1] % mod * dp2[i-1] % mod\
		   + dp2[i-1] * dp3[i-1] % mod * dp3[i-1] % mod * (n-2) % mod
	dp3[i] = dp2[i-1] * dp2[i-1] % mod * dp3[i-1] % mod * 3 % mod\
		   + dp3[i-1] * dp3[i-1] % mod * dp3[i-1] % mod * (n-3) % mod
	dp4[i] = dp1[i-1] * dp4[i-1] % mod * dp4[i-1] % mod\
		   + dp2[i-1] * dp5[i-1] % mod * dp5[i-1] % mod * (n-1) % mod
	dp5[i] = dp2[i-1] * dp4[i-1] % mod * dp5[i-1] % mod * 2 % mod\
		   + dp3[i-1] * dp5[i-1] % mod * dp5[i-1] % mod * (n-2) % mod
	dp6[i] = dp4[i-1] * dp4[i-1] % mod * dp6[i-1] % mod\
		   + dp5[i-1] * dp5[i-1] % mod * dp6[i-1] % mod * (n-1) % mod
	dp1[i] %= mod
	dp2[i] %= mod
	dp3[i] %= mod
	dp4[i] %= mod
	dp5[i] %= mod
	dp6[i] %= mod

#print(dp6[0])

print(dp6[k-1] * dp6[k-1] % mod * dp6[k-1] % mod * n % mod)
0