数组嵌套(力扣第565题)

565.数组嵌套

​ 索引从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}

提示:

N是[1, 20,000]之间的整数。
A中不含有重复的元素。
A中的元素大小在[0, N-1]之间。

分析:

​ 可以将S集合看作是一个环形链表,那么集合的大小就是环形链表的长度。因为数组的索引是0N-1,数组的元素值也是0N-1,所以元素值就可以看作是下一个结点的“地址”,比如对于示例1,S[0]集合来说,就是形成了这样一个环形链表:

​ 0->5->6->2->0,

​ 可以定义一个队首指针head,head = nums[0],然后定义一个遍历指针变量point,初始值指向队首元素的下一个元素:point=nums[head],然后它进行的向后走的操作是:point=nums[point]

​ 当head == point时,说明环形链表走完。

​ 因此,我们对数组中每个元素进行这样的操作之后,求出最大值就是最终的结果。

public int arrayNesting(int[] nums) {
    if (nums == null || nums.length == 0){
        return 0;
    }
    int maxLength = 0;
    for (int i = 0; i < nums.length; i++) {

            int head = nums[i];
            int point = nums[head];
            int curLength = 1;
            while (point != head) {
                point = nums[point];
                curLength++;
            }
            if (curLength > maxLength) {
                maxLength = curLength;
            }

    }
    return maxLength;
}

​ 该方法虽然解决了问题,但是时间复杂度太高,性能不好,有点像暴力法了。其实可以优化一下的,因为我们要求的是长度最大的那个环形链表的长度,而每个环形链表都是一个闭环,即环外的元素永远不会指向环中的元素,环中的元素也不会指向环外元素,求所有的S集合的操作相当于将原始数组的元素划分为了几个不相交的集合。那么既然是一个闭环,那么从环中任何一个元素出发,最终形成的都是相同的环,长度自然也就相同。既然这样就无需对原始数组逐个遍历求长度了,而是对于同一个环中的元素,访问过之后就无需再访问了。这样会大大节省时间。

​ 例如:

​ S[0] : 0->5->6->2->0

​ S[2] : 2->0->5->6->2

​ S[5] : 5->6->2->0->5

public int arrayNesting2(int[] nums) {
    if (nums == null || nums.length == 0){
        return 0;
    }
    int maxLength = 0;
    for (int i = 0; i < nums.length; i++) {

        int count = 0;

        for (int j = i; nums[j] != -1 ; ) {
            count++;
            int temp = nums[j];
            nums[j] = -1;
            j = temp;
        }

        if (count > maxLength){
            maxLength = count;
        }
    }

    return maxLength;
}
posted @ 2020-11-26 13:27  有心有梦  阅读(138)  评论(0编辑  收藏  举报