F maki和tree 换根dp

F maki和tree
题意：给你一颗树，每个节点被染成黑色和白色中的一种。问你有多少对<u,v>在u到v的路径上只有一个黑色点。
题解：正解是并查集，但是我觉得能用换根dp写，现场赛没写出来，但是dp的大致思路是对的，赛后改了改，好久没写了，换根都有细节出错了。但是一眼还是换根dp。最后还是补了这个题。设dp[u][0]代表到节点u一个黑色点都没经过的点对数，dp[u][1]代表到节点u经过一个黑色点的点对数。然后换根，写完了。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5+10;
struct node
{
    ll l,r;
};
char a[maxn];
ll dp[maxn][2],ans = 0;
vector<int>G[maxn];
vector<node> sf[maxn];
void dfs(int u,int fa)
{
    for(auto v:G[u])
    {
        if(v==fa) continue;
        dfs(v,u);
        if(a[u]=='B') { if(a[v]=='W') dp[u][1] += dp[v][0]+1 ;  dp[u][0] = 0; }
        if(a[u]=='W') { if(a[v]=='W') dp[u][0] += dp[v][0]+1 , dp[u][1] += dp[v][1]; else dp[u][1] += dp[v][1] + 1; }
    }
}
void dfs2(int u,int fa)
{
    dp[u][0] = dp[u][1] = 0;
    sf[u].push_back(node{dp[u][0],dp[u][1]});
    for(auto v:G[u])
    {
       if(a[u]=='B') { if(a[v]=='W') dp[u][1] += dp[v][0]+1 ; dp[u][0] = 0;  }
        if(a[u]=='W') { if(a[v]=='W') dp[u][0] += dp[v][0]+1 , dp[u][1] += dp[v][1]; else dp[u][1] += dp[v][1] + 1; }
        sf[u].push_back(node{dp[u][0],dp[u][1]});
    }
    ans += dp[u][1];
    node tmp = node{0,0};
    for(int i=G[u].size()-1;i>=0;i--)
    {
        int v = G[u][i];
        dp[u][0] = tmp.l + sf[u][i].l ; dp[u][1] = tmp.r + sf[u][i].r;
        if(a[u]=='B') {if(a[v]=='W')tmp.r += dp[v][0] +1 ;  }
        if(a[u]=='W') { if(a[v]=='W') tmp.l += dp[v][0] + 1, tmp.r += dp[v][1] ; else tmp.r += dp[v][1] + 1; }
       if(v!=fa) dfs2(v,u);
    }
}
int main()
{
     int n; cin>>n;
     string s; cin>>s;
     for(int i=0;i<n;i++) a[i+1] = s[i];
     memset(dp,0,sizeof(dp));
     int u,v;
     for(int i=1;i<n;i++)
     {
          cin>>u>>v;
          G[u].push_back(v);
          G[v].push_back(u);
     }
     dfs(1,0);
     dfs2(1,0);
    cout<<ans/2<<endl;
}