51nod 1253 Kundu and Tree

容斥原理+并查集。。第一次见这样的题，涨姿势了，不看题解想不出。
看了题解后才发现很简单。
题解：
http://blog.csdn.net/crybymyself/article/details/68062911
http://www.cnblogs.com/Stomach-ache/p/3931848.html

一个讲的细，用dfs求的连通分量。
一个讲的粗略，用的并查集。。
思路：首先考虑所有的黑边是没用的，所以用并查集构造出一个个全是黑边的连通块。如果有满足要求的3个点a,b,c，若把a和b之间的红色边去掉，a和b将不连通。同理，如果把a到b，b到c，c到a中的红色边都去掉，那么a，b，c将属于不同的连通分量。假设有k个连通块，每个连通块有cnt[i]个点，这就转化成从k个连通块中选择a，b，c三个点且两两不再同一连通块中的方案数。
res = 所有点选三个点的方案数-三个点在同一个块中的方案数-两个点在同一个连通块另一个点在别的块方案数。。

using System;
using System.IO;

namespace timeless
{
    class Program
    {
        private static long MAXN = 50010;
        private static long[] f;
        private static long[] size;
        private static long[] cnt;
        private static long mod = (long)1e9 + 7;
        static void Main(string[] args)
        {
            StreamReader sr = new StreamReader(Console.OpenStandardInput());
            StreamWriter sw = new StreamWriter(Console.OpenStandardOutput());
            f = new long[MAXN];
            size = new long[MAXN];
            cnt = new long[MAXN];
            long n = Convert.ToInt64(sr.ReadLine());
            f[0] = size[0] = cnt[0] = 0;
            for (int i = 1; i <= n; ++i)
            {
                f[i] = i;
                size[i] = 1;
                cnt[i] = 0;
            }
            string[] str;
            long s,e;
            char ch;
            for (int i = 0; i < n-1; ++i)
            {
                str = sr.ReadLine().Split(' ');
                s = Convert.ToInt64(str[0]);
                e = Convert.ToInt64(str[1]);
                ch = str[2].ToCharArray()[0];
                if (ch == 'b')
                    merge(s, e);
            }
            long sum = 0;
            long cntLen = 0;
            long res = 0;
            for (int i = 1; i <= n; ++i)
            {
                if (f[i] == i)
                {
                    cnt[cntLen++] = size[i];
                    sum += size[i];
                    //在一个连通块里选择三个点的情况
                    res -= calC(size[i]);
                }
            }
            res += calC(sum);//在所有点中选三个点的情况
            for (int i = 0; i < cntLen; ++i)
            {
                //在一个块里选两个点的情况
                res -= cnt[i] * (cnt[i]-1) / 2 * (sum - cnt[i]);
            }
            res = res % mod;
            sw.WriteLine(res);
            sw.Flush();
            sw.Close();
            sr.Close();
        }

        public static long calC(long x)
        {
            return x * (x - 1) * (x - 2) / 6;
        }

        public static void merge(long x, long y)
        {
            x = getf(x);
            y = getf(y);
            if (x != y)
            {
                size[x] += size[y];
                f[y] = x;
            }
        }

        public static long getf(long x)
        {
            if (x == f[x])
                return x;
            f[x] = getf(f[x]);
            return f[x];
        }
    }
}