我有两个问题,任何帮助都会受到高度关注。
我有一个矩阵A ={ 0 1 0 0 1 1 0 0}
。现在我找到了0的索引的位置并保存在向量B={0 2 3 6 7}
中。
如何将A中的矢量B索引的元素提取到新的矢量中,而不会损坏原始矢量A?即我希望得到C= {0 0 0 0 0}
,这是来自A的数据,由B编制索引。
如何删除B中索引的A中的元素?
我尝试了类似这样的问题。 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]);
}
答案 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)。
/// 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());
}
//! 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());