結果

問題 No.2367 Painting Gascket
ユーザー shobonvipshobonvip
提出日時 2023-06-30 01:54:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 109 ms / 2,000 ms
コード長 1,959 bytes
コンパイル時間 319 ms
コンパイル使用メモリ 86,956 KB
実行使用メモリ 81,676 KB
最終ジャッジ日時 2023-09-21 12:31:03
合計ジャッジ時間 5,235 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 74 ms
71,588 KB
testcase_01 AC 76 ms
71,388 KB
testcase_02 AC 80 ms
71,236 KB
testcase_03 AC 76 ms
71,100 KB
testcase_04 AC 75 ms
71,236 KB
testcase_05 AC 75 ms
71,348 KB
testcase_06 AC 75 ms
71,380 KB
testcase_07 AC 75 ms
71,500 KB
testcase_08 AC 105 ms
81,676 KB
testcase_09 AC 99 ms
79,872 KB
testcase_10 AC 91 ms
77,000 KB
testcase_11 AC 104 ms
80,908 KB
testcase_12 AC 94 ms
76,972 KB
testcase_13 AC 94 ms
78,212 KB
testcase_14 AC 92 ms
76,860 KB
testcase_15 AC 93 ms
76,708 KB
testcase_16 AC 98 ms
78,344 KB
testcase_17 AC 105 ms
81,072 KB
testcase_18 AC 101 ms
79,936 KB
testcase_19 AC 99 ms
79,020 KB
testcase_20 AC 105 ms
80,928 KB
testcase_21 AC 99 ms
79,268 KB
testcase_22 AC 105 ms
81,408 KB
testcase_23 AC 101 ms
80,148 KB
testcase_24 AC 97 ms
78,096 KB
testcase_25 AC 96 ms
78,176 KB
testcase_26 AC 93 ms
76,836 KB
testcase_27 AC 104 ms
80,168 KB
testcase_28 AC 103 ms
79,764 KB
testcase_29 AC 78 ms
71,508 KB
testcase_30 AC 78 ms
71,104 KB
testcase_31 AC 79 ms
71,480 KB
testcase_32 AC 77 ms
71,592 KB
testcase_33 AC 106 ms
81,404 KB
testcase_34 AC 107 ms
81,668 KB
testcase_35 AC 108 ms
81,548 KB
testcase_36 AC 109 ms
81,644 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