使用BFS的最短距离

时间:2012-06-20 09:45:32

标签: c++ algorithm shortest-path breadth-first-search

我正在努力解决this SPOJ problem。该问题要求找到每个黑色(1​​)像素的最短路径。

由于它是未加权的图表,我使用了BFS。

输入:

3 3
010
000
000

它正在给予:

323
434
343

而不是:

101
212
323

这是我的代码

#include<iostream>
#include<queue>
#include<string.h>
using namespace std;
typedef pair < int, int >ii;
int R, C, i, j;
queue < ii > myQueue;

int visit[100][100];
int dist[100][100];
void bfs(ii s)
{
    int i, j;
    int count = 0;
    ii node;
    memset(visit, 0, sizeof(visit));
    memset(dist, 0, sizeof(dist));
    myQueue.push(s);
    dist[node.first][node.second] = 0;

    while (!myQueue.empty()) {
        node = myQueue.front();
        myQueue.pop();
        if (visit[node.first][node.second])
            continue;
        visit[node.first][node.second] = 1;

        //cout << node.first << " " << node.second << "\n";
        i = node.first;
        j = node.second;

        if (j - 1 < R && j - 1 >= 0) {
            myQueue.push(make_pair(i, j - 1));
            if(dist[i][j - 1] == 0)
            dist[i][j - 1] = dist[i][j] + 1;
        }
        if (j + 1 < R && j + 1 >= 0) {
            myQueue.push(make_pair(i, j + 1));
            if(dist[i][j+1] == 0)
            dist[i][j + 1] = dist[i][j] + 1;
        }
        if (i - 1 < C && i - 1 >= 0) {
            myQueue.push(make_pair(i - 1, j));
            if(dist[i-1][j] == 0)
            dist[i - 1][j] = dist[i][j] + 1;
        }
        if (i + 1 < C && i + 1 >= 0) {
            myQueue.push(make_pair(i + 1, j));
            if(dist[i+1][j] == 0)
            dist[i + 1][j] = dist[i][j] + 1;
        }
    }
}

int main()
{
    char input[100][100];
    scanf("%d %d", &R, &C);
    for (i = 0; i < R; i++)
        scanf("%s", &input[i]);
    int GRID[R][C];
    for (i = 0; i < R; i++)
        for (j = 0; j < C; j++)
            GRID[i][j] = input[i][j] - '0';
    for (i = 0; i < R; i++)
        for (j = 0; j < C; j++) {
            if (GRID[i][j] == 1)
                bfs(make_pair(i, j));
        }
    for (i = 0; i < R; i++) {
        for (j = 0; j < C; j++) {
            printf("%d", dist[i][j]);
        }
        printf("\n");
    }
}

ideone

2 个答案:

答案 0 :(得分:3)

试试这个:

if (j - 1 < R && j - 1 >= 0) {
    myQueue.push(make_pair(i, j - 1));
    if(dist[i][j - 1] == 0)
    dist[i][j - 1] = dist[i][j] + 1;
}

为所有dist [] []执行此操作。

答案 1 :(得分:0)

您的结果加倍可能是因为您在配对顶点之间运行BFS两次。

但我不确定。