hdu 1142 A Walk Through the Forest(最短路径+dfs)

http://acm.hdu.edu.cn/showproblem.php?pid=1142

大致题意：若a，b之间有边，走ab边的条件是b到目的地的最短路径小于a到目的地的最短路径。按这个要求走，问最后从1到2的路径数目。

思路：先预处理到2的最短路径。然后从1dfs寻求路径数目。有点像DAG上的动态规划求路径数目。

#include <stdio.h>
#include <algorithm>
#include <set>
#include <map>
#include <vector>
#include <math.h>
#include <string.h>
#define LL long long
#define _LL __int64

using namespace std;
const int INF = 0x3f3f3f3f;

int n,m;
int mapp[1100][1100];
int dis[1100];
int vis[1100];
int cnt[1100];

void init()
{
	for(int i = 1; i <= n; i++)
	{
		for(int j = 1; j <= n; j++)
		{
			if(i != j)
				mapp[i][j] = INF;
			else mapp[i][j] = 0;
		}
	}
}

void dijstra(int s)
{
	memset(vis,0,sizeof(vis));
	memset(dis,INF,sizeof(dis));

	vis[s] = 1;

	for(int i = 1; i <= n; i++)
		dis[i] = mapp[s][i];

	for(int i = 1; i < n; i++)
	{
		int min = INF;
		int pos;

		for(int j = 1; j <= n; j++)
		{
			if(!vis[j] && min > dis[j])
			{
				min = dis[j];
				pos = j;
			}
		}

		vis[pos] = 1;

		for(int j = 1; j <= n; j++)
		{
			if(!vis[j] && dis[j] > dis[pos] + mapp[pos][j])
				dis[j] = dis[pos] + mapp[pos][j];
		}
	}
}

//求1到2的路径数目
int dfs(int s)
{
	if(s == 2)
		return cnt[2] = 1;

	if(cnt[s] >= 0)
		return cnt[s];
	else
	{
		int res = 0;
		for(int i = 1; i <= n; i++)
		{
			if(mapp[s][i] != INF && dis[i] < dis[s])
				res += dfs(i);
		}
		return cnt[s] = res;
	}
}

int main()
{
	int u,v,w;
	while(~scanf("%d",&n) && n)
	{
		scanf("%d",&m);
		init();

		while(m--)
		{
			scanf("%d %d %d",&u,&v,&w);
				mapp[u][v] = mapp[v][u] = w;
		}

		dijstra(2); //预处理最短路径
		
		memset(cnt,-1,sizeof(cnt));
		int ans = dfs(1);
		printf("%d\n",ans);
	}
	return 0;
}