[51nod1253]Kundu and Tree

　　树包含N个点和N-1条边。树的边有2中颜色红色('r')和黑色（'b')。给出这N-1条边的颜色，求有多少节点的三元组(a,b,c)满足：节点a到节点b、节点b到节点c、节点c到节点a的路径上，每条路径都至少有一条边是红色的。
　　注意(a,b,c), (b,a,c)以及所有其他排列被认为是相同的三元组。输出结果对1000000007取余的结果。

　Input
　　第1行：1个数N（1 <= N <= 50000）
　　第2 - N行：每行2个数加一个颜色，表示边的起始点和结束的以及颜色。
　Output
　　输出1个数，对应符合条件的3元组的数量。

 

 

　　奇特的思路。。黑边没有什么用...就把黑边相连的合成一个联通块。

　　只要选择的三个点在不同的联通块内就合法了...

　　答案就变成 随便选三个点的方案数 - 三个点在同个联通块内的方案数 - 两个点在同个联通块内的方案数

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstdlib>
#define ll long long
#define ull unsigned long long
#define ui unsigned int
//#define d double
#define ld long double
const int maxn=50023,modd=1000000007;
int c[maxn][4],fa[maxn],sz[maxn];
int i,j,k,n,m;

int ra,fh;char rx;
inline int read(){
    rx=getchar(),ra=0,fh=1;
    while(rx<'0'&&rx!='-')rx=getchar();
    if(rx=='-')fh=-1,rx=getchar();
    while(rx>='0')ra=ra*10+rx-48,rx=getchar();return ra*fh;
}
inline int getfa(int x){return fa[x]!=x?fa[x]=getfa(fa[x]):x;}
inline void MOD(int &x){if(x>=modd)x-=modd;}
inline void UPD(int &x){if(x<0)x+=modd;}
int main(){
    n=read();
    c[0][0]=1;register int i,j;
    for(i=1;i<=n;i++){
        c[i][0]=sz[i]=1,fa[i]=i;
        for(j=1;j<=3;j++)MOD(c[i][j]=c[i-1][j-1]+c[i-1][j]);//,printf(" %d %d   c:%d\n",i,j,c[i][j]);
    }
    int x,y;
    for(i=1;i<n;i++){
        x=read(),y=read();
        if(getchar()=='b')x=getfa(x),y=getfa(y),fa[x]=y,sz[y]+=sz[x];
    }
    int ans=c[n][3];
    for(i=1;i<=n;i++)if(fa[i]==i)
        UPD(ans-=(c[sz[i]][3]+1ll*(n-sz[i])*c[sz[i]][2])%modd);
    printf("%d\n",ans);
}