hdu-1142-A Walk Through the Forest

#include<iostream>
#include<cstring>
using namespace std;
const int MAX=1001;
const int INF=99999999;
int m,n;
int map[MAX][MAX];
int dis[MAX];
int use[MAX];
void Init()
{
memset(use,0,sizeof(use));
for(int i=1; i<MAX; i++)
for(int j=1; j<MAX; j++)
map[i][j]=map[j][i]=INF;
}
void dijkstra(int s)
{
bool vis[MAX];
memset(vis,false,sizeof(vis));
int pos=s;
vis[pos]=true;
dis[pos]=0;
for(int i=1; i<=n; i++)
if(i!=pos)
dis[i]=map[pos][i];
for(int i=1; i<n; i++)
{
int min=INF;
for(int j=1; j<=n; j++)
{
if(!vis[j]&&dis[j]<min)
{
min=dis[j];
pos=j;
}
}
vis[pos]=true;
for(int k=1; k<=n; k++)
{
if(!vis[k]&&dis[k]>dis[pos]+map[pos][k])
dis[k]=dis[pos]+map[pos][k];
}
}
}
int dfs(int x)
{
int sum=0;
if(x==2)
return 1;
if(use[x])
return use[x];
for(int i=1; i<=n; i++)
{
if(map[x][i]!=INF&&dis[x]>dis[i]&&x!=i)
{
sum+=dfs(i);
}
}
use[x]=sum;
return use[x];
}
int main()
{
int a,b,time;
while(cin>>n&&n!=0)
{
cin>>m;
Init();
for(int i=1; i<=m; i++)
{
cin>>a>>b>>time;
if(map[a][b]>time||map[b][a]>time)
map[a][b]=map[b][a]=time;
}
dijkstra(2);
cout<<dfs(1)<<endl;
}
}