您现在的位置是：首页 >技术交流 >力扣总结，深度优先题网站首页技术交流

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;
    }
};

565. 数组嵌套
     中等
     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;
    }
};