哈理工oj (acm.hrbust.edu.cn) 1347 森林中的路 [dijkstra + DP]

题目连接：

http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1347

代码：

/*顶点从0开始标号，工作地点在0,家在1。
 * 先用dijkstra求出顶点1到每个点的最短路径。
 * 结果记录在dist[]中。
 * 然后ＤＰ思想求解答案。
 * ＤＦＳ搜索每一个顶点，要求这个点到1号点的距离大于前一个点到1号点的距离。
 * 所谓前一个点就是从那个点搜到当前点的点。
 * 每一步结果保存在ans[]中
 */
#include<stdio.h>
#include<string.h>
#include<limits.h>

#define MAXN 1001

int map[MAXN][MAXN];
int s[MAXN]; //标记数组，在dijstra中标记是否找到当前点的最小距离。在dfs中标记是否搜到过该点。
int dist[MAXN];
int ans[MAXN];
int n, m;

void dijkstra()
{
	for(int i = 0; i < n; i++) {
		s[i] = 0;
		dist[i] = map[1][i];
	}
	s[1] = 1;
	dist[1] = 0;
	int u = 1;
	for(int i = 0; i < n-1; i++) {
		int min = INT_MAX;
		for(int j = 0; j < n; j++) {
			if(s[j] == 0 && dist[j] < min) {
				min = dist[j];
				u = j;
			}
		}
		s[u] = 1;
		for(int j = 0; j < n; j++) {
			if(s[j] == 0 && map[u][j] < INT_MAX && map[u][j] + dist[u] < dist[j]) {
				dist[j] = map[u][j] + dist[u];
			}
		}
	}
}

int dfs(int np)
{
	if(s[np] == 1) return ans[np];
	ans[np] = 0;
	s[np] = 1;
	for(int i = 0; i < n; i++) {
		if(map[i][np] < INT_MAX && dist[i] > dist[np]) {
			ans[np] += dfs(i);
		}
	}
	return ans[np];
}

int main()
{
	while(scanf("%d", &n) != EOF) {
		if(n == 0) break;
		scanf("%d", &m);
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				map[i][j] = INT_MAX;
			}
		}
		for(int i = 0; i < m; i++) {
			int a, b, c;
			scanf("%d %d %d", &a, &b, &c);
			map[a-1][b-1] = map[b-1][a-1] = c;
		}
		dijkstra();
		memset(s, 0, sizeof(s));
		s[0] = 1;
		ans[0] = 1;
		printf("%d\n", dfs(1));
	}
	return 0;
}