283. 移动零

2020-11-19  本文已影响0人  infinite_stack

给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。

示例:

输入: [0,1,0,3,12]
输出: [1,3,12,0,0]

说明:

必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。

解法一(效率差)

双重循环
外层:遍历元素
内层:往后找到第一个不是0的元素
交换0和之后第一个不是0的元素
后面全是0,提前退出循环

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == 0) {
                int k = -1;
                for (int j = i+1; j < nums.size(); j++) {
                    if (nums[j] != 0) {
                        k = j;
                        break;
                    }
                }
                if (k == -1)
                    return;
                int t = nums[i];
                nums[i] = nums[k];
                nums[k] = t;
            }
        }
    }
};

解法二(常规解法,必须掌握)

记录z为写入下标,非0值移动到该下标处,0跳过,遍历一遍后非0值全部移动到z之前了,这时候将[z,n)下标的置0

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int z = 0;
        int n = nums.size();
        for (int i = 0; i < n; i++) {
            if (nums[i] != 0) {
                nums[z++] = nums[i];
            }
        }
        while (z < n) {
            nums[z++] = 0;
        }
    }
};
上一篇 下一篇

猜你喜欢

热点阅读