zcmu-1398: 工程(dfs）

题目描述

某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

输入

本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。

输出

对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1.

样例输入

3 3

0 1 1

0 2 3

1 2 1

0 2

3 1

0 1 1

1 2

样例输出

2 -1

注释：注意初始化为零。

代码：

#include<bits/stdc++.h>
using namespace std;
#define maxn 300
int map1[maxn][maxn];
int s[maxn][maxn];
int a1[maxn];
int n,m,a,b,x,s1,t1,mx;
void dfs(int w,int sum){
	if(sum>mx)return;
	if(w==t1){
		mx=min(mx,sum);
		return ;
	}
	for(int i=0;i<n;i++){
		if(map1[w][i]!=0&&a1[i]==0){
			a1[i]=1;
			dfs(i,sum+s[w][i]);
			a1[i]=0;
		}
	}
}
int main(){
	while(~scanf("%d %d",&n,&m)){
        memset(map1,0,sizeof(map1));
		memset(s,0,sizeof(s));
		memset(a1,0,sizeof(a1));
		for(int i=1;i<=m;i++){
			scanf("%d %d %d",&a,&b,&x);
			map1[a][b]=1;
			map1[b][a]=1;
			s[a][b]=x;
			s[b][a]=x;
		}
		scanf("%d %d",&s1,&t1);
		mx=10000000;
		a1[s1]=1;
		dfs(s1,0);
		if(mx==10000000){
			cout<<"-1"<<endl;
		}
		else{
		cout<<mx<<endl;
	}
	}
}