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

C++에서 이 삼중항을 연결하는 노드의 수가 최대가 되도록 삼중항 찾기


이 튜토리얼에서는 이 삼중항을 연결하는 노드의 수가 최대가 되도록 삼중항을 찾는 프로그램에 대해 논의할 것입니다.

이를 위해 N개의 노드가 있는 트리가 제공됩니다. 우리의 임무는 경로에 포함된 노드가 최대로 결합하는 세 개의 노드를 찾는 것입니다.

예시

#include <bits/stdc++.h>
#define ll long long int
#define MAX 100005
using namespace std;
vector<int> nearNode[MAX];
bool isTraversed[MAX];
//storing the required nodes
int maxi = -1, N;
int parent[MAX];
bool vis[MAX];
int startnode, endnode, midNode;
//implementing DFS to search nodes
void performDFS(int u, int count) {
   isTraversed[u] = true;
   int temp = 0;
   for (int i = 0; i < nearNode[u].size(); i++) {
      if (!isTraversed[nearNode[u][i]]) {
         temp++;
         performDFS(nearNode[u][i], count + 1);
      }
   }
   if (temp == 0) {
      if (maxi < count) {
         maxi = count;
         startnode = u;
      }
   }
}
void performDFS2(int u, int count) {
   isTraversed[u] = true;
   int temp = 0;
   for (int i = 0; i < nearNode[u].size(); i++) {
      if (!isTraversed[nearNode[u][i]] && !vis[nearNode[u][i]]) {
         temp++;
         performDFS2(nearNode[u][i], count + 1);
      }
   }
   if (temp == 0) {
      if (maxi < count) {
         maxi = count;
         midNode = u;
      }
   }
}
//finding endnote of diameter
void performDFS1(int u, int count) {
   isTraversed[u] = true;
   int temp = 0;
   for (int i = 0; i < nearNode[u].size(); i++) {
      if (!isTraversed[nearNode[u][i]]) {
         temp++;
         parent[nearNode[u][i]] = u;
         performDFS1(nearNode[u][i], count + 1);
      }
   }
   if (temp == 0) {
      if (maxi < count) {
         maxi = count;
         endnode = u;
      }
   }
}
void calcTreeVertices() {
   performDFS(1, 0);
   for (int i = 0; i <= N; i++)
      isTraversed[i] = false;
   maxi = -1;
   performDFS1(startnode, 0);
   for (int i = 0; i <= N; i++)
      isTraversed[i] = false;
   int x = endnode;
   vis[startnode] = true;
   while (x != startnode) {
      vis[x] = true;
      x = parent[x];
   }
   maxi = -1;
   for (int i = 1; i <= N; i++) {
      if (vis[i])
         performDFS2(i, 0);
   }
}
int main() {
   N = 4;
   nearNode[1].push_back(6);
   nearNode[2].push_back(0);
   nearNode[1].push_back(7);
   nearNode[3].push_back(0);
   nearNode[1].push_back(2);
   nearNode[4].push_back(0);
   calcTreeVertices();
   cout << "Nodes: (" << startnode << ", " << endnode << ", " << midNode << ")";
   return 0;
}

출력

Nodes: (0, 0, 0)