結果

問題 No.2367 Painting Gascket
ユーザー shobonvipshobonvip
提出日時 2023-06-30 05:41:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 68 ms / 2,000 ms
コード長 2,043 bytes
コンパイル時間 239 ms
コンパイル使用メモリ 82,116 KB
実行使用メモリ 73,680 KB
最終ジャッジ日時 2024-07-07 06:20:31
合計ジャッジ時間 3,171 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,492 KB
testcase_01 AC 38 ms
53,012 KB
testcase_02 AC 45 ms
57,288 KB
testcase_03 AC 36 ms
54,352 KB
testcase_04 AC 34 ms
52,684 KB
testcase_05 AC 34 ms
53,752 KB
testcase_06 AC 36 ms
53,440 KB
testcase_07 AC 35 ms
53,084 KB
testcase_08 AC 67 ms
72,448 KB
testcase_09 AC 63 ms
70,960 KB
testcase_10 AC 54 ms
67,480 KB
testcase_11 AC 67 ms
71,224 KB
testcase_12 AC 56 ms
68,336 KB
testcase_13 AC 57 ms
69,472 KB
testcase_14 AC 54 ms
68,148 KB
testcase_15 AC 55 ms
68,080 KB
testcase_16 AC 57 ms
70,332 KB
testcase_17 AC 67 ms
72,152 KB
testcase_18 AC 65 ms
70,700 KB
testcase_19 AC 60 ms
70,600 KB
testcase_20 AC 67 ms
71,696 KB
testcase_21 AC 62 ms
69,936 KB
testcase_22 AC 66 ms
73,128 KB
testcase_23 AC 63 ms
70,920 KB
testcase_24 AC 60 ms
68,960 KB
testcase_25 AC 59 ms
69,108 KB
testcase_26 AC 53 ms
68,268 KB
testcase_27 AC 62 ms
71,236 KB
testcase_28 AC 62 ms
70,060 KB
testcase_29 AC 34 ms
53,288 KB
testcase_30 AC 36 ms
53,048 KB
testcase_31 AC 37 ms
52,460 KB
testcase_32 AC 37 ms
53,268 KB
testcase_33 AC 64 ms
72,936 KB
testcase_34 AC 66 ms
71,792 KB
testcase_35 AC 68 ms
72,200 KB
testcase_36 AC 68 ms
73,680 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