leetcode array nesting

本来代码

#include<iostream>

#include<vector>

using namespace std;

class Solution {

public:

    int arrayNesting(vector<int>& nums) {

        int n=(int)nums.size();

        int max=-1;

        for(int i=0;i<=n-1;i++)

        {

            int a[n];

            for(int i=0;i<=n-1;i++)

                a[i]=-1;

            int count=0;

            int t=nums[i]; a[t]=1;  count++;

            while(a[nums[t]]==-1)

            {

                a[nums[t]]=1;

                t=nums[t];

                count++;

            }

            max= max>count? max:count;

            

        }

        return max;

    }

};

两组数据超时

改进后

#include<iostream>

#include<vector>

using namespace std;

class Solution {

public:

    int arrayNesting(vector<int>& nums) {

        int n=(int)nums.size();

        int max=-1;

        int visited[n+1];

        for(int i=0;i<=n;i++)

            visited[i]=0;

        for(int i=0;i<=n-1;i++)

        {

            if(visited[i]==1)

                continue;

            int a[n];

            for(int i=0;i<=n-1;i++)

                a[i]=-1;

            int count=0;

            int t=nums[i]; a[t]=1; visited[t]=1; count++;

            while(a[nums[t]]==-1)

            {

                a[nums[t]]=1;

                visited[nums[t]]=1;

                t=nums[t];

                count++;

            }

            max= max>count? max:count;

            

        }

        return max;

    }

};