(C++) - LeetCode (easy) 671. Second Minimum Node In a Binary Tree

https://leetcode.com/problems/second-minimum-node-in-a-binary-tree/description/

Second Minimum Node In a Binary Tree - LeetCode

자료구조와 tree 순회 문제였습니다.

📕 풀이방법

📔 입력 및 초기화

tree의 node들을 고유하게 저장하는 set변수 s를 선언해줍니다.

📔 풀이과정

1. preOrder를 수행해 각 node를 s에 insert해줍니다.

2. s의 원소들을 vector setVec을 선언해 옮겨 담아줍니다.

📔 정답 출력 | 반환

원소가 1개라면 두 번째로 작은 원소가 없으므로 -1을 아니라면 해당 원소를 setVec에서 반환합니다.

---

📕 Code

📔 C++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
 
class Solution {
public:
    set <int> s;
    void preOrder(TreeNode* root) {
        if(root == NULL) return;
        s.insert(root->val);
        preOrder(root->left);
        preOrder(root->right);
    }

    int findSecondMinimumValue(TreeNode* root) {
        preOrder(root);
        vector <int> setVec(s.begin(), s.end());
        if(setVec.size() == 1) return -1;
        return setVec[1]; 
    }
};

---

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.