Program to find minimum number of steps required to catch the opponent in C++

Suppose we have a list of tree edges in the form [u, v], this indicates there is an undirected edge between u and v. And we also have two values x and y. If we are at node x, and our opponent is at node y. In the first round, we move, then in the next round the opponent moves and so on. The opponent can select to not make a move in a round. We have to find the minimum number of rounds it we need to catch the opponent.

So, if the input is like edges = [[0, 1], [0, 2], [1, 3], [1, 4]], x = 0, y = 3, then the output will be 3, as at first, we move from node 0 to 1. Then opponent stays in the current node 3. Then we move to node 3.

To solve this, we will follow these steps −

N := 1^5 + 5
Define an array visited of size: N and another array visited2 of size: N fill them with −1
create adjacency list graph for N nodes
for each edge it in edges list, do
- insert it[v] at the end of graph[it[u]]
- insert it[u] at the end of graph[it[v]]
Define one queue w
insert u into q
visited[u] := 0
while q is not empty, do −
- node := first element of q
- delete element from q
- for each node it in graph[node]
  - if visited[it] is same as −1, then −
    - visited[it] := visited[node] + 1
    - insert it into q
insert v into q
ret := 0
visited2[v] := 0
while q is not empty), do −
- node := first element of q
- delete element from q
- ret := maximum of ret and 2 * (visited[node])
- for each node it in graph[node]
  - if visited2[it] is same as -1 and visited2[node] + 2 < visited[it], then −
    - visited2[it] := visited2[node] + 1
    - insert it into q
return ret

Let us see the following implementation to get better understanding −

Example

Live Demo

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
int visited[N];
int visited2[N];
vector<int> graph[N];
class Solution {
   public:
   int solve(vector<vector<int>>& edges, int u, int v) {
      memset(visited, −1, sizeof visited);
      memset(visited2, −1, sizeof visited2);
      for (int i = 0; i < N; i++)
      graph[i].clear();
      for (auto& it : edges) {
         graph[it[0]].push_back(it[1]);
         graph[it[1]].push_back(it[0]);
      }
      queue<int> q;
      q.push(u);
      visited[u] = 0;
      while (!q.empty()) {
         int node = q.front();
         q.pop();
         for (auto& it : graph[node]) {
            if (visited[it] == −1) {
               visited[it] = visited[node] + 1;
               q.push(it);
            }
         }
      }
      q.push(v);
      int ret = 0;
      visited2[v] = 0;
      while (!q.empty()) {
         int node = q.front();
         q.pop();
         ret = max(ret, 2 * (visited[node]) − 1);
         for (auto& it : graph[node]) {
            if (visited2[it] == −1 && visited2[node] + 2 <
            visited[it]) {
               visited2[it] = visited2[node] + 1;
               q.push(it);
            }
         }
      }
      return ret;
   }
};
int solve(vector<vector<int>>& edges, int u, int v) {
   return (new Solution())−>solve(edges, u, v);
}
int main(){
   vector<vector<int>> edge = {{0, 1},{0, 2},{1, 3},{1, 4}};
   int x = 0, y = 3;
   cout << solve(edge, x, y);
}

Input

[
   [0, 1],
   [0, 2],
   [1, 3],
   [1, 4]
], 0, 3

Output

Arnab Chakraborty

Published on 25-Dec-2020 05:48:30

Previous Page Print Page