矢量中的移动元素没有按预期工作

moving elements in a vector is not working as expected

本文关键字:工作 元素 移动      更新时间:2023-10-16

我正试图将每个元素的x值移动到向量的开始,以便所有元素的x值都在向量的前面,但它不工作,所以你能告诉我我做错了什么,请?

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
template <typename Container, typename Arg>
void move_x(Container& c, Arg x)
{
    typename Container::iterator it = find(c.begin(), c.end(), x);
    if (it!=c.end()) {
        c.insert(c.begin(), *it);
       remove (it, c.end(), x);
    }
}
int main()
{
    int x=1;
    vector <int> v{1,2,4,6,7,1,3,1,1,8,9};
    move_x(v, x);
    for(auto i:v)
        cout<<v[i];
    return 0;
}

当我运行它时,我得到了这个输出

411613848811

一旦插入到容器中,迭代器就不再有效了

    c.insert(c.begin(), *it); // This invalidates 'it'    
    remove (it, c.end(), x); // oops! trying to use invalid iterator

使用std::rotate提供了更好的选择,它不会使迭代器失效:

template <typename Container, typename Arg>
void move_x(Container& c, Arg x)
{
    typedef typename Container::iterator It;
    It write_it = c.begin(), read_it = c.begin();
    for (;;) {
        It found_it = find(read_it, c.end(), x);
        if (found_it==c.end()) break;
        read_it = found_it;
        ++read_it;
        std::rotate(write_it,found_it,read_it);
        ++write_it;
    }
}

只要你处理的是像int这样简单的数据项,这是一个很好的方法:

template <typename Container, typename Arg>
void move_x(Container& c, Arg x)
{
    typename Container::reverse_iterator it = std::remove(c.rbegin(),c.rend(),x);
    for (;it!=c.rend();++it) {
        *it = x;
    }
}

这是一个固定的实现,你在你的代码:

template <typename Container, typename Arg>
void move_x(Container& c, Arg x)
{
    typename Container::iterator it = find(c.begin(), c.end(), x);
    if (it!=c.end()) {
       c.erase(it);
       c.insert(c.end(), x);
    }
}

您的实现的一个问题是insert在任何地方,但结束可能导致重新分配,无论如何将使插入位置后的任何内容无效。因此,根据定义,由于您在开头插入it,因此将无效。

第二个问题是cout<<v[i];,它实际上应该是cout<<i;

使用反向迭代器并移动所有x的更好实现。这一个擦除,因为它去,并保持一个count,然后count插入完成时。在反向迭代器中使用erase有点棘手:

template <typename Container, typename Arg>
void move_all_x(Container& c, Arg x)
{
    unsigned int count = 0 ;
    for( typename Container::reverse_iterator it = c.rbegin() ; it != c.rend(); )
    {
       if( *it == x )
       {
         c.erase(--(it++).base() ) ;
         ++count ;
       }
       else
       {
          ++it ;
       }
    }
    for( unsigned int i = 0; i < count; ++i )
      {
         c.insert(c.begin(), x) ;
      }
}

您可以使用std::partition,它将把范围内满足给定谓词的所有元素移到范围的开头,并返回一个指向第一个不满足谓词的元素的迭代器。

template <typename Container, typename Arg>
void move_x(Container& c, Arg x)
{
    typename Container::iterator endrange = 
        std::partition(c.begin(), c.end(), [&x](Arg ele){ return ele == x; });
}

在这种情况下,我们没有使用返回值,但我认为它可能会很有用。

输出错误。应该有for (auto i:v) cout << i;而不是v[i]。使用正确的算法也可以看到垃圾

需要一个循环来处理所有匹配(或使用count和insert)。v定义为list<int>:

template <typename Container, typename Arg>
void move_x(Container& c, Arg x)
{
  for (auto it = find(c.begin(), c.end(), x); 
       it != c.end(); 
       it = find(it, c.end(), x)) {
    c.insert(c.begin(), x); 
    it = c.erase(it);
  }
}