leetCode #283 move zeroes

大家好，我是小鸭酱，博客地址为：http://www.cnblogs.com/xiaoyajiang

解法一

->  从nums[0]开始，依次扫描0的个数count，每个元素都应该向前移动count个位置，最后再将最后count个元素赋值为0

实现如下：

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int count = 0;
        for (int i = 0; i < nums.size(); ++i)
        {
            if (nums[i] != 0)
            {
                nums[i - count] = nums[i];
            }
            else
            {
                ++count;
            }
        }
        while (count)
        {
            nums[nums.size() - count] = 0;
            --count;
        }
    }
};

 

解法二

->  扫描一遍nums中的元素，用p来表示可以放置非零元素的索引。只有当扫描的元素非零时，我们需要将这个元素放置到p的位置，并将p的位置再向后挪。

或这样理解：

->  将nums中非零元素全部放置在[0,p) 这个区间，剩余的部分赋值为0

代码如下

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int p = 0;
        for (int i = 0; i < nums.size(); ++i)
        {
            if (nums[i])
            {
                nums[p++] = nums[i];
            }
        }
        for (int j = p; j < nums.size(); ++j)
        {
            nums[j] = 0;
        }
    }
};

 

细心的你可能已经发现了，解法一和解法二本质是一样的，只是一个记录的是跨度count，一个记录的是下标p。

 

解法三

我们对上面的步骤进行优化，也就是把赋值（=）换成交换（swap()），这样就无需再对后面的0元素赋值了。

代码如下：

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int p = 0;
        for (int i = 0; i < nums.size(); ++i)
        {
            if (nums[i])
            {
                swap(nums[p++], nums[i]);
            }
        }
    }
};

 

优化解法三

考虑没有0元素的情况，那么原始的解法三会出现元素与自己交换的情况，对此，我们对两种情况进行处理，即，若i==p和i!=p这两种情况，前者并不需要做swap。

改进后的代码如下：

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int p = 0;
        for (int i = 0; i < nums.size(); ++i)
        {
            if (nums[i])
            {
                if (i != p)
                {
                    swap(nums[p++], nums[i]);    
                }
                else
                {
                    ++p;
                }
            }
        }
    }
};