移动数组中的元素

这篇博客讨论了如何在给定数组中移动元素,要求空间复杂度为 O(1)。文章提供了两种解法,包括数学角度的解法和原位移动的策略,重点介绍了原位移动的方法,通过求余确定元素位置并确保不覆盖原有元素,通过循环和计数器实现高效移动。

移动数组中的元素

问题描述

给定一个数组,将数组中的元素向右移动 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]

示例 2:

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

说明:

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

解法

数组中元素的移动是数组插入、删除时常遇到的,也是数组效率比较低的操作,所以掌握移动方法是很重要的。

首先提供一种开辟新的内存空间的解法,虽然这不符合题目要求,但是这是一种从数学角度出发的解法,可以借鉴。

    public void rotate(int[] nums, int k) {
        int[] temp=new int[nums.length];
        for(int i=0;i<nums.length;i++){
            temp[(i+nums.length+k)%nums.length]=nums[i];
        }
        for(int i=0;i<temp.length;i++){
            nums[i]=temp[i];
        }

    }

下一种解法是原位移动的方法:

image.png

以[1,2,3,4,5,6],k=2为例,如上图,继续沿用通过求余的方法确定元素位置,但是关键是移动一个就得跟着移动下一个,何时结束呢?

例如:上图我们先移动1至3 的位置,为了避免覆盖,我们将3记录下来,那么3继续移动到5的位置,记录下5,5又移动到1原来的位置,所以结束的条件就是回到了最初的位置,所以我们需要一个int变量来保存原来的数,也需要一个int变量标识最开始的位置,并且还要一个int变量来记录当前位置,当当前位置到达最初位置时,结束循环,继续遍历。最终当所有数都被移动过了,那程序结束,所以还要一个count来记录移动次数,次数等于元素个数时结束程序。

    public void rotate(int[] nums, int k) {
            k = k % nums.length;
            int count = 0;
            for (int start = 0; count < nums.length; start++) {
                int current = start;
                int prev = nums[start];
                do {
                    int next = (current + k) % nums.length;
                    int temp = nums[next];
                    nums[next] = prev;
                    prev = temp;
                    current = next;
                    count++;
                } while (start != current);
        }
    

count++;
} while (start != current);
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值