计蒜客 元素移除

元素移除

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

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

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

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

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

格式：

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

 1 #include"iostream"
 2 #define MAX 10000
 3 using namespace std;
 4 int main()
 5 {
 6     int n, a[MAX], elem, ans;
 7     cin >> n;
 8     for (int i = 0; i<n; i++)
 9         scanf("%d", &a[i]);
10
11     cin >> elem;
12     ans = n;
13
14     for (int i = 0; i<n; i++)
15     {
16         if (a[i] == elem)
17         {
18             ans--;
19         }
20     }
21
22     cout << ans;
23
24 }