結果
問題 | No.824 Many Shifts Hard |
ユーザー | leaf_1415 |
提出日時 | 2019-04-29 14:12:27 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 514 ms / 2,000 ms |
コード長 | 1,171 bytes |
コンパイル時間 | 379 ms |
コンパイル使用メモリ | 54,604 KB |
実行使用メモリ | 222,008 KB |
最終ジャッジ日時 | 2023-08-25 21:56:30 |
合計ジャッジ時間 | 6,037 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge14 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,380 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 35 ms
28,452 KB |
testcase_03 | AC | 157 ms
94,564 KB |
testcase_04 | AC | 127 ms
64,012 KB |
testcase_05 | AC | 381 ms
183,228 KB |
testcase_06 | AC | 6 ms
4,548 KB |
testcase_07 | AC | 88 ms
35,764 KB |
testcase_08 | AC | 312 ms
170,472 KB |
testcase_09 | AC | 284 ms
124,700 KB |
testcase_10 | AC | 27 ms
14,576 KB |
testcase_11 | AC | 71 ms
31,932 KB |
testcase_12 | AC | 163 ms
83,828 KB |
testcase_13 | AC | 250 ms
155,564 KB |
testcase_14 | AC | 380 ms
221,876 KB |
testcase_15 | AC | 344 ms
203,636 KB |
testcase_16 | AC | 41 ms
32,948 KB |
testcase_17 | AC | 373 ms
220,496 KB |
testcase_18 | AC | 2 ms
4,376 KB |
testcase_19 | AC | 376 ms
222,008 KB |
testcase_20 | AC | 512 ms
221,944 KB |
testcase_21 | AC | 3 ms
4,380 KB |
testcase_22 | AC | 514 ms
221,860 KB |
ソースコード
#include <iostream> #define llint long long #define mod 1000000007 using namespace std; llint n, K; llint dp[305][305][305], dp2[305][305]; llint sum[305]; int main(void) { cin >> n >> K; dp[0][0][1] = 1; for(int i = 0; i < K; i++){ for(int j = 0; j <= K+1; j++){ for(int k = 0; k <= K+1; k++){ if(j == k){ (dp[i+1][j+1][k+1] += dp[i][j][k]) %= mod; (dp[i+1][j][k] += dp[i][j][k] * (n-1) % mod) %= mod; } else{ (dp[i+1][j+1][k] += dp[i][j][k]) %= mod; (dp[i+1][j][k+1] += dp[i][j][k]) %= mod; (dp[i+1][j][k] += dp[i][j][k] * (n-2) % mod) %= mod; } } } } dp2[0][0] = 1; for(int i = 0; i < K; i++){ for(int j = 0; j <= K+1; j++){ (dp2[i+1][j+1] += dp2[i][j]) %= mod; (dp2[i+1][j] += dp2[i][j] * (n-1) % mod) %= mod; } } for(int j = 0; j <= K; j++){ for(int k = 0; k < j+1; k++){ sum[j] += dp[K][k][j+1] % mod, sum[j] %= mod; } } llint ans = 0; for(int i = 1; i <= n; i++){ llint cnt = 0; for(int j = 0; j <= min(K, n-i); j++){ if(j < n-i) cnt += sum[j], cnt %= mod; else cnt += dp2[K][j], cnt %= mod; } ans += cnt * i % mod, ans %= mod; } cout << ans << endl; return 0; }