今天碰到一道算法题。关于迪杰斯特拉算法
有关该算法的介绍链接。点击打开链接
http://www.cnblogs.com/biyeymyhjob/archive/2012/07/31/2615833.html
题目如下。
小兔的叔叔从外面旅游回来给她带来了一个礼物,小兔高兴地跑回自己的房间,拆开一看是一个棋盘,小兔有所失望。不过没过几天发现了棋盘的好玩之处。从起点(0,0)走到终点(n,n)的最短路径数是C(2n,n),现在小兔又想如果不穿越对角线(但可接触对角线上的格点),这样的路径数有多少?小兔想了很长时间都没想出来,现在想请你帮助小兔解决这个问题,对于你来说应该不难吧!
1 3 12 -1
1 1 2 2 3 10 3 12 416024
这是典型的无向图的题,即没有负方向。找出最短路径。
代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
double map[1005][1005],dis[1005];//距离可能为小数。
int nearly[1005],aim[1005], T, S, D;
bool vis[1005];
int maxcity=0;
void findv()
{
int thmin,pos=0;
memset(vis, false, sizeof(vis));
vis[0] = 1;//0是原点
for (int i = 0; i <= maxcity; i++)
dis[i] = map[0][i];
for (int i = 1; i <= maxcity; i++)//遍历每一个起点.
{
for (int j = 1; j <= maxcity; j++)//遍历后面所有的终点。
{//每次从最小距离开始的城市找下一个城市。
if (!vis[j] && dis[j] > (dis[pos] + map[pos][j]))//如果没有访问过,并且map中有值
{//if(dis[j]!=0)//如果距离不为0,也就是已经记录过某个值
dis[j] = (map[pos][j] + dis[pos]);//>= dis[j] ? dis[j] : (map[i][j] + dis[i]);
//else dis[j] = map[i][j]+dis[i];//如果没有记录过值
}
}
thmin = 0xfffffff;
for (int i = 0; i <= maxcity; i++)//遍历距离。找到最短的距离,设置那个点为已访问。
{
if (!vis[i] &&dis[i] < thmin)//没走过的城市的最小的距离
thmin = dis[i], pos = i;
}
vis[pos] = 1;
}
}
int main()
{
while (cin >> T>> S >> D)
{
memset(dis, 0, sizeof(dis));
for (int i = 0; i < 1005; i++)//给map统一赋值。城市本身距离为0;其他所有不相关城市距离为无限0xffffff。
{
for (int j = 0; j < 1005; j++)
map[i][j] = 0xffffff;
map[i][i] = 0;
}
memset(vis, 0, sizeof(vis));
//cout << dis[3] << endl << map[1][3] << endl;
if (T == -1)return 0;
int from, to,dist;
maxcity = 0;double mindis=0xffffff;
for(int i=0;i<T;i++)
{
cin >> from >> to>>dist;
maxcity = max(max(maxcity, from), to);//记录最大的城市名称。以免遍历1005次。
if(map[from][to]>dist)
map[from][to] = map[to][from] = dist;//双向的距离应该是一样的。;
}
for (int i = 1; i <= S; i++)
{
cin >> nearly[i];
map[0][nearly[i]] = map[nearly[i]][0] = 0;//起点不一定只有一个,把0看成真正的起点,0到任意第一个城市的
//距离都设0,其余为无限
}
for (int i = 1; i <= D; i++)
cin >> aim[i];
findv();
for (int i = 1; i <= D; i++)
mindis = min(mindis, dis[aim[i]]);//在想去的城市里找到最小的那个距离
cout << mindis << endl;
}
return 0;
}