2021 ICPC Southeastern Europe Regional Contest 补题

C

对于一种颜色而言，直接同色+1，不同色-1，树形DP计算和为正的连通子图个数，每次效率为$O(n^2)$。
然而对于子树合并的树形DP，还有一个性质：如果第二维的上届是$m$，总的效率是$O(nm)$的！
对于本题，和的绝对值必然不超过同色的节点数，所以每次的效率可以做到$O(n\times 节点数)$！

K

如果已经确定根，那么直接贪心就好了，所以问题在于找到最优的根。
答案的第一个肯定是编号最小的叶节点（记为v），所以考虑在访问完v之后，退回到u，这时有两种情况：
1.u是根
2.根在u连接的其它子树中的一个（记为T）
对于第二种情况，原本不知道该如何下手；不妨考虑最终的答案：在把与u相连的其它子树遍历完后，加上u，然后遍历T。显然T的最小叶节点应该是与u相连的子树中最大的（记为mx），那么会发现与第一种情况的区别在于：下一个位置是u还是mx，那么直接据此判断u是否为根即可。如果u不是根，把u看做v，就转化为了原来的问题。
对于字典序的贪心，冲着最终答案考虑就会可做很多！