Leetcode 543. 二叉樹的直徑

1、問題分析

題目鏈接：https://leetcode-cn.com/problems/diameter-of-binary-tree/
  本質上就是一個二叉樹的遍歷問題。代碼我已經進行了詳細的註釋，理解應該沒有問題，讀者可以作爲參考，如果看不懂(可以多看幾遍)，歡迎留言哦！我看到會解答一下。

2、問題解決

  筆者以C++方式解決。

#include "iostream"

using namespace std;

#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;

    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
private:
    // 定義最大直徑 初始值爲 0
    int result = 0;
public:
    int diameterOfBinaryTree(TreeNode *root) {
        // 如果樹爲空 直接返回 0
        if (root == NULL) {
            return 0;
        }
        // 尋找最大直徑
        dfs(root);

        // 返回處理後的結果
        return result;
    }

    /**
     * 計算 以 root 節點爲根節點的樹的高度
     * @param root
     * @return
     */
    int dfs(TreeNode *root) {
        // 如果節點爲空 返回 -1
        if (root == NULL) {
            return -1;
        }

        // 如果是葉子節點 返回高度爲 0
        if (root->left == nullptr && root->right == nullptr) {
            return 0;
        }

        TreeNode *rootLeft = root->left;
        TreeNode *rootRight = root->right;

        // 分別計算左右子樹的高度
        int a = dfs(rootLeft) + 1;
        int b = dfs(rootRight) + 1;

        // 左右子樹的高度相加，就是二叉樹的直徑
        // 選取較大的保存
        if (a + b > result) {
            result = a + b;
        }

        // 返回左右子樹中高度較大的返回作爲樹的高度
        return max(a, b);
    }
};

int main() {
    TreeNode *pNode1 = new TreeNode(1);
    TreeNode *pNode2 = new TreeNode(2);
    TreeNode *pNode3 = new TreeNode(3);
    TreeNode *pNode4 = new TreeNode(4);
    TreeNode *pNode5 = new TreeNode(5);
//    TreeNode *pNode6 = new TreeNode(6);
//    TreeNode *pNode7 = new TreeNode(7);

    pNode1->left = pNode2;
    pNode1->right = pNode3;
    pNode2->left = pNode4;
    pNode2->right = pNode5;
//    pNode4->left = pNode6;
//    pNode5->right = pNode7;

    Solution *pSolution = new Solution;
    int i = pSolution->diameterOfBinaryTree(pNode1);
    cout << i << endl;
    system("pause");
    return 0;
}

運行結果

有點菜，有時間再優化一下。

3、總結

  難得有時間刷一波LeetCode, 這次做一個系統的記錄，等以後複習的時候可以有章可循，同時也期待各位讀者給出的建議。算法真的是一個照妖鏡，原來感覺自己也還行吧，但是算法分分鐘教你做人。前人栽樹，後人乘涼。在學習算法的過程中，看了前輩的成果，受益匪淺。
感謝各位前輩的辛勤付出，讓我們少走了很多的彎路！
哪怕只有一個人從我的博客受益，我也知足了。
點個贊再走唄！歡迎留言哦！