leetcode题解(2)

题目描述:
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。

示例 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]

来源:力扣(LeetCode)


c/c++

  • 第一次尝试:

思路:时间复杂度O(N),需要额外的内存空间 这不是一个好方法:
1,创建一个同等长度的数组
2,根据移动后的索引大小是否大于原数组大小 将原数组中的每个值都放入空数组中对应的位置上
arr[i + (k % numsSize)] = nums[i] / arr[i + (k % numsSize) - numsSize] = nums[i];
3,把新数组复制给原数组,程序完毕

void rotate(int* nums, int numsSize, int k) {
	
	// int arr[numsSize];
	int * arr;
	arr = (int*)malloc(sizeof(int) * numsSize);

	for (int i = 0; i < numsSize; i++) {

		if ((i + (k % numsSize)) < numsSize) {
			arr[i + (k % numsSize)] = nums[i];
		}
		else {
			arr[i + (k % numsSize) - numsSize] = nums[i];
		}

	}
	for (int i = 0; i < numsSize; i++) {
		nums[i] = arr[i];
	}
}

执行结果:

  • 参考后修改:

新思路:三次旋转

三次旋转
观察可得:数组的翻转中,有(k % Size)个尾部元素将会移动到数组头部,其余的头部元素会后移成为尾部元素
因此:我们首先将整个数组翻转,然后将头部(k % l)个头部元素翻转,最后再将其余尾部元素翻转 (3次旋转)

c

void reverse(int *a, int l, int h)
{
    for(int i = 0; i < (h - l + 1) / 2; i++){
        int t = a[l + i];
        a[l + i] = a[h - i];
        a[h - i] = t;
    }
}
void rotate(int* nums, int numsSize, int k){
    if(!nums || numsSize == 0)
        return;
    k = k % numsSize;
    reverse(nums, 0, numsSize - 1 - k);
    reverse(nums, numsSize - k, numsSize - 1);
    reverse(nums, 0, numsSize - 1);
}

执行结果:

c++

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
       int len=nums.size();

       reverse(nums.begin(), nums.end() - k%len); //reverse翻转的是[begin,end)区间
       reverse(nums.end()- k%len,nums.end());
       reverse(nums.begin(),nums.end());
    }
};

执行结果:


python代码实现三次旋转

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n=len(nums)
        k=k%n
        def swap(l,r):
            while(l<r):
                nums[l],nums[r]=nums[r],nums[l]
                l=l+1
                r=r-1
        swap(0,n-k-1)
        swap(n-k,n-1)
        swap(0,n-1)


java代码实现实现三次旋转*

public class Solution {
    public void rotate(int[] nums, int k) {
        k %= nums.length;
        reverse(nums, 0, nums.length - 1);
        reverse(nums, 0, k - 1);
        reverse(nums, k, nums.length - 1);
    }
    public void reverse(int[] nums, int start, int end) {
        while (start < end) {
            int temp = nums[start];
            nums[start] = nums[end];
            nums[end] = temp;
            start++;
            end--;
        }
    }
}

总结:

1.在局部函数中创建一个数组,不能直接使用(int arr[N])创建,这样创建出来的数组只是一个局部变量,无法传出来,而应该使((int*) malloc(sizeof()))动态创建。

2,原地翻转一个数组的方法 : 三次旋转法 环形旋转

发布了12 篇原创文章 · 获赞 29 · 访问量 2674

猜你喜欢

转载自blog.csdn.net/a13352912632/article/details/104066532
今日推荐