leetcode笔记:Remove Element

一.题目描述

这里写图片描述

二.解题技巧

这道题和Remove Duplicates from Sorted Array类似,不同的是这里只要删除值等于给定值的元素即可,可以采用和前面的题相同的做法:可以将原来的数组看作一个栈,设定一个栈顶指针,在对数组进行遍历的时候,判断元素是否等于给定值,如果等于,则直接进行数组的下一个元素,如果不等于,则将该元素放入到栈顶,然后更新指针并处理数组的下一个元素。之所以能这么做,是因为我们在遍历过程中,对于数组元素的访问的下标总是不小于栈顶指针的,因此,可以将原来的数组直接当作一个栈来使用,这样就能做到in-place处理了。

这道题相对比较简单,也没有多少边界条件要考虑,是一道练手的题。这里在进行删除后数组的元素的个数的统计的时候,有一个技巧,即栈顶指针就已经说明了数组中元素的个数了,因此,可以直接返回栈顶指针就可以了(这里的栈顶指针是用数组的下标表示的)。当然,也可以统计等于给定值的元素的个数,然后用原来的数组个数减去该值就可以得到新的数组的值了。

三.示例代码

#include <iostream>

using namespace std;

class Solution
{
public:
    int RemoveElement(int A[], int n, int Elem)
    {
        int num = 0;
        for (int i = 0; i < n; i++)
        {
            if (A[i] != Elem)
                A[num++] = A[i];  // 支持in place运算
        }
        return num;
    }
};

测试代码:

#include "Solution.h"
#include <algorithm>

int main()
{
    int Elem = 5; // 删除数组内数值为5的元素
    int length = 100;  // 数组大小
    int A[100];
    for (int i = 0; i < length; i++)
        A[i] = rand() % 20 - 10;

    sort(A, A + length);  // 便于观察,对输入数组先进行排序

    cout << "输入数组长度为:" << length << endl;
    cout << "输入数组为:" << endl;
    for (int j = 0; j < 100; j++)
        cout << A[j] << " ";
    cout << endl << endl;

    cout << "删除数值为:" << Elem << "的元素" << endl << endl;

    Solution s;
    int result = s.RemoveElement(A, 100, Elem);

    cout << "输入数组长度为:" << result << endl;
    cout << "输出数组为:" << endl;
    for (int k = 0; k < result; k++)
    {
        cout << A[k] << " ";
    }
    cout << endl;

    getchar();
    return 0;
}

一个测试结果:

leetcode笔记:Remove Element_第1张图片

四.体会

这道题实现简单,没有很严格的边界条件,可以寻求别的解决方式。

你可能感兴趣的:(LeetCode,Algorithm,C++,指针,stack)