3.28百度笔试b卷第3题

今天回想起来突然思路打开了，当局者迷啊，题目、思路和代码如下，请各位大佬看看有没有问题。

原题：
给出一棵树，每个结点用红色或蓝色染色。定义一条边的贡献为删掉该条边得到的两个子树的同色联通块数的差的绝对值，一棵树的贡献为树上每条边的贡献之和，求该树的贡献。

样例输入：

4

RRBB

1 2

2 3

1 4

样例输出：

2

样例解释：

删除1-2这条边，两个子树的同色联通块都为2，差的绝对值为0

删除2-3这条边，两个子树的同色联通块分别为1,2，差的绝对值为1

删除1-4这条边，两个子树的同色联通块分别为2,1，差的绝对值为1

故答案为0+1+1=2。

颜色连通块我没理解错的话应该是，不管多少个节点，只要连通且颜色相同就算一个连通块。示例有点不充分让人混淆，比如三个连通的颜色相同的节点a-b-c，是算两个连通块a-b和b-c还是算一个整体呢，我认为是一个整体。

思路和代码见图