代码挑战:迁移0元素|Java
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:
You must do this in-place without making a copy of the array.
Minimize the total number of operations.
方案:
class Solution {
public void moveZeroes(int[] nums) {
int index = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] != 0) {
nums[index++] = nums[i];
}
}
while (index < nums.length) {
nums[index++] = 0;
}
}
}
测试案例:
@Test
public void test1() {
Solution solution = new Solution();
int[] nums = { 0, 1, 0, 3, 12 };
solution.moveZeroes(nums);
int[] expected = { 1, 3, 12, 0, 0 };
for (int i = 0; i < nums.length; i++)
{
Assert.assertEquals(nums[i], expected[i]);
}
}
@Test
public void test2() {
Solution solution = new Solution();
int[] nums = { 1, 9, 8, 4, 0, 0, 2, 7, 0, 6, 0, 9 };
solution.moveZeroes(nums);
int[] expected = { 1, 9, 8, 4, 2, 7, 6, 9, 0, 0, 0, 0 };
for (int i = 0; i < nums.length; i++)
{
Assert.assertEquals(nums[i], expected[i]);
}
- 点赞
- 收藏
- 关注作者
评论(0)