DFS求有中转站数限制的路径最少花费

Description Time Limit: 1sec Memory Limit:256MB

某天，X要从1地走到n地，途中可能会经过许多中转站。某些中转站之间连接着双向道路，这些双向道路具有一定长度，可以认为走这些路的花费即为路的长度。此外，由于进入中转站有着许多麻烦的手续，因此X希望在中途经过不超过k个中转站的情况下到达b地（从1地出发以及进入n地均不计入耗费中转站的次数内），并求出最短的花费。

注意起点和终点不计入总的中转站花费。

Input

第一行T表示数据个数.,T<=10

以下T组数据中，第一行三个数N,m,k。N为点数，m为边数，k为途径中转站的最多数量。N<=10,0<=m（边数）<N*(N-1)/2，k<=N-2

下面m 行，每行三个整数x,y<=N，d<=100并且x,y从1开始。表示有一条边连接x,y，并且距离为d

无重边与自环

Output

输出一个数，为最短距离，不可达输出-1

每组数据一行

Sample Input

2
3 2 0
1 2 1
2 3 1
3 2 1
1 2 1
2 3 1

Sample Output

-1
2

分析

思想：因为有中转站数量的限制，所以我用的是深度优先搜索DFS，然后把每条符合要求的路径放入优先级队列中，最后取出花费最少的路径即可。

用矩阵存储边信息

自己到自己的距离为0，不可直接到达的距离为INF（无穷大）。

priority_queue<int,vector<int>,greater<int> > q; //此时为小顶堆，因为默认是大顶堆,less<int> 
void init(){//初始化
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i==j) e[i][j]=0;//自己到自己的距离为0
            else e[i][j]=INF;//初始为无穷大，说明此事城市i和城市j之间还没有路相通
        }
    }
    //在这里清空最小堆
	while(!q.empty()) q.pop(); 
}

DFS找最短路

记录当前节点、花费数和经过的节点数turn，根据判断当前节点是否为目标节点以及turn-1是否小于等于最大可经过的中转站数，来判断是否把路径放入优先级队列。

void dfs(int now,int step,int turn){
	if(turn-1<=k&&now==n){
		q.push(step);
		return;
	}
	else if(turn-1>k){
		return;
	}
    
    vis[now] = 1;
    for(int j=1;j<=n;j++){
        if(e[now][j]!=INF&&!vis[j]){
            vis[j]=1;
            dfs(j,step+e[now][j],turn+1);
            vis[j]=0;
        }
    }
    return;
}

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#define INF 99999999
using namespace std;
const int maxn=105;
int e[maxn][maxn];
int vis[maxn];
int minx=INF;
int sx,ex;
int n,m,k;
priority_queue<int,vector<int>,greater<int> > q; //此时为小顶堆，因为默认是大顶堆,less<int> 
void init(){//初始化
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i==j) e[i][j]=0;//自己到自己的距离为0
            else e[i][j]=INF;//初始为无穷大，说明此事城市i和城市j之间还没有路相通
        }
    }
    //在这里清空最小堆
	while(!q.empty()) q.pop(); 
}

void dfs(int now,int step,int turn){
	if(turn-1<=k&&now==n){
		q.push(step);
		return;
	}
	else if(turn-1>k){
		return;
	}
    
    vis[now] = 1;
    for(int j=1;j<=n;j++){
        if(e[now][j]!=INF&&!vis[j]){
            vis[j]=1;
            dfs(j,step+e[now][j],turn+1);
            vis[j]=0;
        }
    }
    return;
}
/*void dfs(int now,int sum,int turn){
	if(turn>k) return;
	if(now==n){
		q.push(step);
		return;
	}
}*/
int main()
{
    int a,b,c;
    int T;
    cin>>T;
    for(int t = 0;t < T;t++){
    	scanf("%d %d %d",&n,&m,&k);
    	init();
	    for(int i=1;i<=m;i++){
	        scanf("%d %d %d",&a,&b,&c);
	        e[a][b]=c;
	        e[b][a]=c;
	    }
	    if(k == 0&&e[1][n]==INF){
	    	printf("-1\n");
	    }
	    else{
	    	memset(vis,0,sizeof(vis));
		    vis[1]=1;
		    dfs(1,0,0);
		    if(q.empty()){
		    	printf("-1\n");
		    }
		    else{
		    	printf("%d\n",q.top());
		    	/*while(!q.empty()){
		    		printf("%d\n",q.top());
		    		q.pop();
		    	}*/
			}
	    } 

    }

}