본문 바로가기

Algorithm/자료구조

(C++) - LeetCode (easy) 144. Binary Tree Preorder Traversal

반응형

https://leetcode.com/problems/binary-tree-preorder-traversal/

 

Binary Tree Preorder Traversal - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com

preorder를 구현해보는 문제였습니다.

📕 풀이방법

📔 풀이과정

preorder란 binary tree를 순회하는 방식입니다. root node -> root의 왼쪽 자식 -> root의 오른쪽 자식 순으로 순회하는 algorithm으로 재귀로 구현하면 비교적 간단하게 구현됩니다.

재귀함수 preorderTraversal을 구현합니다.

1. 반환형이 vector이므로 root->NULL이라면 tree의 말단이므로 기저에 해당합니다.vector <int> ()를 반환해줍니다.

2. 현재 node가 root이므로 이 경우 vector v를 호출해 저장해줍니다.

3. 왼쪽 자식에 대해 preorderTraversal을 호출해줍니다. 이 결과를 vector a를 선언해 저장해줍니다.

4. 오른쪽 자식에 대해 preorderTraversal을 호출해줍니다. 이 결과를 vector b를 선언해 저장해줍니다.

5. v에 a와 b의 원소들을 각각 push_back해줍니다.

6. v를 반환합니다.


📕 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:
    vector<int> preorderTraversal(TreeNode* root) {
        if(root == NULL) return vector<int>();
        vector <int> totalTraversal;
        totalTraversal.push_back(root->val);
        vector <int> leftTraversal = preorderTraversal(root->left);
        vector <int> rightTraversal = preorderTraversal(root->right);
        totalTraversal.insert(totalTraversal.end(), leftTraversal.begin(), leftTraversal.end());
        totalTraversal.insert(totalTraversal.end(), rightTraversal.begin(), rightTraversal.end());
        return totalTraversal;
    }
};

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