《代码随想录》：移除元素

最新推荐文章于 2024-10-15 17:06:18 发布

小李童学

最新推荐文章于 2024-10-15 17:06:18 发布

阅读量56

点赞数

文章标签：算法 Powered by 金山文档

本文链接：https://blog.csdn.net/m0_72984169/article/details/129380355

版权

文章提供了两种解决LeetCode第27题的方法。第一种是暴力破解，通过双重循环移除指定值，时间复杂度为O(n^2)。第二种方法是双指针法（快慢指针），以O(n)的时间复杂度更高效地完成任务。两种方法都满足原地修改输入数组且额外空间复杂度为O(1)的要求。

摘要由CSDN通过智能技术生成

力扣题号：27 移除元素

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

方法1：暴力破解

#include<bits/stdc++.h>
using namespace std;
//暴力破解法
class Solution{
    public:
        int removeElement(vector<int>& nums,int val){
            int size = nums.size();
            for(int i=0;i<size;i++){
                if(nums[i]==val){
                    for(int j=i+1;j<size;j++){
                        nums[j-1]=nums[j];
                    }
                i--;
                size--;
                }
            }
            return size;
        }
};

int main(){
    vector<int> nums;
    int val;
    int a;
    cout<<"输入数组:"<<endl;
    while(cin>>a){
        nums.push_back(a);
        char c = cin.get();
        if(c=='\n'){
            break;
        }
    }
    cout<<"输入移除数："<<endl;
    cin>>val;
    Solution S1;
    int size=S1.removeElement(nums,val);
    cout<<"移除后数组长度为："<<size<<endl;
    system("pause");
    return 0;
}

注意点：

二重循环提里j=i+1

i--和size--容易遗忘，需谨记

时间复杂度：O（n2）

空间复杂度：O（1）

方法2：双指针法（快慢指针法）

#include<bits/stdc++.h>
using namespace std;
//双指针法
class Solution{
    public:
        int removeElement(vector<int>& nums, int val){
            int slowIndex=0;
            for(int fastIndex=0;fastIndex<nums.size();fastIndex++){
                if(val!=nums[fastIndex]){
                    nums[slowIndex++]=nums[fastIndex];
                }
            }
            return slowIndex;
        }
};

int main(){
    vector<int> nums;
    int val;
    int a;
    cout<<"输入数组："<<endl;
    while(cin>>a){
        nums.push_back(a);
        char c=cin.get();
        if(c=='\n'){
            break;
        }
    }
    cout<<"输入移除数："<<endl;
    cin>>val;
    Solution S1;
    int size = S1.removeElement(nums,val);
    cout<<"移除后数组长度为："<<size<<endl;
    system("pause");
    return 0;
}

时间复杂度：O（n）

空间复杂度：O（1）

注：

双指针很常用