leetcode872.叶子相似的树

孤者浪人 提交于 2020-01-23 17:01:10

题目描述

请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。

举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。

如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。

如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。

提示:

给定的两颗树可能会有 1 到 100 个结点。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/leaf-similar-trees

解法:

先遍历把两棵树叶子节点的值分别存入数组,再判断两个数组是否相等。

class Solution {
public:
	bool leafSimilar(TreeNode* root1, TreeNode* root2) {
		vector<int> arr1;
		vector<int> arr2;
		getLeafArray(root1, arr1);
		getLeafArray(root2, arr2);
		for (int i = 0; i < arr1.size(); i++) {
			if (arr1[i] != arr2[i]) {
				return false;
			}
		}
		return true;
	}
	void getLeafArray(TreeNode* root, vector<int>& arr) {
		if (root == NULL) {
			return;
		}
		getLeafArray(root->left, arr);
		getLeafArray(root->right, arr);
		if (root->left == NULL && root->right == NULL) {
			arr.push_back(root->val);
		}
	}
};
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!