Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++의 주어진 세트에 있는 모든 노드에서 도달 가능한 모든 노드 찾기

<시간/>

하나의 무방향 그래프와 정점 세트가 있다고 가정합니다. 주어진 세트에 있는 모든 정점에서 도달 가능한 모든 노드를 찾아야 합니다.

따라서 입력이 다음과 같으면

C++의 주어진 세트에 있는 모든 노드에서 도달 가능한 모든 노드 찾기

두 개의 연결된 구성 요소이므로 출력은 [1,2,3] 및 [4,5]가 됩니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • nodes :=그래프의 노드 수
  • 노드+1 크기의 방문 배열을 정의합니다. 0으로 채우기
  • 하나의 지도 정의
  • comp_sum :=0
  • 초기화 i의 경우:=0, i
  • u :=arr[i]
  • 방문한[u]가 거짓이면 -
    • (comp_sum 1 증가)
    • m[visited[u]] :=노드 u에서 g의 bfs 순회, comp_sum도 계산
  • m[visited[u]] 인쇄 순회

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;
class Graph{
   public:
      int nodes;
      list<int> *adj_list;
      Graph(int);
      void insert_edge(int, int);
      vector<int> BFS(int, int, int []);
};
Graph::Graph(int nodes) {
   this->nodes = nodes;
   adj_list = new list<int>[nodes+1];
}
void Graph::insert_edge(int u, int v) {
   adj_list[u].push_back(v);
   adj_list[v].push_back(u);
}
vector<int> Graph::BFS(int comp_sum, int src,int visited[]){
   queue<int> queue;
   queue.push(src);
   visited[src] = comp_sum;
   vector<int> reachableNodes;
   while(!queue.empty()) {
      int u = queue.front();
      queue.pop();
      reachableNodes.push_back(u);
      for (auto itr = adj_list[u].begin(); itr != adj_list[u].end(); itr++) {
         if (!visited[*itr]) {
            visited[*itr] = comp_sum;
            queue.push(*itr);
         }
      }
   }
   return reachableNodes;
}
void displayReachableNodes(int n, unordered_map <int, vector<int> > m) {
   vector<int> temp = m[n];
   for (int i=0; i<temp.size(); i++)
      cout << temp[i] << " ";
   cout << endl;
}
void get_all_reachable(Graph g, int arr[], int n) {
   int nodes = g.nodes;
   int visited[nodes+1];
   memset(visited, 0, sizeof(visited));
   unordered_map <int, vector<int> > m;
   int comp_sum = 0;
   for (int i = 0 ; i < n ; i++) {
      int u = arr[i];
      if (!visited[u]) {
         comp_sum++;
         m[visited[u]] = g.BFS(comp_sum, u, visited);
      }
      cout << "Reachable Nodes from " << u <<" are\n";
      displayReachableNodes(visited[u], m);
   }
}
int main() {
   int nodes = 5;
   Graph g(nodes);
   g.insert_edge(1, 2);
   g.insert_edge(2, 3);
   g.insert_edge(4, 5);
   int arr[] = {2, 4, 1};
   int n = sizeof(arr)/sizeof(int);
   get_all_reachable(g, arr, n);
}

입력

g.insert_edge(1, 2);
g.insert_edge(2, 3);
g.insert_edge(4, 5);

출력

Reachable Nodes from 2 are
2 1 3
Reachable Nodes from 4 are
4 5
Reachable Nodes from 1 are
2 1 3