其实就是一道单源最短路的题目,就是想测试一下我的SPFA板子。
双向边看成单向WA2。。。
AC代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int ,int> pii;
const int INF = 0x3f3f3f3f;
const int N = 3e3+10;
const int M = 7e3+10;
struct EDGE{
int next;
int to;
int w;
}edge[M];
int head[N],dis[N];
bool inq[N];
int T,C,Ts,Te,cnt = 0;
void add(int u, int v, int w){
edge[++cnt].next = head[u];
edge[cnt].to = v;
edge[cnt].w = w;
head[u] = cnt;
}
void SPFA(){
memset(dis, INF, sizeof dis);
dis[Ts] = 0;
int p; queue<int > que;
que.push(Ts); inq[Ts] = true;
while(!que.empty()){
p = que.front();que.pop();inq[p] = false;
for(int i = head[p]; ~i; i = edge[i].next){
int u = edge[i].to;
if(dis[u] > dis[p] + edge[i].w){
dis[u] = dis[p] + edge[i].w;
if(!inq[u]){
inq[u] = true;
que.push(u);
}
}
}
}
printf("%d\n", dis[Te]);
}
int main(int argc, char const *argv[])
{
memset(head, -1, sizeof head);
cnt = 0;
int u, v, w;
scanf("%d %d %d %d",&T, &C, &Ts, &Te);
for(int i=1;i<=C;++i){
scanf("%d %d %d",&u, &v, &w);
add(u, v, w);
add(v, u, w);
}
SPFA();
return 0;
}
/*
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
7
*/