LCP 67. 装饰树 –链接–
中等
10
相关企业
力扣嘉年华上的 DIY 手工展位准备了一棵缩小版的 二叉 装饰树 root 和灯饰，你需要将灯饰逐一插入装饰树中，要求如下：
完成装饰的二叉树根结点与 root 的根结点值相同
若一个节点拥有父节点，则在该节点和他的父节点之间插入一个灯饰（即插入一个值为 -1 的节点）。具体地：
在一个 父节点 x 与其左子节点 y 之间添加 -1 节点， 节点 -1、节点 y 为各自父节点的左子节点，
在一个 父节点 x 与其右子节点 y 之间添加 -1 节点， 节点 -1、节点 y 为各自父节点的右子节点，
现给定二叉树的根节点 root ，请返回完成装饰后的树的根节点。 示例 1：
输入： root = [7,5,6]
输出：[7,-1,-1,5,null,null,6]
解释：如下图所示，image.png
示例 2：
输入： root = [3,1,7,3,8,null,4]
输出：[3,-1,-1,1,null,null,7,-1,-1,null,-1,3,null,null,8,null,4]
解释：如下图所示image.png
该题思路简单，无非是利用递归走一遍树，在两节点之间插入-1，有一点我需要注意的是，递归函数的return 值，（该值可以不用），我们是不可以使用return中途停止递归和函数的执行，返回最初的调用函数的，因此我们也需要辨清此时调用的参数和返回值具体意义
class Solution {
public:
TreeNode* expandBinaryTree(TreeNode* root) {
if(root!=nullptr)
{
expandBinaryTree(root->left);
if(root->left!=nullptr)
{
TreeNode* T=new TreeNode(-1);
T->left=root->left;
root->left=T;
}
expandBinaryTree(root->right);
if(root->right!=nullptr)
{
TreeNode *T=new TreeNode(-1);
T->right=root->right;
root->right=T;
}
}
return root;
}
};
- 数组嵌套
中等
313
相关企业
索引从0开始长度为N的数组A，包含0到N - 1的所有整数。找到最大的集合S并返回其大小，其中 S[i] = {A[i], A[A[i]], A[A[A[i]]], … }且遵守以下的规则。
假设选择索引为i的元素A[i]为S的第一个元素，S的下一个元素应该是A[A[i]]，之后是A[A[A[i]]]… 以此类推，不断添加直到S出现重复的元素。
示例 1:
输入: A = [5,4,0,3,1,6,2]
输出: 4
解释:
A[0] = 5, A[1] = 4, A[2] = 0, A[3] = 3, A[4] = 1, A[5] = 6, A[6] = 2.
其中一种最长的 S[K]:
S[0] = {A[0], A[5], A[6], A[2]} = {5, 6, 2, 0}
使用染色法，又由题的的*S的下一个元素应该是A[A[i]]，之后是A[A[A[i]]]*这不就是提示我们用dfs了吗，哈哈哈哈
思路：
直接对数组进行dfs遍历（for循环），对每一轮遍历设置一个“颜色”，即一个标记，防止重复访问。然后选出遍历次数最多的数组元素
看代码
class Solution {
int Max,Cnt;
int hash[200010];
public:
void dfs(vector<int>&nums,int u,int color)
{
if(hash[u]!=-1)
{
return ;
}
hash[u]=color;
++Cnt;
dfs(nums,nums[u],color);
}
int arrayNesting(vector<int>& nums) {
int i;
int n=nums.size();
int color=0;
Max=0;
memset(hash,-1,sizeof(hash));
for(i=0;i<n;++i)
{
if(hash[i]==-1)
{
Cnt=0;
dfs(nums,i,++color);
Max=max(Max,Cnt);
}
}
return Max;
}
};
