671. Second Minimum Node In a Binary Tree

Given a non-empty special binary tree consisting of nodes with the non-negative value, where each node in this tree has exactly two or zero sub-node. If the node has two sub-nodes, then this node's value is the smaller value among its two sub-nodes. More formally, the property root.val = min(root.left.val, root.right.val) always holds.

Given such a binary tree, you need to output the second minimum value in the set made of all the nodes' value in the whole tree.

If no such second minimum value exists, output -1 instead.

 

 

Example 1:

Input: root = [2,2,5,null,null,5,7]
Output: 5
Explanation: The smallest value is 2, the second smallest value is 5.

Example 2:

Input: root = [2,2,2]
Output: -1
Explanation: The smallest value is 2, but there isn't any second smallest value.

 

Constraints:

  • The number of nodes in the tree is in the range [1, 25].
  • 1 <= Node.val <= 231 - 1
  • root.val == min(root.left.val, root.right.val) for each internal node of the tree.

Rust Solution

struct Solution;
use rustgym_util::*;

trait SecondMinimum {
    fn find_second_minimum_value(&self, min: &mut Option<i32>) -> Option<i32>;
    fn option_min(a: Option<i32>, b: Option<i32>) -> Option<i32>;
}

impl SecondMinimum for TreeLink {
    fn find_second_minimum_value(&self, min: &mut Option<i32>) -> Option<i32> {
        if let Some(node) = self {
            let node = node.borrow();
            let left = &node.left;
            let right = &node.right;
            if min.is_none() {
                *min = Some(node.val);
            }
            if min.unwrap() == node.val {
                let min_left = Self::find_second_minimum_value(&left, min);
                let min_right = Self::find_second_minimum_value(&right, min);
                Self::option_min(min_left, min_right)
            } else {
                Some(node.val)
            }
        } else {
            None
        }
    }
    fn option_min(a: Option<i32>, b: Option<i32>) -> Option<i32> {
        match (a, b) {
            (Some(a), Some(b)) => Some(i32::min(a, b)),
            (Some(a), None) => Some(a),
            (None, Some(b)) => Some(b),
            (None, None) => None,
        }
    }
}

impl Solution {
    fn find_second_minimum_value(root: TreeLink) -> i32 {
        let mut min = None;
        if let Some(second_min) = root.find_second_minimum_value(&mut min) {
            second_min
        } else {
            -1
        }
    }
}

#[test]
fn test() {
    let root = tree!(2, tree!(2), tree!(5, tree!(5), tree!(7)));
    assert_eq!(Solution::find_second_minimum_value(root), 5);
    let root = tree!(2, tree!(2), tree!(2));
    assert_eq!(Solution::find_second_minimum_value(root), -1);
}

Having problems with this solution? Click here to submit an issue on github.