GEEK编程练习— —数组去重

数组消除重复元素的题目一般来说都是最基本的，没什么太大的难度，而且大部分时候直接使用STL算法或容器的性质就可搞定。让我们来看题目吧~

题目一

输入一个已经排好序的数组，去除重复元素，输出新数组的长度。
要求：在连续的内存中进行操作，不要分配额外的内存空间给另一个数组。
举栗：输入[1,1,2]，输出2。

分析

如果可以额外创建容器，可以使用set，先把vector的值给set，然后再传回来。因为set本身就具有无重复的性质。下面先会给出用set的代码，当然不适合本题目。那么不可以额外创建，还可以使用STL算法unique。
如果不使用STL算法，我们可以考虑，自己写循环进行判断操作。

代码

set版本（非本题答案）

//Remove Duplicates from Sorted Array
//Date:2016-04-05
//Author:Sin_Geek
//时间负责度O(n)，空间复杂度O(n)

#include <iostream>
#include <vector>
#include <set>

using namespace std;

int main()
{
    vector<int> a{1, 1, 2, 2, 3};
    set<int> b(a.cbegin(), a.cend());
    a.assign(b.cbegin(), b.cend());

    cout << a.size() <<endl;

    return 0;
}

unique版本

//Remove Duplicates from Sorted Array
//Date:2016-04-05
//Author:Sin_Geek
//时间负责度O(n)，空间复杂度O(1)

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    vector<int> a{1, 1, 2, 2, 3};

    cout << distance(a.begin(), unique(a.begin(), a.end())) << endl;

    return 0;
}

循环判断

//Remove Duplicates from Sorted Array
//Date:2016-04-05
//Author:Sin_Geek
//时间负责度O(n)，空间复杂度O(1)

#include <iostream>
#include <vector>

using namespace std;

int main()
{
    vector<int> a{1, 1, 2, 2, 3};

    if (a.empty())
    {
       cout << 0 << endl;
       return 0;
    }


    int index = 0;
    for (int i = 1; i < a.size(); ++i)
    {
        if (a[index] != a[i])
            a[++index] = a[i];
    }

    cout << index + 1 << endl;

    return 0;
}

题目二

输入一个已经排好序的数组，同一个元素最多允许出现2次，去除多余的元素，输出新数组的长度。
举栗：输入[1,1,1,2,2,3]，输出5。

分析

使用一个变量记录元素出现的次数。因为是已经排序的数组，使用一个变量即可，若是没排序的数组，需要引入hashmap来记录出现次数。

代码

//Date:2016-04-05
//Author:Sin_Geek
//时间负责度O(n)，空间复杂度O(1)
#include <vector>
#include <iostream>

using namespace std;

int main()
{
    vector<int> nums = {1,1,1,2,2,3};
    if (nums.size() <= 2)
        cout << nums.size();
    int index = 2;
    for (unsigned i = 2; i < nums.size(); i++)
    {
        if (nums[i] != nums[index - 2])
            nums[index++] = nums[i];
    }
    cout << index;

    return 0;
}