题目大意
给定几条路,起点+终点+权值,给定几个指定起点,给定几个指定终点,问从起点到终点的最短路是多少
总结
最短路算法,需要避开坑(不然会超时),一开始我用Floyd算法求出任意两点之间的最短路然后再查询统计,求出起点到终点的最短路,结果发现可以在计算的过程中就统计好了,然后我又用迪杰克斯拉算法写了一份只需要每次计算从起点到第一个遇到的终点的最短路,取所有起点最短路最小值即为结果。
代码
Floyd:
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mem(a,n) memset(a,n,sizeof(a))
#define FRER() freopen("input.txt", "r", stdin)
#define FREW() freopen("output.txt", "r", stdin)
typedef long long LL;
using namespace std;
const int inf = 0x3f3f3f3f, maxn = 1000 + 10;
int road[maxn][maxn], ans, n;
int beg[maxn], en[maxn];
void Floyd()
{
for(int k=1;k<=n;k++) {
for (int i = 1; i <= n; i++) {
if(road[i][k]==inf) continue; // 一开始超时加一个这个就过了=_=
for (int j = 1; j <= n; j++) {
road[i][j] = min(road[i][j], road[i][k] + road[k][j]);
if (beg[i] && en[j]) ans = min(ans, road[i][j]);
}
}
}
}
void init()
{
mem(beg,0);
mem(en,0);
mem(road, 0x3f);
ans=inf;
}
int main()
{
//FRER();
int T, S, D;
int u, v, cost;
while(~scanf("%d%d%d",&T,&S,&D)) {
init();
for(int i=0;i<T;i++) {
scanf("%d%d%d", &u, &v, &cost);
road[u][v] = road[v][u] = min(cost, road[u][v]);
n=max(n,max(u,v));
}
int t;
for(int i=0;i<S;i++) {
scanf("%d",&t);
beg[t]=1;
}
for(int i=0; i<D;i++){
scanf("%d", &t);
en[t]=1;
}
Floyd();
printf("%d\n", ans);
}
ret
dijkstra
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mem(a,n) memset(a,n,sizeof(a));
#define FRER() freopen("input.txt", "r", stdin)
#define FREW() freopen("output.txt", "r", stdin)
typedef long long LL;
using namespace std;
const int inf = 0x3f3f3f3f, maxn = 1000 + 10;
int beg[maxn],en[maxn];
int road[maxn][maxn];
int vis[maxn];
int x[maxn];
int d[maxn];
int n;
int dj(int z) {
if(x[z]) return 0;
int mindis, v;
for(int i=1;i<=n;i++) {
vis[i]=0;
d[i]=road[z][i];
}
vis[z]=1;
for(int k=1; k<n;k++) {
mindis=inf;
for(int i=1;i<=n;i++) {
if(mindis>d[i]&&!vis[i]) {
mindis=d[i];
v=i;
}
}
vis[v] = 1;
if(x[v]) return d[v];
for(int j=1;j<=n;j++) {
if(!vis[j]&&d[j]>d[v]+road[v][j]) {
d[j]=d[v]+road[v][j];
}
}
}
}
int main()
{
// FRER();
int T, S, D;
int u, v, w;
while(~scanf("%d%d%d",&T,&S,&D)){
n=-1;
mem(road,0x3f);
mem(x,0);
mem(d,0);
for(int i=0;i<T;i++){
scanf("%d%d%d",&u,&v,&w);
if(road[u][v]>w) road[u][v]=road[v][u]=w;
n=max(n,max(u,v));
}
for(int i=1;i<=S;i++){
scanf("%d",&beg[i]);
}
for(int j=1;j<=D;j++)
{
int t;
scanf("%d",&t);
x[t]=1;
}
int ans=inf;
for(int i=1;i<=S;i++){
ans=min(dj(beg[i]), ans);
}
printf("%d\n", ans);
}
return 0;
}