BFS在创建图形时出现分段错误,尝试进行调试,不确定出什么问题

问题描述 投票:0回答:1

我已经开始学习图论,并从Hackerrank https://www.hackerrank.com/challenges/bfsshortreach/problem处提出了一个问题,该问题基本上要求执行和BFS,将同一级别的所有元素都标记为6 *级别,并将所有不可达的节点标记为-1。我按以下方式尝试过,创建adjList时遇到段错误,是否以错误的方式访问输入vector<vector<int>> edges?或者是别的什么。这是我的2-3 bfs代码,所以我还想知道我的实现是否可行,它是否正在执行应做的工作? (我知道这个问题要求ans vector不是当前访问的1-n顺序,也许我可以使用地图,直方图或具有原始值的对,然后在第二秒或以后对它进行排序,还没弄清楚那部分)谢谢。

代码:

vector<vector<int>> Graph(vector<vector<int>>& edges) {
    vector<vector<int>> ans;
    for(auto i: edges) {
        ans[i[0]].push_back(i[1]);
        ans[i[1]].push_back(i[0]);
    } return ans;
}

vector<int> bfs(int n, int m, vector<vector<int>>& edges, int s) { // no. of vertex, no. of edges, edges, start
    vector<vector<int>> adjList;
    // adjList.resize(n);
    adjList = Graph(edges);
    vector<int> ans(n + 1);
    vector<bool> visited(n + 1);
    queue<int> q;
    visited[s] = true;
    q.push(s);
    int i = 0;
    while(!q.empty()) {
        int temp = q.front(); q.pop();
        // int len = q.size();
        for(int j: adjList[temp]) {
            if(!visited[j]) {
                ans.push_back(6*i);
                visited[j] = true;
                q.push(j);
            }
        } i++;
    }
    for(int i = 1; i <= n; i++) if(i % 6 != 0 && i != s) ans[i] = -1;
    ans.erase(ans.begin() + s);
    return ans;
}

经过测试

int main() {
    vector<vector<int>> edges = {{1, 2}, {1, 3} };
    vector<int> som = bfs(4, 2, edges, 1);
    return 0;
}
c++ segmentation-fault breadth-first-search
1个回答
0
投票

vector<vector<int>> ans必须调整为Graph中的顶点数。

工作代码:

vector<vector<int>> Graph(vector<vector<int>>& edges, int n) {
    vector<vector<int>> ans;
    ans.resize(n);
    for(auto i: edges) {
        ans[i[0]].push_back(i[1]);
        ans[i[1]].push_back(i[0]);
    } return ans;
}

vector<int> bfs(int n, int m, vector<vector<int>>& edges, int s) {
    vector<vector<int>> adjList;
    adjList = Graph(edges, n + 1);
    vector<int> ans(n + 1, -1);
    ans[s] = 0; // first distance
    vector<bool> visited(n + 1);
    queue<int> q;
    visited[s] = true;
    q.push(s);
    while(!q.empty()) {
        int temp = q.front(); q.pop();
        for(int j: adjList[temp]) {
            if(!visited[j]) {
                ans[j] = 6 + ans[temp];   // just use it as DP array!
                visited[j] = true;
                q.push(j);
            }
        }
    }
    ans.erase(ans.begin() + s); // remove starting point(as per question)
    ans.erase(ans.begin()); // remove 0 (graph starts from 1)
    return ans;
}
© www.soinside.com 2019 - 2024. All rights reserved.