UVALive 6658 - Fiasco(BFS)

题目：

http://acm.hust.edu.cn/vjudge/contest/view.action?cid=84227#problem/I

题意：

根据给出的最短路的错误代码，输出可以得到正确答案的数据。

错误代码的思路是: 每次取最短的线段加入路径。

根据代码思路：先将所有的线段排序，从源点开始bfs，按照遍历的顺序赋值边。

没有赋值到的线段，则赋值剩下的数据。

AC.

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
#include <queue>

using namespace std;
const int maxn = 25005;
int dis[maxn], a[maxn], b[maxn];
struct edge {
    int to, w, id;
    edge(int tt, int ww, int ii) {
        to = tt;  w = ww; id = ii;
    }
};
vector<edge> g[maxn];
int n, m, st;

void init()
{
    memset(dis, 0, sizeof(dis));
    for(int i = 0; i <= n; ++i) {
        g[i].clear();
    }
}

int ans[maxn], vis[maxn];
int cnt;
void bfs()
{
    queue<int> que;
    que.push(st);
    memset(vis, 0, sizeof(vis));
    memset(ans, 0, sizeof(ans));
    vis[st] = 1;

    cnt = 0;
    while(!que.empty()) {
        int u = que.front(); que.pop();

        for(int i = 0; i < g[u].size(); ++i) {
            int v = g[u][i].to, id = g[u][i].id;
            if(!vis[v]) {
                ans[id] = dis[cnt++];
                vis[v] = 1;
                que.push(v);
            }
        } 
    }
}
int main()
{
    //freopen("in", "r", stdin);
    int T, ca = 1;
    scanf("%d", &T);
    while(T--) {
        scanf("%d %d %d", &n, &m, &st);

        init();
        for(int i = 0; i < m; ++i) {
            int u, v, w;
            scanf("%d %d %d", &u, &v, &w);
            a[i] = u; b[i] = v;
            dis[i] = w;
            g[u].push_back(edge(v, w, i));
            g[v].push_back(edge(u, w, i));
        }

        sort(dis, dis+m);
        bfs();
        for(int i = 0; i < m; ++i) {
            if(ans[i] == 0) {
                ans[i] = dis[cnt++];
            }
        }

        printf("Case %d:\n", ca++);
        for(int i = 0; i < m; ++i) {
            printf("%d %d %d\n", a[i], b[i], ans[i]);
        }

    }
    return 0;
}