元素移除

元素移除

给定一个数组和一个数（该数不一定在数组中），从数组里删掉这个数字，返回剩下的数组长度。

如：A[] = {1, 2, 3, 4, 5}，要删除数字 3，那么返回数组长度为 4。
亲爱的小伙伴们，题目是不是很简单呢？

提示：int removeElement(int A[], int n, int elem)

其中，n代表数组长度，elem代表要删掉的元素。

输入格式
第一行输入一个数 n (1 ≤ n ≤ 100)，接下来一行 n 个整数，表示数组 A 的所有元素 A i (0 ≤ Ai ≤100)，输入要删除的元素 elem(0 ≤ elem ≤ 100)。

输出格式
输出一个整数，表示剩余数组长度。

样例输入

2
3 3
3

样例输出

0

Code：

#include<iostream>
using namespace std;

int removeElement(int A[], int n, int elem); //声明函数 

int main()
{
    int n, elem;
    cin >> n; //输入个数 
    int *A = new int[n]; //动态分配数组，当然这里最多只有100个数可以用下面这个方式声明数组
    //int A[101]; 最好设置大一点，防止越界，所以这里去101 
    for(int i = 0; i < n; i ++) {
        cin >> A[i]; //输入数组的数值 
    }
    cin >> elem; //输入要删除的数字 
    cout << removeElement(A, n, elem) << endl; //调用函数，并且打印函数的返回值 

    return 0;
}

int removeElement(int A[], int n, int elem)
{
    int cnt = n; //cnt初始化为n 
    for(int i = 0; i < n; i ++) {
        if(A[i] == elem) { //如果这个是要删除的数字，则cnt需要减少一个 
            cnt --;
        }
    }

    return cnt; //返回cnt，即为剩余的个数 
}