代码挑战：迁移0元素|Kotlin

Given an array nums, write a function to move all 0's to the end of it while maintaining the relative order of the non-zero elements.

Example:

Input: [0,1,0,3,12]
Output: [1,3,12,0,0]

Note:

1. You must do this in-place without making a copy of the array.

2. Minimize the total number of operations.

方案：

internal class Solution {
    fun moveZeroes(nums: IntArray) {
        var index = 0
        for (i in nums.indices) {
            if (nums[i] != 0) {
                nums[index++] = nums[i]
            }
        }
        while (index < nums.size) {
            nums[index++] = 0
        }
    }
}

测试案例：

@Test
fun test1() {
    val solution = Solution()
    val nums = intArrayOf(0, 1, 0, 3, 12)
    solution.moveZeroes(nums)
    val expected = intArrayOf(1, 3, 12, 0, 0)
    for (i in nums.indices) {
        Assert.assertEquals(nums[i], expected[i])
    }
}

@Test
fun test2() {
    val solution = Solution()
    val nums = intArrayOf(1, 9, 8, 4, 0, 0, 2, 7, 0, 6, 0, 9)
    solution.moveZeroes(nums)
    val expected = intArrayOf(1, 9, 8, 4, 2, 7, 6, 9, 0, 0, 0, 0)
    for (i in nums.indices) {
        Assert.assertEquals(nums[i], expected[i])
    }
}