【C++】 Vector 的erase用法

Problem Statement

给你一个下标从 0 开始长度为 偶数 的整数数组 nums 。

只要 nums 不是 空数组,你就重复执行以下步骤:

  • 找到 nums 中的最小值,并删除它。
  • 找到 nums 中的最大值,并删除它。
  • 计算删除两数的平均值。

两数 a 和 b 的 平均值 为 (a + b) / 2 。

  • 比方说,2 和 3 的平均值是 (2 + 3) / 2 = 2.5 。

返回上述过程能得到的 不同 平均值的数目。

注意 ,如果最小值或者最大值有重复元素,可以删除任意一个。

input

第一行有1个数,即整数数组的元素个数n
第二行有n个数,即整数数组的元素

output

上述过程能得到的 不同 平均值的数目

先来看代码
#include<iostream>
#include<set>
#include<vector>
using namespace std;
int main(){
    int n;
    cin>>n;
    vector<int> arr;
    set<float> ans;
    int n1=n;
    while(n1--){
        int a=0;
        cin>>a;
        arr.push_back(a);
    }
    while(arr.size()){
        auto iterMax=arr.begin();auto iterMin=arr.begin();
        for(auto iter=arr.begin();iter!=arr.end();iter++){
            if(*iter>*iterMax){
                iterMax=iter;
            }
            if(*iter<*iterMin){
                iterMin=iter;
            }
        }
        for(auto i=arr.begin();i!=arr.end();i++){
            cout<<*i;
        }
        cout<<endl;
        float aver=(*iterMax+*iterMin)/2.0;
        ans.insert(aver);
        cout<<*iterMax<<' '<<*iterMin<<endl;
        arr.erase(iterMax);arr.erase(iterMin);    
        cout<<endl;    
    }
    cout<<ans.size();
}

Sample Input

6
4 1 4 0 3 5

sample input通过了,但是在后面的随机测试中并未通过。经检查,发现测试案例的输出为:

4 1 4 0 3 5 
5 0

4 1 4 3 
4 1

1 3 
3 1

2

可以看到,erase删除的并非4和1,而是4和4;

所以erase(iterator position)在删除vector中的元素后,会将该元素的后面所有元素都往前挪一位。因此,原先的迭代器指向的元素就不是原来那个了,而是它的后一位元素。

所以,只需要将代码改为先删除后面的元素,再删除前面的元素,即可以避免迭代器指向错误的元素。

经过修改,代码成功通过。

#include<iostream>
#include<set>
#include<vector>
using namespace std;
int main(){
    int n;
    cin>>n;
    vector<int> arr;
    set<float> ans;
    int n1=n;
    while(n1--){
        int a=0;
        cin>>a;
        arr.push_back(a);
    }
    while(arr.size()){
        auto iterMax=arr.begin();auto iterMin=arr.begin();
        for(auto iter=arr.begin();iter!=arr.end();iter++){
            if(*iter>*iterMax){
                iterMax=iter;
            }
            if(*iter<*iterMin){
                iterMin=iter;
            }
        }
        float aver=(*iterMax+*iterMin)/2.0;
        ans.insert(aver);
        if(iterMax>iterMin){
            arr.erase(iterMax);arr.erase(iterMin);
        }else{
            arr.erase(iterMin);arr.erase(iterMax);
        }
    }
    cout<<ans.size();
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值