結果
問題 | No.196 典型DP (1) |
ユーザー | maspy |
提出日時 | 2020-01-01 14:21:44 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 583 ms / 2,000 ms |
コード長 | 1,100 bytes |
コンパイル時間 | 271 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 57,064 KB |
最終ジャッジ日時 | 2024-05-01 20:54:40 |
合計ジャッジ時間 | 26,705 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 453 ms
43,620 KB |
testcase_01 | AC | 450 ms
44,136 KB |
testcase_02 | AC | 463 ms
43,884 KB |
testcase_03 | AC | 454 ms
44,388 KB |
testcase_04 | AC | 468 ms
44,396 KB |
testcase_05 | AC | 460 ms
44,008 KB |
testcase_06 | AC | 459 ms
44,008 KB |
testcase_07 | AC | 448 ms
44,136 KB |
testcase_08 | AC | 478 ms
44,396 KB |
testcase_09 | AC | 449 ms
44,392 KB |
testcase_10 | AC | 464 ms
44,264 KB |
testcase_11 | AC | 465 ms
44,008 KB |
testcase_12 | AC | 458 ms
44,392 KB |
testcase_13 | AC | 462 ms
44,144 KB |
testcase_14 | AC | 523 ms
44,008 KB |
testcase_15 | AC | 482 ms
44,268 KB |
testcase_16 | AC | 492 ms
43,852 KB |
testcase_17 | AC | 478 ms
44,140 KB |
testcase_18 | AC | 493 ms
44,008 KB |
testcase_19 | AC | 512 ms
43,880 KB |
testcase_20 | AC | 560 ms
44,264 KB |
testcase_21 | AC | 550 ms
44,004 KB |
testcase_22 | AC | 547 ms
44,268 KB |
testcase_23 | AC | 558 ms
44,908 KB |
testcase_24 | AC | 526 ms
44,136 KB |
testcase_25 | AC | 529 ms
44,140 KB |
testcase_26 | AC | 552 ms
56,680 KB |
testcase_27 | AC | 572 ms
57,064 KB |
testcase_28 | AC | 552 ms
56,812 KB |
testcase_29 | AC | 544 ms
56,932 KB |
testcase_30 | AC | 542 ms
56,556 KB |
testcase_31 | AC | 560 ms
44,272 KB |
testcase_32 | AC | 568 ms
43,880 KB |
testcase_33 | AC | 558 ms
44,008 KB |
testcase_34 | AC | 583 ms
43,752 KB |
testcase_35 | AC | 571 ms
44,288 KB |
testcase_36 | AC | 578 ms
44,260 KB |
testcase_37 | AC | 531 ms
44,132 KB |
testcase_38 | AC | 565 ms
44,392 KB |
testcase_39 | AC | 559 ms
44,016 KB |
testcase_40 | AC | 575 ms
43,888 KB |
testcase_41 | AC | 463 ms
43,872 KB |
testcase_42 | AC | 467 ms
44,332 KB |
testcase_43 | AC | 464 ms
44,648 KB |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np N,K = map(int,readline().split()) m = map(int,read().split()) AB = zip(m,m) MOD = 10 ** 9 + 7 graph = [[] for _ in range(N+1)] for a,b in AB: a+=1; b+=1 graph[a].append(b) graph[b].append(a) root = 1 parent = [0] * (N+1) order = [] stack = [root] while stack: x = stack.pop() order.append(x) for y in graph[x]: if y == parent[x]: continue parent[y] = x stack.append(y) size = [1] * (N+1) # 部分木の大きさ for v in order[::-1]: p = parent[v] size[p] += size[v] def convolve(A,B): LA = len(A); LB = len(B) if LA > LB: A,B = B,A LA,LB = LB,LA C = np.zeros(LA+LB-1,np.int64) for i,x in enumerate(A): C[i:i+LB] += A[i] * B % MOD C %= MOD return C dp = [np.array([1],np.int64) for _ in range(N+1)] for v in order[::-1]: p = parent[v] dp[v] = np.append(dp[v],1) dp[p] = convolve(dp[p], dp[v]) answer = dp[root][K] print(answer)