Lucky Tree

题目：http://codeforces.com/problemset/problem/109/C

题意：一棵树n个节点，组成一个图，每条边都有权值，对于i、j、k三个数，计算所有的 i 到 j 和 i 到 k 的路径上有幸运数的三元组个数。

题解：用并查集来做，对于任意节点 i ，找出与 i 连通的且某段路径的权值是幸运数的点的个数num，则总个数即是1~n的num*(num-1）.但是因为有些点之间是间接连接的，无法直接判断，所以用并查集来更新，计算出每个点的根节点的num值（额，每个点的num值也无法直接计算，因为有的间接连接的点也符合，但可以通过两点的权值计算点与 i 的不符合的点的个数，用n减去它就是所需的  num）。在后面计算总数时，遍历每个点时用其根节点的num值来计算。

 

这两天整个人敲题都是迷的.........

#include <map>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <limits>
#include <cstdio>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Scc(c) scanf("%c",&c)
#define Scs(s) scanf("%s",s)
#define Sci(x) scanf("%d",&x)
#define Sci2(x, y) scanf("%d%d",&x,&y)
#define Sci3(x, y, z) scanf("%d%d%d",&x,&y,&z)
#define Scl(x) scanf("%I64d",&x)
#define Scl2(x, y) scanf("%I64d%I64d",&x,&y)
#define Scl3(x, y, z) scanf("%I64d%I64d%I64d",&x,&y,&z)
#define Pri(x) printf("%d\n",x)
#define Prl(x) printf("%I64d\n",x)
#define Prc(c) printf("%c\n",c)
#define Prs(s) printf("%s\n",s)
#define For(i,x,y) for(int i=x;i<y;i++)
#define For_(i,x,y) for(int i=x;i<=y;i++)
#define FFor(i,x,y) for(int i=x;i>y;i--)
#define FFor_(i,x,y) for(int i=x;i>=y;i--)
#define Mem(f, x) memset(f,x,sizeof(f))
#define LL long long
#define ULL unsigned long long
#define MAXSIZE 100005
#define INF 0x3f3f3f3f
const int mod=1e9;
const double PI = acos(-1.0);

using namespace std;

int pre[MAXSIZE];
int num[MAXSIZE];
int find(int x)
{
    if(x!=pre[x])
        return pre[x]=find(pre[x]);
    return x;
}
int check(int n)
{
    while(n)
    {
        if(n%10!=4&&n%10!=7)
            return 0;
        n/=10;
    }
    return 1;
}
void hhh(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)
        return ;
    pre[y]=x;
    num[x]+=num[y];
}
int main()
{
    int n;
    Sci(n);
    For_(i,1,n)
    {
        num[i]=1;
        pre[i]=i;
    }

    int a,b,c;
    For_(i,1,n-1)
    {
        Sci3(a,b,c) ;
        if(!check(c))
        {
            hhh(a,b);
        }
    }
    LL sum=0;
    For_(i,1,n)
    {
        int tmp=n-num[ find( i) ];
        sum+=(LL)(tmp-1)*tmp;//这个地方的（LL）不能省，乘积结果应该是会超出int范围
    }
    Prl(sum);
    return 0;
}