27. 移除元素(C++)

题干:
https://leetcode.cn/problems/remove-element/

  • 双指针

和本专栏中的“ 26 删除有序数组中的重复项”很相似,我们采用双指针解决问题。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {

        //if (nums.empty()) { throw "the array is empty!"; } //测试中有空数组哈哈哈,很尴尬不符合答案
        if (nums.empty()) { return 0; }

        int fir{ 0 };//第一个指针,指向被覆盖的空间
        int sec{ 0 };//第二个指针,遍历数组

        while (sec != nums.size()) {
            if (nums[sec] == val) {   }
            else {
                nums[fir] = nums[sec];
                fir++;
            }
            sec++;//因为sec++在if-else中都需要递增,所以写到了这里
        }

        return fir;//fir代表返回数组的size(),fir-1是返回数组的最大下标
    }
};

优化后的代码,赋值操作变少了。很值得一看(代码写得越复杂,逻辑问题可能就越多,像这样简洁的代码才是我们追求的)

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {

        if (nums.empty()) { return 0; }
		//注意区间的开闭
        int left{ 0 };//left左边区间 [0,left) 全不等于val
        int right{ static_cast<int>(nums.size()-1) };//right右边区间(right,size)全等于val

        while (left <= right) {
            if (nums[left] == val) {
                nums[left] = nums[right];
                right--;//只有确保了right被赋值为val,right才左移
            }
            else {
                left++;//只有left不等于val,left才右移
            }
        }

        return left;//或者right+1
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值