九度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[502];
int step[502];
 
int main(int argc, char const *argv[])
{
    while(scanf("%d %d %d",&n,&m,&s) != EOF) {
        while(!que.empty()) {
            que.pop();
        }
        for(int i = 0; i < n; i++) {
            edge[i].clear();
        }
        memset(step, 0, 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 = 0; i < sizep; i++) {
                int tp = edge[p][i].first;
                int ct = edge[p][i].second;
                int pt = step[p] + ct;
                if(step[tp] == 0 || pt < step[tp]) {
                    step[tp] = pt;
                    que.push(tp);
                }
            }
        }
        if(step[s] == 0) {
            puts("help!");
        }
        else {
            printf("%d
",step[s]);
        }
    }
    return 0;
}

这段代码犯了两个错误，一是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[502];
int step[502];

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 = 0; i < 502; i++) {
            edge[i].clear();
        }
        memset(step, 0, 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 = 0; 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] == 0 || pt < step[tp]) {
                    step[tp] = pt;
                    que.push(tp);
                }
            }
        }
        if(step[s] == 0) {
            puts("help!");
        }
        else {
            printf("%d
",step[s]);
        }
    }
    return 0;
}