Remove Element的两种解法

标签：

/*************************************************************************
	> 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的两种解法

标签：

原文地址：http://blog.csdn.net/u011487593/article/details/45674673