元素移除

题目描述

给定一个数组和一个数（该数不一定在数组中），从数组里删掉这个数字，返回剩下的数组长度。 如：$A[] = {1, 2, 3, 4, 5}$，要删除数字 $3$，那么返回数组长度为 $4$。 亲爱的小伙伴们，题目是不是很简单呢？ 提示：int removeElement(int A[], int n, int elem) 其中，n代表数组长度，elem代表要删掉的元素。

输入

第一行输入一个数 $n(1 \leq n \leq 100)$，接下来一行 $n$ 个整数，表示数组 $A$ 的所有元素 $A_i(0 \leq A_i \leq 100)$，接着输入要删除的元素 $elem(0 \leq elem \leq 100)$。

输出

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

样例输入

2
3 3
3

样例输出

0
分析：还是map对每个数字进行计数，删除时将对应数字的second置零即可。

#include <iostream>
#include <string>
#include <cstdio>
#include <cmath>
#include <map>
#include <cstring>
#include <algorithm>
#define range(i,a,b) for(int i=a;i<=b;++i)
#define rerange(i,a,b) for(int i=a;i>=b;--i)
#define LL long long
#define fill(arr,tmp) memset(arr,tmp,sizeof(arr))
using namespace std;
map<int,int>data;
int n,aim;
void init(){
    cin>>n;
    range(i,1,n){
        int tmp;
        cin>>tmp;
        data[tmp]++;
    }
    cin>>aim;
}
void solve(){
    map<int,int>::iterator iter;
    for(iter=data.begin();iter!=data.end();++iter)
        if(iter->first==aim){
            n-=iter->second;
            break;
        }
    cout<<n<<endl;
}
int main(){
    init();
    solve();
    return 0;
}

View Code