maki和tree （图论）详解

题目链接：https://ac.nowcoder.com/acm/contest/3002/F
题目描述：


给我启发最大的一张图：

解题思路
深搜统计每个黑点相连的白点的个数，一个黑点的ans为所有与它相连的白点个数加上，两两两条线之间的乘积，把每个黑点的ans相加就是答案。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
const int maxn=1e5+5;
char q[maxn];
vector<int>v[maxn];
ll sum;
void  dfs(int x,int fa)   ///常规dfs
{
    if(q[x]=='B')
    {
        return;
    }
    sum++;
    for(int i=0; i<v[x].size(); i++)
    {
        int y=v[x][i];
        if(y!=fa)
        {
            dfs(y,x);
        }
    }
}
ll w[maxn];
int main()
{
    int n;
    scanf("%d",&n);
    scanf("%s",q+1);

    for(int i=1; i<=n-1; i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);

    }
    ll ans=0;
    ll cnt=0;
    ll t1,t2;

    for(int i=1; i<=n; i++)
    {
        t1=t2=0;

        if(q[i]=='B')
        {
            for(int j=0; j<v[i].size(); j++)
            {
                sum=0;
                dfs(v[i][j],v[i][j]);
                w[++cnt]=sum;
                ans+=sum;
                t1+=sum;
                t2+=sum*sum;
            }
        }
         ans+=(t1*t1-t2)/2; 
         ///注意：(( a+b+c+d)^2-(a*a+b*b+c*c+d*d) )/2=a*b+a*c+a*d+b*c+b*d+c*d
    }

    printf("%lld\n",ans);
}