結果

問題 No.2367 Painting Gascket
ユーザー shobonvipshobonvip
提出日時 2023-06-30 05:41:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 114 ms / 2,000 ms
コード長 2,043 bytes
コンパイル時間 359 ms
コンパイル使用メモリ 87,168 KB
実行使用メモリ 81,648 KB
最終ジャッジ日時 2023-09-21 12:31:10
合計ジャッジ時間 5,627 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 74 ms
71,384 KB
testcase_01 AC 75 ms
71,288 KB
testcase_02 AC 89 ms
71,212 KB
testcase_03 AC 76 ms
71,336 KB
testcase_04 AC 75 ms
71,496 KB
testcase_05 AC 75 ms
71,408 KB
testcase_06 AC 75 ms
71,560 KB
testcase_07 AC 75 ms
71,092 KB
testcase_08 AC 114 ms
81,520 KB
testcase_09 AC 101 ms
79,980 KB
testcase_10 AC 94 ms
76,808 KB
testcase_11 AC 106 ms
81,064 KB
testcase_12 AC 94 ms
76,940 KB
testcase_13 AC 98 ms
78,272 KB
testcase_14 AC 93 ms
76,812 KB
testcase_15 AC 93 ms
76,912 KB
testcase_16 AC 98 ms
78,588 KB
testcase_17 AC 105 ms
81,172 KB
testcase_18 AC 105 ms
80,044 KB
testcase_19 AC 97 ms
79,024 KB
testcase_20 AC 105 ms
80,960 KB
testcase_21 AC 99 ms
79,268 KB
testcase_22 AC 106 ms
81,468 KB
testcase_23 AC 102 ms
80,168 KB
testcase_24 AC 100 ms
78,228 KB
testcase_25 AC 98 ms
78,192 KB
testcase_26 AC 97 ms
76,900 KB
testcase_27 AC 106 ms
80,220 KB
testcase_28 AC 101 ms
79,860 KB
testcase_29 AC 75 ms
71,404 KB
testcase_30 AC 73 ms
71,404 KB
testcase_31 AC 76 ms
71,196 KB
testcase_32 AC 74 ms
71,260 KB
testcase_33 AC 104 ms
81,488 KB
testcase_34 AC 107 ms
81,592 KB
testcase_35 AC 108 ms
81,648 KB
testcase_36 AC 110 ms
81,552 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] = max(0, n-1)
dp2[0] = max(0, n-2)
dp3[0] = max(0, n-3)
dp4[0] = max(0, n-1)
dp5[0] = max(0, n-2)
dp6[0] = max(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 * max(0, 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 * max(0, 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 * max(0, 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 * max(0, 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 * max(0, 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 * max(0, 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