我已经开始学习图论,并从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;
}
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;
}