逛动物园
描述:Jack周日去动物园看“猩猩”,为了直达目的地,Jack想根据动物园地图计算从园区入口到“猩猩”馆的最短路径是多长?
输入: 输入中第一行第一个数为动物园展览馆总数n(包括0号馆),第二个数为“猩猩”馆的数字编号(入口编号为0号馆)。后面的输入是一个n×n的矩阵,表示动物园地图,其中,矩阵节点之间用一个空格隔开,矩阵节点的值n[i][j]代表i号馆到j号馆的距离,0表示馆间不通或无效路径。
输入:
5 4
0 7 3 10 15
7 0 5 13 12
3 5 0 5 10
10 13 5 0 11
15 12 10 11 0
输出 :13(园区入口到“猩猩“馆的最短路径。)
样例输入:
2 1
0 5
5 0
样例输出 :
5
知识点 :图的广度优先遍历 ,Dijkstra算法。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
int MinDistance(vector<vector<int> > &vec,const int &end){
int m=vec.size();
int k=0;
vector<bool> visited;//记录各馆是否被访问
visited.resize(m);
vector<int> dist;//从0馆到各馆的路径
dist.resize(m);
for(int j=1;j<m;j++)//给dist[]赋初值
dist[j]=vec[0][j];
visited[0]=true;//一开始0馆就已被访问
while(!visited[end]){//直到目标馆被访问
int mindis=INT_MAX;
for(int j=1;j<m;j++){ //找出未访问的馆中路径最短的馆号
if((dist[j]<mindis)&&(!visited[j])){
mindis=dist[j];
k=j;
}
}
visited[k]=true;//将此路径最短的馆标记为已访问
for(int j=1;j<m;j++){ //Dijkstra算法计算带权最短路径
if(!visited[j]){
if(dist[k]+vec[k][j]<dist[j])
dist[j]=dist[k]+vec[k][j];
}
}
}
return dist[end];//返回到目标馆的最短路径值
}
};
int main(){
int tol,num;
int i,j;
while(cin>>tol>>num){
vector<vector<int> > vec(tol, vector<int>(tol));
for(i=0;i<tol;i++){
for(j=0;j<tol;j++)
cin>>vec[i][j];
}
Solution so;
cout << so.MinDistance(vec,num)<< endl;
}
system("pause");
return 0;
}