Cho cây có gốc tại đỉnh 1, gồm n đỉnh, mỗi đỉnh i có trọng số wi (có thể âm). Cây con gốc tại v gồm v và tất cả con cháu của nó. Định nghĩa T(v) là tổng trọng số của mọi đỉnh trong cây con gốc v.
Hãy tìm maxvT(v). Sử dụng quy hoạch động trên cây (subtree DP) duyệt O(n).
Ví dụ: Cây 1−2, 1−3 với w=[−5,4,4]: T(2)=4, T(3)=4, T(1)=−5+4+4=3, đáp án 4.
Dòng đầu chứa n. Dòng thứ hai chứa n số nguyên w1…wn. n−1 dòng tiếp theo, mỗi dòng hai số là một cạnh.
1≤n≤2⋅105, ∣wi∣≤109.
In ra một số nguyên là tổng trọng số cây con lớn nhất.
Ví dụ:
Đầu vào:
3
-5 4 4
1 2
1 3
Đầu ra:
4
Giải thích:
Đang tải editor...