Challenge——spfa

Challenge

题目描述

wys和zerinf经常出题来虐Zhuyu。有一天, wys搞了一个有向图,每条边的长度都是1。 他想让Zhuyu求出点1到点 N 的最短路。
“水题啊。”, Zhuyu这么说道。
所以zerinf把某些边的长度增加了1（也就是说，每条边的长度不是1就是2）。现在，可怜的Zhuyu要向你求助了。

输入格式

第1行，两个整数 N (1 ≤ N ≤ 10^5) 和 M (1 ≤ M ≤ 10^6), 点和边的数量。
第2到 M + 1行: 三个整数 Ui, Vi, Wi (1 ≤ Wi ≤ 2), 从点 Ui 到 Vi 长度为 Wi 的边。

输出格式

一个整数，表示点1到点N的最短路。数据保证至少存在一条路径。

样例输入1

3 3
1 2 1
2 3 1
1 3 2

样例输出1

2

#include<bits/stdc++.h>
using namespace std;
int n,m,v,u,pre[100002],w,dis[100002],cnt;
queue<int> q;
bool used[100002];
struct ed{int to,w,next;
}e[1000002];
void add(int u,int v,int w,int b){e[++cnt].to=v;e[cnt].next=pre[u];e[cnt].w=w;pre[u]=b;
}
int main(){memset(pre,-1,sizeof(pre));memset(dis,0x3f,sizeof(dis));scanf("%d%d",&n,&m);for(int i=1;i<=m;i++){scanf("%d%d%d",&u,&v,&w);add(u,v,w,i);}q.push(1);used[1]=1;dis[1]=0;while(!q.empty()){int now=q.front();//q.pop();used[now]=0;for(int i=pre[now];i!=-1;i=e[i].next ){//printf("%d\n",i);if(e[i].w+dis[now]<dis[e[i].to]){dis[e[i].to]=e[i].w+dis[now];if(!used[e[i].to]){used[e[i].to]=1;q.push(e[i].to);}}}}printf("%d\n",dis[n]);
}
/*
6 9
1 2 1
2 4 3
4 6 15
1 3 12
2 3 1
4 3 4
3 5 5
4 5 13
5 6 411
*/