J - 单源最短路

原题

题目描述

给一个 n(1 \leq n \leq 2500)n(1≤n≤2500) 个点 m(1 \leq m\leq 6200 )m(1≤m≤6200) 条边的无向图，求 ss 到 tt 的最短路。

输入格式

第一行四个由空格隔开的整数 nn、mm、ss、tt。
之后的 mm 行，每行三个正整数 s_isi​、t_iti​、w_i(1 \leq w_i \leq 10 ^ 9)wi​(1≤wi​≤109)，表示一条从 s_isi​ 到 t_iti​ 长度为 w_iwi​ 的边。

输出格式

一个整数表示从 ss 到 tt 的最短路长度。数据保证至少存在一条道路

样例

Input

7 11 5 4
2 4 2
1 4 3
7 2 2
3 4 3
5 7 5
7 3 3
6 1 1
6 3 4
2 4 3
5 6 3
7 2 1

output

7

思路

板子题，单源最短路径(dij):选定一个起始点，求该点到图中任意点的距离。将图G的顶点集V分成两个子集U和V-U，U是已求得最短路径的顶点集。 引入一维数组Dist，记录各顶点所求的当前最短路径信息。

代码

#include<iostream>
#include<algorithm>
#include<string.h> 
using namespace std;
#define ll long long

ll f[2505][2505];
int a,b;
ll dis[2505];
int vis[2505];
ll MAX=9999999999999;
ll len;
int n,m;
int x,y; 
int  main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> n >> m>>x>>y;
	memset(f,1,sizeof(f));
	for(int i=1;i<=m;i++)
	{
		cin >> a >> b >> len;
		f[a][b]=min(f[a][b] ,len);
		f[b][a]=f[a][b];	
	}
	for(int i=1;i<=n;i++)
	{
		f[i][i]=0;
		dis[i]=f[x][i];
	}
	ll MIN;
	int k;
	for(int i=1;i<=n;i++)
	{
		MIN=MAX;
		for(int j=1;j<=n;j++)
		{
			if(vis[j] != 1 && dis[j]<MIN)
			{
				MIN=dis[j];
				k=j;
			}
		}
		vis[k]=1;
		for(int j=1;j<=n;j++)
			dis[j]=min(dis[j],dis[k]+f[k][j]);
	}
	cout<<dis[y];
	return 0;
 }