HDU1142——A Walk Through the Forest（spfa+记忆化搜索）

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<queue>
#include<algorithm>
#define inf 10000000
using namespace std;
struct node
{
int to,next,w;
}city[1000010];
int mark[1050];
int head[1050];
int dp[1050];
int dis[1050];
int n,m;int cnt;
void add(int a,int b,int c)
{
city[cnt].w=c;
city[cnt].to=b;
city[cnt].next=head[a];
head[a]=cnt++;
}
void spfa()
{
int l,i;
queue<int>q;
while(!q.empty()) q.pop();
mark[2]=1;
dis[2]=0;
q.push(2);
while(!q.empty())
{
l=q.front();
q.pop();
mark[l]=0;
for(i=head[l];i+1;i=city[i].next)
{
if(dis[city[i].to]>dis[l]+city[i].w)
{
dis[city[i].to]=dis[l]+city[i].w;
if(mark[city[i].to]==0)
{
q.push(city[i].to);
mark[city[i].to]=1;
}
}
}
}
}
int dfs(int st)
{
int sum=0;
if(dp[st]!=-1) return dp[st];
if(st==2) return 1;
for(int i=head[st];i+1;i=city[i].next)
{
  if(dis[st]>dis[city[i].to])
  sum+=dfs(city[i].to);
  }
dp[st]=sum;
  return sum;
}


int main()
{
while(scanf("%d",&n)!=EOF)
{
if(n==0) break;
scanf("%d",&m);
int x,y,z;
for(int i=1;i<=n;i++)
{
head[i]=-1;
dis[i]=inf;
dp[i]=-1;
}
memset(mark,0,sizeof(mark));
cnt=0;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
spfa();
// printf("%d\n",dis[1]);
int pp=dfs(1);
printf("%d\n",pp);
}
return 0;
}