九度oj 题目1411：转圈

题目描述：

在一个有向图有n个顶点（编号从1到n），给一个起点s，问从起点出发，至少经过一条边，回到起点的最短距离。

输入：

输入包括多组，每组输入第一行包括三个整数n,m,s（1<=n<=500,0<=m<=10000,1<=s<=n），接下来有m行，每行包括三个整数a,b,c(1<=a,b<=n,1<=c<=1000)，表示有一条a到b的边，长度为c。

输出：

对每组输入。输出最短距离，如果没有这个一条路径输出"help!"。

样例输入：

5 6 1
1 2 1
2 3 2
3 4 1
4 5 1
3 1 3
4 1 1

样例输出：

5

看网上的做法，都使用迪杰斯特拉算法做的，而我采用的是广度优先搜索的办法
但一开始的代码提交错误，如下

 #include <cstdio>
 #include <cstdlib>
 #include <cstring>
 #include <queue>
 #include <vector>
 using namespace std;

 int n,m,s;
 queue <int> que;
 typedef pair<int, int> Edge;
 vector<Edge> edge[];
 int step[];

 int main(int argc, char const *argv[])
 {
     while(scanf("%d %d %d",&n,&m,&s) != EOF) {
         while(!que.empty()) {
             que.pop();
         }
         for(int i = ; i < n; i++) {
             edge[i].clear();
         }
         memset(step, , sizeof(step));
         while(m--) {
             int a, b, d;
             scanf("%d %d %d",&a,&b,&d);
             edge[a].push_back(Edge(b,d));
         }
         que.push(s);
         while(!que.empty()) {
             int p = que.front();que.pop();
             int sizep = edge[p].size();

             for(int i = ; i < sizep; i++) {
                 int tp = edge[p][i].first;
                 int ct = edge[p][i].second;
                 int pt = step[p] + ct;
                 if(step[tp] ==  || pt < step[tp]) {
                     step[tp] = pt;
                     que.push(tp);
                 }
             }
         }
         if(step[s] == ) {
             puts("help!");
         }
         else {
             printf("%d\n",step[s]);
         }
     }
     return ;
 }

这段代码犯了两个错误，一是20行, clear不应该只到n，万一第二次输入的n比第一次的n小呢?

第二是题目中有 起点s到起点s的边，需要特殊处理，代码如下

 #include <cstdio>
 #include <cstdlib>
 #include <cstring>
 #include <queue>
 #include <vector>
 using namespace std;

 int n,m,s;
 queue <int> que;
 typedef pair<int, int> Edge;
 vector<Edge> edge[];
 int step[];

 int main(int argc, char const *argv[])
 {
     freopen("input.txt","r",stdin);
     while(scanf("%d %d %d",&n,&m,&s) != EOF) {
         while(!que.empty()) {
             que.pop();
         }
         for(int i = ; i < ; i++) {
             edge[i].clear();
         }
         memset(step, , sizeof(step));
         while(m--) {
             int a, b, d;
             scanf("%d %d %d",&a,&b,&d);
             edge[a].push_back(Edge(b,d));
         }
         que.push(s);
         while(!que.empty()) {
             int p = que.front();que.pop();
             int sizep = edge[p].size();

             for(int i = ; i < sizep; i++) {
                 int tp = edge[p][i].first;
                 int ct = edge[p][i].second;
                 int pt = step[p] + ct;
                 if(p == s) {
                     pt = ct;
                 }
                 if(step[tp] ==  || pt < step[tp]) {
                     step[tp] = pt;
                     que.push(tp);
                 }
             }
         }
         if(step[s] == ) {
             puts("help!");
         }
         else {
             printf("%d\n",step[s]);
         }
     }
     return ;
 }