sdutacm 1566 幸运数（bfs)

http://acm.sdut.edu.cn/web/problem.php?action=showproblem&problemid=1566

这是上个星期测试赛的最后一个题目，那天真是悲剧了，一个也没做出来，；

思路：采用VONGONG的思路做的，用bfs搜索来找：

View Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
int map[55][55];
int p(int w)
{
    int temp;
    while(w!=0)
    {
      temp=w%10;
      if(temp!=4&&temp!=7)
      return 0;
      w=w/10;
    }
    return 1;
}
int bian(int w,int n)
{
    int visit[60],ans[60],ba[60],i;
    memset(visit,0,sizeof(visit));
    memset(ba,0,sizeof(ba));
    visit[w]=1;
    ans[1]=w;
    int a,b,te;
    a=1;
    b=2;
    int ss=0;
    while(a<b)
    {
       te=ans[a];
       a++;
       if(ba[te])
       ss++;
       for(i=1;i<=n;i++)
        if(!visit[i]&&map[te][i]!=-1)
        {
           if(map[te][i]==1||ba[te]==1)
           ba[i]=1;
           visit[i]=1;
           ans[b++]=i;
        }
    }
    return ss;
}
int main()
{
    int a,b,len,sum,n,i;
   while(scanf("%d",&n)!=EOF)
   {
       sum=0;
       memset(map,-1,sizeof(map));
       for(i=1;i<n;i++)
       {
           scanf("%d%d%d",&a,&b,&len);
           len=p(len);
           map[a][b]=map[b][a]=len;
       }
       int tt;

       for(i=1;i<=n;i++)
       {
          tt=bian(i,n);
          if(tt>1)
          sum+=tt*(tt-1);
       }
       printf("%d\n",sum);
   }
   return 0;
}