【2017多校】HDU6035 Colorful Tree 【據說是樹形DP】

官方題解:單獨考慮每一種顏色，答案就是對於每種顏色至少通過一次這種的路徑條數之和。反過來思考只須要求有多少條路徑沒有通過這種顏色便可。直接作能夠採用虛樹的思想（不用真正建出來），對每種顏色的點按照 dfs 序列排個序，就能求出這些點把原來的樹劃分紅的塊的大小。這個過程實際上能夠直接一次 dfs 求出。ios 令sum[i]=已經dfs遍歷過的點中,以i顏色的點爲根的子樹的節點個數之和(不重複計數)

>>阅读原文<<