将数组的元素向右移动

Question

我知道有一些解决方案可以移动阵列。但是没有解决方案适合我。代码应实现如下：

shift(int[] array, int places) 方法接收 array，将元素 places 向右移动 次并将“剩余”元素替换为“0”。

到目前为止我有：

public static int[] shiftWithDrop(int[] array, int places) {
            
    if (places == 0 || array == null) {
        return null;
    }
    for (int i = array.length-places-1; i >= 0; i-- ) {
        array[i+places] = array[i];
        array[i] = 0;
    }               
    return array;
}

此代码只能以某种方式工作，但不会返回所需的结果。我错过了什么？

Answer 1

这段代码有几个问题：

1. places == 0时返回null -- 不移位，需要返回原始数组
2. 在给定的循环实现中，可能会跳过数组的主要部分，而不是将第一个 places 元素替换为 0，实际上数组开头的一些元素设置为 0。

另外最好更改方法的签名以在可变参数array.之前设置位置

因此，为了解决这些问题，提供了以下解决方案：

public static int[] shiftWithDrop(int places, int ... array) {
    if(array == null || places <= 0) {
        return array;
    }
    for (int i = array.length; i-- > 0;) {
        array[i] = i  < places ? 0 : array[i - places];
    }
    
    return array;
}    

测试：

System.out.println(Arrays.toString(shiftWithDrop(1, 1, 2, 3, 4, 5)));
System.out.println(Arrays.toString(shiftWithDrop(2, new int[]{1, 2, 3, 4, 5})));
System.out.println(Arrays.toString(shiftWithDrop(3, 1, 2, 3, 4, 5)));
System.out.println(Arrays.toString(shiftWithDrop(7, 1, 2, 3, 4, 5)));

输出：

[0, 1, 2, 3, 4]
[0, 0, 1, 2, 3]
[0, 0, 0, 1, 2]
[0, 0, 0, 0, 0]