C(最短路_spfa+前向星)
阿新 • • 發佈：2019-01-09
C
Time Limit: 7000ms Memory limit: 65536K 有疑問？點這裡^_^
題目描述
給出一個帶權無向圖，包含n個點，m條邊。求出s，e的最短路。保證最短路存在。輸入
多組輸入。 對於每組資料。 第一行輸入n，m(1<= n && n<=5*10^5,1 <= m && m <= 2*10^6)。接下來m行，每行三個整數，u,v,w,表示u，v之間有一條權值為w(w >= 0)的邊。
最後輸入s，e。
輸出
對於每組資料輸出一個整數代表答案。示例輸入
3 1 1 2 3 1 2
示例輸出
3
提示
來源
zmx示例程式
卡資料卡的太厲害了，差點超記憶體，spfa+前向星果斷水過，但是TLE了好幾次，我又把程式碼刪掉重新寫的，然後AC了 ，都一樣 不知道為什麼，在下面貼出來，希望各路大神給找找，萬分感謝。
TLE的程式碼 7010ms
#include <stdio.h> #include <string.h> #include <stdlib.h> #include <algorithm> #include <iostream> #include <queue> #include <string> using namespace std; #define inf 1e19+7 struct node { int v,w; int next; } edge[2000010]; int dis[500010]; int vis[500010]; int head[500010]; int cnt; int n; void add(int u,int v,int w) { edge[cnt].v=v; edge[cnt].w=w; edge[cnt].next=head[u]; head[u]=cnt++; } void SPFA(int s,int e) { int i; queue<int>q; memset(vis,0,sizeof(vis)); for(i=0; i<=n; i++) dis[i]=inf; dis[s]=0; vis[s]=1; q.push(s); while(!q.empty()) { int u=q.front(); q.pop(); vis[u]=0; for(i=head[u]; i!=-1; i=edge[i].next) { int v=edge[i].v; if(dis[v]>dis[u]+edge[i].w) { dis[v]=dis[u]+edge[i].w; if(!vis[v]) { vis[v]=1; q.push(v); } } } } } int main() { int m,i; int u,v,w; int s,e; while(~scanf("%d %d",&n,&m)) { memset(head,-1,sizeof(head)); cnt=0; for(i=1; i<=m; i++) { scanf("%d %d %d",&u,&v,&w); add(u,v,w); add(v,u,w); } scanf("%d %d",&s,&e); SPFA(s,e); printf("%d\n",dis[e]); } return 0； ｝
AC程式碼 1200ms
#include <stdio.h> #include <string.h> #include <stdlib.h> #include <algorithm> #include <iostream> #include <queue> #include <string> using namespace std; #define inf 999999999 struct node { int v,w; int next; }edge[4000010]; int dis[500010]; int vis[500010]; int head[500010]; int cnt; int n; void add(int u,int v,int w) { edge[cnt].v=v; edge[cnt].w=w; edge[cnt].next=head[u]; head[u]=cnt++; } void SPFA(int s,int e) { int i; queue<int>q; memset(vis,0,sizeof(vis)); for(i=0; i<=n; i++) { dis[i]=inf; } dis[s]=0; vis[s]=1; q.push(s); while(!q.empty()) { int u=q.front(); q.pop(); vis[u]=0; for(i=head[u];i!=-1;i=edge[i].next) { int v=edge[i].v; if(dis[v]>dis[u]+edge[i].w) { dis[v]=dis[u]+edge[i].w; if(!vis[v]) { vis[v]=1; q.push(v); } } } } } int main() { int m; int u,v,w; int s,e; while(~scanf("%d %d",&n,&m)) { memset(head,-1,sizeof(head)); cnt=0; while(m--) { scanf("%d %d %d",&u,&v,&w); add(u,v,w); add(v,u,w); } scanf("%d %d",&s,&e); SPFA(s,e); printf("%d\n",dis[e]); } return 0; }