从c ++向量中的指定索引中提取元素

时间:2014-09-15 14:48:30

标签: c++ visual-c++ vector

我有两个问题,任何帮助都会受到高度关注。

我有一个矩阵A ={ 0 1 0 0 1 1 0 0}。现在我找到了0的索引的位置并保存在向量B={0 2 3 6 7}中。

  1. 如何将A中的矢量B索引的元素提取到新的矢量中,而不会损坏原始矢量A?即我希望得到C= {0 0 0 0 0},这是来自A的数据,由B编制索引。

  2. 如何删除B中索引的A中的元素?

  3. 我尝试了类似这样的问题。 2,但没有成功。

    ///// erasing the elements in particular locations
    
    sort (B.begin(), B.end());
    
    for(int i=A.size() - 1; i >= 0; i--){
    
       A.erase(A.begin() + B[i]);
    }
    

4 个答案:

答案 0 :(得分:2)

<强> Q1:
如果你很高兴制作一个新的载体,PaulMcKenxie的answer就是你想要的:

std::vector<int> C;
for (size_t i = 0; i < B.size(); ++i )
   C.push_back(A[B[i]]);

<强> Q2:
否则,您需要删除未由B编制索引的每个实例。 这是相对复杂的,因为通过以您的方式删除条目,您强制进行可以/(将?)使迭代器/指针数据无效的重新定位。

对此最好的解决方案(简单而有效)可能是在上面创建一个临时向量C,然后交换减少的数据。

void delete_indexes(std::vector<int> &data, std::vector<int> &indexes)
{
    std::vector<int> temp;
    for (size_t i = 0; i < indexes.size(); ++i )
    {
       temp.push_back(data[indexes[i]]);
    }
    data.swap(temp);  //does the work
}

int main()
{
    //do stuff
    delete_indexes(A,B);
}

交换选项很快(只是交换而不是删除和写入),临时向量(包含原始数据)在超出范围时被处理掉。

编辑:
This回答也可能是您正在寻找的,假设您有一个函数来生成您可以应用的B的每个元素(即使它是A[i] == 1(代码编辑为适合) :

for(auto it = A.begin(); it != A.end();)
{
  if (criteria_for_B(*it))
    it = A.erase(it);
  else
    ++it;
}

答案 1 :(得分:1)

  

1.如何在A中引导由矢量B索引的元素,而不会损坏原始矢量A?即我想得到C = {0 0 0 0 0},这是来自A的数据,由B索引。

std::vector<int> C;
for (size_t i = 0; i < B.size(); ++i )
   C.push_back(A[B[i]]);

当然,我们假设B没有超出A向量范围的条目。

答案 2 :(得分:0)

对我来说,我使用擦除功能,但使用计数器来设置迭代器:

#include <iostream>
#include <vector>
using namespace std;

int main(){
  vector<int> A;
  A.push_back(0);
  A.push_back(1);
  A.push_back(0);
  A.push_back(0);
  A.push_back(1);
  A.push_back(1);
  A.push_back(0);
  A.push_back(0);

  vector<int> B;
  B.push_back(0);
  B.push_back(2);
  B.push_back(3);
  B.push_back(6);
  B.push_back(7);

  for(int i=0; i<A.size(); i++){
    cout << A[i] << "-";
  }
  cout << endl;

  vector<int> C = A;
  int ii=0;
  for(int i=0; i<B.size(); i++){
    C.erase(C.begin() -ii + B[i] );
    ii++;
  }
  for(int i=0; i<C.size(); i++){
    cout << C[i] << "-";
  }

}

您可以像我一样使用第三个矢量或直接修改A.

我希望这会对你有所帮助!

答案 3 :(得分:0)

以下是一些std-syle通用实用程序(标准c ++ 98)。

1。在索引处提取元素

/// Extract elements from vector at indices specified by range
template <typename ForwardIt, typename IndicesForwardIt>
inline std::vector<typename std::iterator_traits<ForwardIt>::value_type>
extract_at(
    ForwardIt first,
    IndicesForwardIt indices_first,
    IndicesForwardIt indices_last)
{
    typedef std::vector<typename std::iterator_traits<ForwardIt>::value_type>
        vector_type;
    vector_type extracted;
    extracted.reserve(static_cast<typename vector_type::size_type>(
        std::distance(indices_first, indices_last)));
    for(; indices_first != indices_last; ++indices_first)
        extracted.push_back(*(first + *indices_first));
    return extracted;
}

/// Extract elements from collection specified by collection of indices
template <typename TVector, typename TIndicesVector>
inline TVector extract_at(const TVector& data, const TIndicesVector& indices)
{
    return extract_at(data.begin(), indices.begin(), indices.end());
}

2。 Remove elements at indices

//! Remove one element with given index from the range [first; last)
template <typename ForwardIt>
inline ForwardIt remove_at(ForwardIt first, ForwardIt last, const size_t index)
{
    std::advance(first, index);
    for(ForwardIt it = first + 1; it != last; ++it, ++first)
        *first = *it;
    return first;
}

/*!
 * Remove elements in the range [first; last) with indices from the sorted
 * range [indices_first, indices_last)
 */
template <typename ForwardIt, typename SortedIndicesForwardIt>
inline ForwardIt remove_at(
    ForwardIt first,
    ForwardIt last,
    SortedIndicesForwardIt indices_first,
    SortedIndicesForwardIt indices_last)
{
    typedef typename std::vector<bool> flags;
    // flag elements to keep
    flags is_keep(
        static_cast<flags::size_type>(std::distance(first, last)), true);
    for(; indices_first != indices_last; ++indices_first)
        is_keep[static_cast<flags::size_type>(*indices_first)] = false;
    // move kept elements to beginning
    ForwardIt result = first;
    for(flags::const_iterator it = is_keep.begin(); first != last; ++first, ++it)
        if(*it) // keep element
            *result++ = *first; //in c++11 and later use: *result++ = std::move(*first);
    return result;
}

用法(erase-remove idiom):

std::vector<int> vec{0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
std::vector<int> ii{6, 3, 1};
std::sort(ii.begin(), ii.end());
vec.erase(remove_at(vec.begin(), vec.end(), ii.begin(), ii.end()), vec.end());