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;
}