Rotate Array

2017-07-05  本文已影响0人  BigBig_Fish

Rotate an array of n elements to the right by k steps.

For example, with n = 7 and k = 3, the array [1,2,3,4,5,6,7] is rotated to [5,6,7,1,2,3,4].

Note:
Try to come up as many solutions as you can, there are at least 3 different ways to solve this problem.

题目分析

将整个数组向右移动k位,那么第i位的目标位置就是(i+k)%len

代码

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        vector<int> vec(nums);
        int len=nums.size();
        for(int i=0; i<len; i++) {
            nums[(i+k)%len] = vec[i];
        }
        return ;
    }
};

倒置的思路

将后k位倒置和前n-k位倒置,然后将整个数组倒置

class Solution 
    {
    public:
        void rotate(int nums[], int n, int k) 
        {
            k = k%n;
    
            // Reverse the first n - k numbers.
            // Index i (0 <= i < n - k) becomes n - k - i.
            reverse(nums, nums + n - k);
            
            // Reverse tha last k numbers.
            // Index n - k + i (0 <= i < k) becomes n - i.
            reverse(nums + n - k, nums + n);
            
            // Reverse all the numbers.
            // Index i (0 <= i < n - k) becomes n - (n - k - i) = i + k.
            // Index n - k + i (0 <= i < k) becomes n - (n - i) = i.
            reverse(nums, nums + n);
        }
    };
上一篇 下一篇

猜你喜欢

热点阅读