华为机试题：逛动物园

逛动物园

描述：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;
}