图的遍历dfs
程序员文章站
2022-03-03 11:36:24
...
例题:
输入n,m,n表示有n个城市,m表示有m个行,每行有三个数x,y,c表示x到y的距离为c,若没记录则没有通路,求到n的最短距离
#include<cstdio>
#include<cstring>
//#include<iostream>
using namespace std;
const int inf=0x3f3f3f3f;
int min=inf;
int book[101],n,e[101][101];//e[x][y]=c表示下,到y的距离为c
//套路定义
void dfs(int cur,int dis){
int j;
if(dis>min)return ;//不满足最短条件
if(cur==n) {
if(dis<min)min=dis;
return;
}//取最小值
for(j=1;j<=n;j++){
if(e[cur][j]!=inf&&book[j]==0){
book[j]=1;//标记
dfs(j,dis+e[cur][j]);//搜索
book[j]=0;//返回上一步时要把归零
}
}
return ;
}
int main()
{
memset(e, inf, sizeof(e));//图距离
int i,j,m,a,b,c;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
if(i==j)e[i][j]=0;
for(i=1;i<=m;i++) {
scanf("%d%d%d",&a,&b,&c);//存图
e[a][b]=c;//e[b][a]=c;//是否双向图
}
book[1]=1;
dfs(1,0);
printf("%d",min);
return 0;
}
那么这是默认1到五,假如要一到三呢?只需稍稍微微改一下即可算到任意距离
#include<cstdio>
#include<cstring>
//#include<iostream>
using namespace std;
const int inf=0x3f3f3f3f;
int min=inf;
int book[101],n,e[101][101];//e[x][y]=c表示下,到y的距离为c
//套路定义
void dfs(int cur,int b,int dis){
int j;
if(dis>min)return ;//不满足最短条件
if(cur==b) {
if(dis<min)min=dis;
return;
}//取最小值
for(j=1;j<=b;j++){
if(e[cur][j]!=inf&&book[j]==0){
book[j]=1;//标记
dfs(j,b,dis+e[cur][j]);
book[j]=0;
}
}
return ;
}
int main()
{
memset(e, inf, sizeof(e));//图距离
int i,j,m,a,b,c;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
if(i==j)e[i][j]=0;
for(i=1;i<=m;i++) {
scanf("%d%d%d",&a,&b,&c);//存图
e[a][b]=c;//e[b][a]=c;//是否双向图
}
book[1]=1;
int abc;
scanf("%d",&abc);
dfs(1,abc,0);
printf("%d",min);
return 0;
}
是不是很神奇 上一篇: 用DFS 遍历图
下一篇: 图的深度优先遍历与广度优先遍历