如何使用迭代器在向量中的不同位置插入多个元素?

时间:2019-06-29 23:56:09

标签: c++ algorithm c++11 insert stdvector

我想编写一个函数,将向量[2, 1, 4, 0, 5]更改为

[2, 2, 1, 4, 4, 4, 4, 5, 5, 5, 5, 5]

我可以通过将向量弹出到数组中,然后将元素推回向量来做到这一点。

如何使用插入来做到这一点?我可以修改以下程序吗?最有效的方法是什么?

void timesDuplicates(vector<int>& a) 
{
    int s = a.size(), count = 0;
    for(int i = 0; count < s ; i+=a[i], count++) {
        if(a[i] == 0) continue;
        a.insert(a.begin()+i, a[i], a[i]); 
    }
}

2 个答案:

答案 0 :(得分:2)

  

我该如何使用插入来做到这一点?我可以修改以下程序吗?

关于效率,您的向量每次插入时都可能会进行多次重新分配,因为在提供的代码中,没有存储 std::vector::reserve ,即使通过累加也可以完成要素。就像 @IgorTandetnik 指出的那样,也无法转换传递的向量。

最简单的方法是创建一个新矢量,其中根据传递的矢量中存在的元素数量,仅 std::vector::insert 个元素。

以下是示例代码。 See Live

#include <iostream>
#include <vector>
#include <numeric>  // std::accumulate

std::vector<int> timesDuplicates(const std::vector<int>& vec)
{
    std::vector<int> result;
    // reserve the amount of memory for unwanted reallocations
    result.reserve(std::accumulate(std::cbegin(vec), std::cend(vec), 0));
    // you do not need to check element == 0 here
    // as std::vector::insert(end, 0, 0) will insert nothing
    for (const int element : vec) result.insert(result.end(), element, element);
    // return the result
    return result;
}

int main()
{
    const auto result{ timesDuplicates({ 2, 1, 4, 0, 5 }) };
    for (const int ele : result) std::cout << ele << " ";
    return 0;
}

或者,如果您不相信 NRVO or copy elision 会发生,请将向量result作为参数( ref )传递给函数,保留所需的内存后。

#include <iostream>
#include <vector>
#include <numeric>  // std::accumulate

void timesDuplicates(
    const std::vector<int>& vec,
          std::vector<int>& result)
{
    for (const int element : vec)
        result.insert(result.end(), element, element);
}

int main()
{
    const std::vector<int> vec{ 2, 1, 4, 0, 5 };
    std::vector<int> result;
    result.reserve(std::accumulate(std::cbegin(vec), std::cend(vec), 0));
    timesDuplicates(vec, result);
    for (const int ele : result) std::cout << ele << " ";
    return 0;
}

答案 1 :(得分:1)

尝试递归使用此代码段。由于您正在弹出并推入新的临时向量,push_back就足够了(插入将需要您找到新的插入位置)

void timesDuplicates(vector<int>& vec, int idx = 0)
{
    static vector<int> result;    
    int v = vec[idx];              // get value
    for (int i = 0; i < v; i++)    // multiply value
    {
        result.push_back(v);   // push value to temp vector (result)
    }    
    if (idx == vec.size() - 1) {   // border condition
        vec.swap(result);      // swap result
        return;                
    }    
    timesDuplicates(vec, ++idx);   // increment index   
}

void main()
{
    vector<int> vec = { 2, 1, 4, 0, 5 };
    timesDuplicates(vec);
    for (auto e : vec)
        cout << e << " ";
    cout << "\n";
}