leetcode 543. 二叉树的直径

leetcode 543. 二叉树的直径

题目详情

题目链接
给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。

• 示例:
  给定二叉树
  
  返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
  注意：两结点之间的路径长度是以它们之间边的数目表示。

我的代码

#include <vector>
#include <algorithm>

using namespace std;

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

class Solution {
public:
    int res = 0;

    int getHeight(TreeNode *pNode) {
        if ((pNode->left == NULL) && (pNode->right == NULL)) {
            return 0;
        }
        if ((pNode->left != NULL) && (pNode->right != NULL)) {
            return max(getHeight(pNode->left), getHeight(pNode->right)) + 1;
        }
        if (pNode->left != NULL) {
            return 1 + getHeight(pNode->left);
        } else {
            return 1 + getHeight(pNode->right);
        }
    }

    int getRootDistance(TreeNode* root) {
        int leftHeight = root->left == NULL? 0:  getHeight(root->left) + 1;
        int rightHeight = root->right == NULL? 0: getHeight(root->right) + 1;
        return leftHeight + rightHeight;
    }

    int diameterOfBinaryTree(TreeNode* root) {
        if (root == NULL) {
            return 0;
        }
        res = max(res, getRootDistance(root));
        if (root->left != NULL) {
            diameterOfBinaryTree(root->left);
        }
        if (root->right != NULL) {
            diameterOfBinaryTree(root->right);
        }
        return res;
    }
};

我的成绩

执行结果：通过
执行用时 :60 ms, 在所有 C++ 提交中击败了13.16%的用户
内存消耗: 43.6 MB, 在所有 C++ 提交中击败了5.16%的用户

一些想法

1. 最长直径不一定过根节点，所以必须比较所有内部节点的经过的最长直径。
2. 递归时要善于使用全局变量

执行用时为 0 ms 的范例

// An highlighted block
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int ans;
    int diameterOfBinaryTree(TreeNode* root) {
        ans = 1;
        depth(root);
        return ans - 1;
    }
    int depth(TreeNode *node)
    {
        if(node == NULL)
            return 0;
            int L = depth(node->left);
            int R = depth(node->right);
            ans = max(ans, L + R + 1);
            return max(L, R) + 1;
    }
};

思考

1. 我的代码过于冗余，进行了多次没必要的遍历。
2. 要善于使用全局变量。