You are given a directed graph of n
nodes numbered from 0
to n - 1
, where each node has at most one outgoing edge.
The graph is represented with a given 0-indexed array edges
of size n
, indicating that there is a directed edge from node i
to node edges[i]
. If there is no outgoing edge from i
, then edges[i] == -1
.
You are also given two integers node1
and node2
.
Return the index of the node that can be reached from both node1
and node2
, such that the maximum between the distance from node1
to that node, and from node2
to that node is minimized. If there are multiple answers, return the node with the smallest index, and if no possible answer exists, return -1
.
Note that edges
may contain cycles.
Example 1:
Input: edges = [2,2,3,-1], node1 = 0, node2 = 1 Output: 2 Explanation: The distance from node 0 to node 2 is 1, and the distance from node 1 to node 2 is 1. The maximum of those two distances is 1. It can be proven that we cannot get a node with a smaller maximum distance than 1, so we return node 2.
Example 2:
Input: edges = [1,2,-1], node1 = 0, node2 = 2 Output: 2 Explanation: The distance from node 0 to node 2 is 2, and the distance from node 2 to itself is 0. The maximum of those two distances is 2. It can be proven that we cannot get a node with a smaller maximum distance than 2, so we return node 2.
Constraints:
n == edges.length
2 <= n <= 105
-1 <= edges[i] < n
edges[i] != i
0 <= node1, node2 < n
Companies: Juspay
Related Topics:
Depth-First Search, Graph
Hints:
- How can you find the shortest distance from one node to all nodes in the graph?
- Use BFS to find the shortest distance from both node1 and node2 to all nodes in the graph. Then iterate over all nodes, and find the node with the minimum max distance.
- Calculate the distances from node
a
andb
separately asda
anddb
. - Find the node with the min max distance from both nodes
// OJ: https://leetcode.com/problems/find-closest-node-to-given-two-nodes
// Author: github.com/lzl124631x
// Time: O(N)
// Space: O(N)
class Solution {
public:
int closestMeetingNode(vector<int>& E, int a, int b) {
int N = E.size(), ans = -1, minDist = INT_MAX;
vector<int> da(N, INT_MAX), db(N, INT_MAX);
auto updateDist = [&](int u, vector<int> &dist) {
int lv = 0;
while (u != -1 && dist[u] == INT_MAX) {
dist[u] = lv++;
u = E[u];
}
};
updateDist(a, da);
updateDist(b, db);
for (int i = 0; i < N; ++i) {
int d = max(da[i], db[i]);
if (d < minDist) {
minDist = d;
ans = i;
}
}
return ans;
}
};