結果
問題 | No.1227 I hate ThREE |
ユーザー | betrue12 |
提出日時 | 2020-09-11 23:06:52 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 167 ms / 2,000 ms |
コード長 | 1,319 bytes |
コンパイル時間 | 2,302 ms |
コンパイル使用メモリ | 211,456 KB |
実行使用メモリ | 18,880 KB |
最終ジャッジ日時 | 2023-08-30 14:45:27 |
合計ジャッジ時間 | 6,186 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 2 ms
4,376 KB |
testcase_03 | AC | 37 ms
18,100 KB |
testcase_04 | AC | 38 ms
18,692 KB |
testcase_05 | AC | 38 ms
17,260 KB |
testcase_06 | AC | 2 ms
4,380 KB |
testcase_07 | AC | 2 ms
4,380 KB |
testcase_08 | AC | 9 ms
4,868 KB |
testcase_09 | AC | 8 ms
4,572 KB |
testcase_10 | AC | 5 ms
4,380 KB |
testcase_11 | AC | 89 ms
17,344 KB |
testcase_12 | AC | 108 ms
18,880 KB |
testcase_13 | AC | 2 ms
4,376 KB |
testcase_14 | AC | 8 ms
4,512 KB |
testcase_15 | AC | 15 ms
5,464 KB |
testcase_16 | AC | 41 ms
8,956 KB |
testcase_17 | AC | 22 ms
6,236 KB |
testcase_18 | AC | 59 ms
11,464 KB |
testcase_19 | AC | 45 ms
9,508 KB |
testcase_20 | AC | 66 ms
12,188 KB |
testcase_21 | AC | 2 ms
4,380 KB |
testcase_22 | AC | 46 ms
9,660 KB |
testcase_23 | AC | 156 ms
18,124 KB |
testcase_24 | AC | 145 ms
18,824 KB |
testcase_25 | AC | 167 ms
18,848 KB |
testcase_26 | AC | 102 ms
17,472 KB |
testcase_27 | AC | 111 ms
18,048 KB |
testcase_28 | AC | 113 ms
18,764 KB |
testcase_29 | AC | 114 ms
18,624 KB |
testcase_30 | AC | 106 ms
17,524 KB |
testcase_31 | AC | 103 ms
17,524 KB |
testcase_32 | AC | 113 ms
18,588 KB |
testcase_33 | AC | 108 ms
17,524 KB |
testcase_34 | AC | 109 ms
18,172 KB |
testcase_35 | AC | 104 ms
17,360 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; const int64_t MOD = 1e9+7; void add(int64_t& a, int64_t b){ a = (a+b) % MOD; } void mul(int64_t& a, int64_t b){ a = a*b % MOD; } int main(){ int N, C; cin >> N >> C; vector<int> edges[1000]; for(int i=0; i<N-1; i++){ int a, b; cin >> a >> b; a--; b--; edges[a].push_back(b); edges[b].push_back(a); } C--; int64_t ans = 0; for(int r=0; r<3; r++){ int M = (C-r)/3 + 1; vector<vector<int64_t>> dp(N, vector<int64_t>(2*N+1, -1)); auto rec = [&](auto&& rec, int i, int p, int v)->int64_t{ if(v < 0 || v >= M) return 0; if(dp[i][v] != -1) return dp[i][v]; int64_t res = 1; for(int j : edges[i]) if(j != p){ mul(res, rec(rec, j, i, v-1) + rec(rec, j, i, v+1)); } dp[i][v] = res; return res; }; for(int i=0; i<=N; i++) dp[0][i] = rec(rec, 0, -1, i); int64_t res = 0; if(M <= 2*N){ for(int i=0; i<M; i++) add(res, dp[0][min(i, M-1-i)]); }else{ for(int i=0; i<N; i++) add(res, 2*dp[0][i]); add(res, (M-2*N)*dp[0][N]); } add(ans, res); } cout << ans << endl; return 0; }