注:本系列题目来自LeetCode(https://leetcode-cn.com/)
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
解题思路:
遇到这种题总是第一时间想到逐个元素循环遍历,但确实是复杂又容易出错。借鉴别人的思路,统计0元素个数,跳过相应数目进行非0元素的交换。另外,本题目不需要函数有返回值,因为是在原数组上修改。
还是要多做题多思考,这样才能开阔视野,熟能生巧。
Python:
class Solution:
def moveZeroes(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
numzero = 0
for i in range(len(nums)):
if (nums[i] == 0):
numzero = numzero + 1
if (nums[i] != 0 and numzero != 0):
nums[i - numzero] = nums[i]
nums[i] = 0
C++
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int index = 0;
for(int i = 0; i < nums.size(); ++i)
{
if(nums[i] != 0)
{
nums[index++] = nums[i];
}
}
while(index < nums.size())
{
nums[index++] = 0;
}
}
};