LeetCode 189. 旋转数组 C++& java

给定一个数组,将数组中的元素向右移动 个位置,其中 是非负数。

示例 1:

输入: [1,2,3,4,5,6,7]k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右旋转 1 步: [7,1,2,3,4,5,6]
向右旋转 2 步: [6,7,1,2,3,4,5]
向右旋转 3 步: [5,6,7,1,2,3,4]

示例 2:

输入: [-1,-100,3,99]k = 2
输出: [3,99,-1,-100]
解释: 
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]

说明:

  • 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
  • 要求使用空间复杂度为 O(1) 的原地算法。

C++:   感觉我写的这种算法 效率不是很高 ,尤其是java的时候格外累赘 , 应该是可以不用定义一个新数组就能完成的,不过我没想出来,只想到了这种笨方法 。

//   思路比较简单 ,将要右移的数放到一个新的数组里,然后将剩下的补齐

class Solution {

public:
    void rotate(vector<int>& nums, int k) {

        if(k>=nums.size())
        {
            k -= nums.size() ;
            return rotate( nums,k) ;
        }
        vector<int> ans(nums.size()) ;
        for(int i = 0 ; i<nums.size();i++)
        {
            if(i<k)
            {
                ans[i] = nums[nums.size()+i-k] ;
            }
            else
            {
                ans[i] = nums[i - k] ;
            }
        }
        nums = ans ;
    }

};

JAVA:

class Solution {
    public void rotate(int[] nums, int k) {
        int[] ans = new int[nums.length];
        if(k>=nums.length)
        {
            k -=nums.length;
            rotate(nums, k) ;
        }
        else
        {
                for(int i = 0;i< nums.length;i++)
                {
                    if(i<k)
                    {
                        ans[i] = nums[nums.length + i-k] ;
                    }
                    else
                    {
                        ans[i] = nums[ i-k] ;
                    }
                }
                for(int i = 0;i< nums.length;i++)     
                {
                    nums[i] = ans[i] ;
                }
        }         
    }
}


猜你喜欢

转载自blog.csdn.net/wx734518351/article/details/80464277
今日推荐