计蒜客 元素移除

元素移除

 

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

如：A[] = {1, 2, 3, 4, 5}, 要删除数字3, 那么返回数组长度为4.

亲爱的小伙伴们，题目是不是很简单呢？

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

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

格式：

输入一个数n，继而输入一个数组A[n],接着输入要删除的元素elem，返回剩余数组长度index.

#include"iostream"
#define MAX 10000
using namespace std;
int main()
{
    int n, a[MAX], elem, ans;
    cin >> n;
    for (int i = 0; i<n; i++)
        scanf("%d", &a[i]);

    cin >> elem;
    ans = n;

    for (int i = 0; i<n; i++)
    {
        if (a[i] == elem)
        {
            ans--;
        }
    }

    cout << ans;

}