UVA 10917 Walk Through the Forest 迪杰斯特拉+记忆化搜索

UVA 10917

题意：有n个地点，有m条带权无向边连接这些点，家的编号为2，公司编号为1，起点是公司，终点是家，只能沿着这样一条路（A，B）走，存在一条从B出发回家的路径，比所有从A出发回家的路径都短，求有多少条这样的路径。

思路：先求从家到每个点的最短路保存在d数组里，如果从A出发，那么只有d[ B ]<d[ A ]才能走B，那么很容易想到记忆化搜索，设dp[ i ]为从 i 出发回家的答案，那么dp[ i ]+=sum(dp[ v ])（当d[ v ]<d[ i ]时）。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn=1e3+5;
const int inf=1e9;
vector<int>G[maxn],dis[maxn];
int n,vis[maxn],d[maxn],dp[maxn];
struct node
{
	int u,dist;
	bool  friend operator<(node c,node d)
	{
		return c.dist>d.dist;
	}
}no,ne;
void init()
{
	for(int i=0;i<=n;i++)
	{
		G[i].clear();
		dis[i].clear();
		vis[i]=dp[i]=0;
		d[i]=inf;
	}
}
void dij()
{
	priority_queue<node>q;
	no.u=2,no.dist=0;
	d[2]=0;
	q.push(no);
	while(!q.empty())
	{
		no=q.top();q.pop();
		if(vis[no.u])
		continue;
		vis[no.u]=1;
		int u=no.u;
		for(int i=0;i<G[u].size();i++)
		{
			int v=G[u][i];
			if(d[v]>d[u]+dis[u][i])
			{
				d[v]=d[u]+dis[u][i];
				ne.u=v;
				ne.dist=d[v];
				q.push(ne);
			}
		}
	}
}
int dfs(int u)
{
	if(dp[u])
	return dp[u];
	if(u==2)
	return dp[u]=1;
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(d[u]>d[v])
		dp[u]+=dfs(v);	
	}
	return dp[u];
}
int main()
{
	int i,u,v,dist,m;
	while(~scanf("%d",&n)&&n)
	{
		init();
		scanf("%d",&m);
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&u,&v,&dist);
			G[u].push_back(v);
			G[v].push_back(u);
			dis[u].push_back(dist);
			dis[v].push_back(dist);
		}
		dij();
		printf("%d\n",dfs(1));
	}
}