543二叉树的直径（递归dfs深度优先搜索+备忘录、树的高度）

最新推荐文章于 2024-11-01 15:19:21 发布

Revendell

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量245

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/Revendell/article/details/112273099

版权

二叉树深度优先搜索备忘录路径长度递归

关键词由CSDN通过智能技术生成

LeetCode 专栏收录该内容

368 篇文章 20 订阅

订阅专栏

1、题目描述

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

注意：两结点之间的路径长度是以它们之间边的数目表示。

2、示例
给定二叉树

1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

3、题解

基本思想：递归dfs深度优先搜索+备忘录，递归计算每个节点的高度并记录到path=max(左节点高度,右节点高度)+1。在这个过程中顺便计算该节点路径=左节点高度+右节点高度。

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<unordered_map>
using namespace std;
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
#define inf 9999
void Init_TreeNode(TreeNode** T, vector<int>& vec, int& pos)
{
	if (vec[pos] == inf || vec.size() == 0)
	{
		*T = NULL;
		return;
	}
	else
	{
		(*T) = new TreeNode(0);
		(*T)->val = vec[pos];
		Init_TreeNode(&(*T)->left, vec, ++pos);
		Init_TreeNode(&(*T)->right, vec, ++pos);
	}
}
class Solution {
public:
    unordered_map<TreeNode*,int> path;
    int res=0;
    int diameterOfBinaryTree(TreeNode* root) {
        //基本思想：递归dfs深度优先搜索+备忘录，递归计算每个节点的高度并记录到path=max(左节点高度,右节点高度)+1
        //在这个过程中顺便计算该节点路径=左节点高度+右节点高度
        height(root);
        return res;
    }
    int height(TreeNode* root)
    {
        if(root==NULL)
            return 0;
        if(path.find(root)!=path.end())
            return path[root];
        int lh=height(root->left);
        int rh=height(root->right);
        res=max(res,lh+rh);
        path[root]=max(lh,rh)+1;
        return max(lh,rh)+1;
    }
};
int main()
{
	TreeNode* root = NULL;
	vector<int> vec = { 2,1,4,inf,inf,5,6,inf,inf,inf,3,inf,inf };
	int pos = 0;
	Init_TreeNode(&root, vec, pos);
	Solution solute;
	cout<<solute.diameterOfBinaryTree(root)<<endl;
	return 0;
}