/*************************************************************************
> File Name: removeElement.cpp
> Author: CourageK
> Mail: kangshaojie@ict.ac.cn
> Created Time: 2015年05月11日 星期一 21时40分41秒
Problem Description:
Given an array and a value, remove all instances of that value in place and return the new length.
The order of elements can be changed. It doesn't matter what you leave beyond the new length.
************************************************************************/
#include<iostream>
#include<algorithm>
#include<vector>
#include<iterator>
using namespace std;
class Solution
{
public:
/*
*Solution 1:顺序遍历删除元素,时间复杂度为O(n)
int removeElement(vector<int>& nums, int val)
{
if(nums.empty()) return 0;
vector<int>::iterator first = nums.begin();
vector<int>::iterator last = nums.end();
// 如果在array中不存在val
if(find(first, last, val) == last)
return nums.size(); // 直接输出数组的大小
// 如果在array中存在val
int lenOfNew = 0;
while(first != nums.end())
{
if(*first == val)
{
first = nums.erase(first); //删除该元素后,first指向这个被删元素的后面
lenOfNew++;
}
else first++;
}
return nums.size();
}
*/
// Solution 2:对数组中的元素由小到大排序,将使得大小相等的数相邻排列,有利于删除(小数据量时采用插入排序,时间复杂度为:O(n^2)
int removeElement(vector<int>& nums, int val)
{
if(nums.empty()) return 0;
vector<int>::iterator first = nums.begin();
vector<int>::iterator last = nums.end();
// 如果在array中不存在val
if(find(first, last, val) == last) // 顺序查找,时间复杂度为O(n)
return nums.size(); // 直接输出数组的大小
// 如果在array中存在val
sort(first, last); // 对数组中的元素由小到大排序,将使得大小相等的数相邻排列,有利于删除(小数据量时采用插入排序,时间复杂度为:O(n^2)
vector<int>::iterator lower = lower_bound(first, last, val); // 找到第一个与val值相等的元素位置,O(logn)
vector<int>::iterator upper = upper_bound(first, last, val); // 找到最后一个与val值相等的元素的下一个位置,O(logn)
nums.erase(lower, upper); // 利用erase函数删除重复的相邻元素
return nums.size();
}
};
Remove Element的两种解法
最新推荐文章于 2024-09-11 02:12:10 发布