cow很热爱加班,他和朋友在凌晨一点吃完海底捞后又一个人回公司加班,为了多加班他希望可以找最短的距离回到公司。
深圳市里有N个(2 <= N <= 1000)个公交站,编号分别为1…N。深圳是大城市,公交车整天跑跑跑。公交站1是cow的位置,公司所在的位置是N。所有公交站中共有T (1 <= T <= 2000)条双向通道。cow对自己的导航能力不太自信,所以一旦开始,他总是沿着一条路线走到底。
cow为了锻炼未来的ACMer,决定让你帮他计算他到公司的最短距离。可以保证存在这样的路存在。
Input
第一行:两个整数:T和N
接下来T行:每一行都用三个空格分隔的整数描述一个轨迹。前两个整数是路线经过的公交站台。第三个整数是路径的长度,范围为1到100。
Output
一个整数,表示cow回到公司的最小距离。
Sample Input
5 5
1 2 20
2 3 30
3 4 20
4 5 20
1 5 100
Sample Output
90
本题在做的时候用的是迪杰斯特拉的方法,思路是在自己的位置上的距离为0,不在自己的位置上的设为无穷值,找到最小的距离,不断地进行更新,最终找到从起点到终点的最小值。
代码如下:
#include"cstdio"
#include"cstring"
#include"iostream"
#include"algorithm"
using namespace std;
#define N 1005
#define INF 0x3f3f3f3f
int e[N][N];
int dist[N];
bool visited[N];
int pre[N];
int t , n , m , s;
void init_e()
{
for(int i = 1 ; i <= n ; i ++)
{
for(int j = 1 ; j <= n ; j ++)
{
if(i == j)
{
e[i][j] = 0;
}
else
{
e[i][j] = INF;
}
}
}
}
void Dijkstra()
{
for(int i = 1 ; i <= n ; i ++)
{
dist[i] = e[1][i];
}
visited[s] = 1;
for(int j = 2 ; j <= n ; j ++)
{
int min = INF , u;
for(int i = 1 ; i <= n ; i ++)
{
if(!visited[i] && dist[i] < min)
{
min = dist[i];
u = i;
}
}
visited[u] = 1;
for(int i = 1 ; i <= n ; i ++)
{
if(!visited[i] && dist[u] + e[u][i] < dist[i])
{
dist[i] = dist[u] + e[u][i];
}
}
}
printf("%d\n",dist[n]);
}
int main()
{
while(~scanf("%d %d",&m,&n))
{
init_e();
int a , b , w;
for(int i = 0 ; i < m ; i ++)
{
scanf("%d %d %d",&a,&b,&w);
if(w < e[a][b])
{
e[a][b] = w;
e[b][a] = w;
}
}
Dijkstra();
}
return 0;
}