ACM学习之路——HDU2544 HDU2066（spfa）

题目链接

HDU2544  http://acm.hdu.edu.cn/showproblem.php?pid=2544

HDU2066 http://acm.hdu.edu.cn/showproblem.php?pid=2066

话说这两道我用的是spfa算法，由于第一次使用这个算法，找bug真是找的我昏天黑地。

第一个要注意的就是链式前向星的位置变量CNT在每一次的循环中都要初始化0；另外，其余的数组最好初始化为-1；以防止越界。

AC代码：

///////////////////////////////////////////////////////// //                        HDU 2066                     ////  Created by 吴尔立                ////  Copyright (c) 2015年 吴尔立. All rights reserved.  ///////////////////////////////////////////////////////////#include <iostream>#include <cstdio>#include <cstring>#include <cmath>#include <cstdlib>           #include <algorithm>#include <cctype>#include <stack>#include <queue>#include <map>#include <string>#include <set>#include <vector>#define ll long long;#define INF 1<<31#define cir(i,a,b)  for (int i=a;i<=b;i++)#define CIR(j,a,b)  for (int j=a;j>=b;j--)#define CLR(x) memset(x,0,sizeof(x))using namespace std;#define maxn 2100int pnt[maxn],nxt[maxn],head[maxn],w[maxn],cnt=0;int T,S,D;bool vis[maxn];int d[maxn];void addedge(int u,int v,int weight){pnt[cnt]=v;nxt[cnt]=head[u];head[u]=cnt;w[cnt]=weight;cnt++;}int  spfa(){queue<int> q;memset(d,0x3f,sizeof(d));memset(vis,false,sizeof(vis)); d[0]=0;q.push(0);vis[0]=true;while (!q.empty()){int x=q.front();q.pop();for (int i=head[x];~i;i=nxt[i]){int v=pnt[i];if (d[x]+w[i]<d[v] || d[v]==0x3f){d[v]=d[x]+w[i];if (!vis[v]){q.push(v);vis[v]=true;}}}vis[x]=false;}return d[1001];}int main(){while(cin >> T >> S >> D){cnt=0;memset(nxt,-1,sizeof(nxt));memset(head,-1,sizeof(head));cir (i,1,T){int a,b,time;scanf("%d%d%d",&a,&b,&time);addedge(a,b,time);addedge(b,a,time);}cir(i,1,S){int u;scanf("%d",&u);addedge(0,u,0);addedge(u,0,0);}cir(i,1,D){int v;scanf("%d",&v);addedge(1001,v,0);addedge(v,1001,0);}int ans=spfa();cout << ans <<endl;}return 0;}

///////////////////////////////////////////////////////// //                        HDU 2544                     ////  Created by 吴尔立 <span style="white-space:pre"></span>                       ////  Copyright (c) 2015年 吴尔立. All rights reserved.  ///////////////////////////////////////////////////////////#include <iostream>#include <cstdio>#include <cstring>#include <cmath>#include <cstdlib>           #include <algorithm>#include <cctype>#include <stack>#include <queue>#include <map>#include <string>#include <set>#include <vector>#define ll long long;#define INF 1<<31#define cir(i,a,b)  for (int i=a;i<=b;i++)#define CIR(j,a,b)  for (int j=a;j>=b;j--)#define CLR(x) memset(x,0,sizeof(x))using namespace std;const int maxn = 10005;int N, M;  //N个路口M条路； int A, B, C;   //A到B用时为Cint pnt[maxn], pre[maxn], head[maxn], d[maxn], dis[maxn]; //d[i] 表示从起点到 i 但的最短路 int counter[maxn], vis[maxn];int cnt = 0; void addedge(int u, int v, int w) {    pnt[cnt] = v; pre[cnt] = head[u]; dis[cnt] = w; head[u] = cnt++;} int spfa(int s) {    queue<int> q;    memset(d, -1, sizeof(d));    CLR(vis);    d[s] = 0;    q.push(s);    vis[s] = 1;    while (!q.empty()) {        int x = q.front();        q.pop();        for (int i = head[x]; i != -1; i = pre[i])   //遍历邻接表         {            int y =  pnt[i];            if (d[y] == -1 || d[x] + dis[i] < d[y]) {                d[y] = d[x] + dis[i];                if (!vis[y]) {                    vis[y] = 1;                    q.push(y);                }            }        }        vis[x] = 0;    }    return 0;} int main() {    while (scanf("%d%d", &N, &M) != EOF && N != 0 && M != 0) {        cnt = 0;        CLR(dis);        memset(head, -1, sizeof(head));        cir(i, 1, M) {            scanf("%d %d %d", &A, &B, &C);            addedge(A, B, C);            addedge(B, A, C);        }        spfa(1);        cout << d[N] << endl;    }    return 0;}