day15 | 二叉樹2

0.引言

● 層序遍歷 10
● 226.翻轉(zhuǎn)二叉樹
● 101.對稱二叉樹 2

1.翻轉(zhuǎn)二叉樹

Category Difficulty Likes Dislikes
algorithms Easy (79.52%) 1525 -

給你一棵二叉樹的根節(jié)點(diǎn) root ,翻轉(zhuǎn)這棵二叉樹,并返回其根節(jié)點(diǎn)。

示例 1:

image.png
輸入:root = [4,2,7,1,3,6,9]
輸出:[4,7,2,9,6,3,1]

示例 2:

image.png
輸入:root = [2,1,3]
輸出:[2,3,1]

示例 3:

輸入:root = []
輸出:[]

提示:

  • 樹中節(jié)點(diǎn)數(shù)目范圍在 [0, 100] 內(nèi)
  • -100 <= Node.val <= 100

Discussion | Solution

1.1遞歸法

/*
 * @lc app=leetcode.cn id=226 lang=cpp
 *
 * [226] 翻轉(zhuǎn)二叉樹
 */

// @lc code=start
/**
 * 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:
  TreeNode* invertTree(TreeNode* root) {
    dfs(root);
    return root;
  }
  // 前序遍歷
 private:
  void dfs(TreeNode* node) {
    if (node == nullptr) return;
    std::swap(cur->left, cur->right);
    dfs(node->left);
    dfs(node->right);
  }
};
// @lc code=end

1.2.迭代法

/*
 * @lc app=leetcode.cn id=226 lang=cpp
 *
 * [226] 翻轉(zhuǎn)二叉樹
 */

// @lc code=start
/**
 * 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:
  TreeNode* invertTree(TreeNode* root) {
    std::stack<TreeNode*> node_stack;
    if (root == nullptr) return root;
    node_stack.push(root);

    while (!node_stack.empty()) {
      TreeNode* cur = node_stack.top();
      node_stack.pop();
      std::swap(cur->left, cur->right);
      if (cur->left) node_stack.push(cur->left);
      if (cur->right) node_stack.push(cur->right);
    }
    return root;
  }
};
// @lc code=end

1.3.層序遍歷BFS

/*
 * @lc app=leetcode.cn id=226 lang=cpp
 *
 * [226] 翻轉(zhuǎn)二叉樹
 */

// @lc code=start
/**
 * 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:
  TreeNode* invertTree(TreeNode* root) {
    std::queue<TreeNode*> que;
    if (root != nullptr) {
      que.push(root);
    }
    while (!que.empty()) {
      for (int i = que.size(); i > 0; --i) {
        TreeNode* cur = que.front();
        que.pop();
        std::swap(cur->left, cur->right);
        if (cur->left) que.push(cur->left);
        if (cur->right) que.push(cur->right);
      }
    }
    return root;
  }
};
// @lc code=end

2.對稱二叉樹

Category Difficulty Likes Dislikes
algorithms Easy (58.64%) 2320 -

給你一個(gè)二叉樹的根節(jié)點(diǎn) root , 檢查它是否軸對稱。

示例 1:

image.png
輸入:root = [1,2,2,3,4,4,3]
輸出:true

示例 2:

image.png
輸入:root = [1,2,2,null,3,null,3]
輸出:false

提示:

  • 樹中節(jié)點(diǎn)數(shù)目在范圍 [1, 1000] 內(nèi)
  • -100 <= Node.val <= 100

進(jìn)階:你可以運(yùn)用遞歸和迭代兩種方法解決這個(gè)問題嗎?


Discussion | Solution

2.1.遞歸法

/*
 * @lc app=leetcode.cn id=101 lang=cpp
 *
 * [101] 對稱二叉樹
 */

// @lc code=start
/**
 * 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:
  bool isSymmetric(TreeNode* root) { return dfs(root, root); }
  // 后續(xù)遍歷
 private:
  bool dfs(TreeNode* left, TreeNode* right) {
    if (left == nullptr && right == nullptr) return true;
    if (left == nullptr || right == nullptr) return false;
    if (left->val != right->val) return false;

    bool out_side = dfs(left->left, right->right);
    bool in_side = dfs(left->right, right->left);
    return in_side && out_side;
  }
};
// @lc code=end

2.2.迭代法-隊(duì)列

/*
 * @lc app=leetcode.cn id=101 lang=cpp
 *
 * [101] 對稱二叉樹
 */

// @lc code=start
/**
 * 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:
  bool isSymmetric(TreeNode* root) {
    if (root == nullptr) return true;
    std::queue<TreeNode*> que;
    que.push(root->left);
    que.push(root->right);

    while (!que.empty()) {
      TreeNode* left = que.front();
      que.pop();
      TreeNode* right = que.front();
      que.pop();

      // 判斷邏輯,與遞歸邏輯相同
      if (left == nullptr && right == nullptr) continue;
      if (left == nullptr || right == nullptr) return false;
      if (left->val != right->val) return false;

      // 外側(cè)
      que.push(left->left);
      que.push(right->right);
      // 內(nèi)側(cè)
      que.push(left->right);
      que.push(right->left);
    }
    return true;
  }
};
// @lc code=end

2.3.迭代法-棧

/*
 * @lc app=leetcode.cn id=101 lang=cpp
 *
 * [101] 對稱二叉樹
 */

// @lc code=start
/**
 * 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:
  bool isSymmetric(TreeNode* root) {
    if (root == nullptr) return true;
    std::stack<TreeNode*> st;
    st.push(root->left);
    st.push(root->right);

    while (!st.empty()) {
      TreeNode* left = st.top();
      st.pop();
      TreeNode* right = st.top();
      st.pop();

      // 判斷邏輯,與遞歸邏輯相同
      if (left == nullptr && right == nullptr) continue;
      if (left == nullptr || right == nullptr) return false;
      if (left->val != right->val) return false;

      // 外側(cè)
      st.push(left->left);
      st.push(right->right);
      // 內(nèi)側(cè)
      st.push(left->right);
      st.push(right->left);
    }
    return true;
  }
};
// @lc code=end
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容