подмножество вектора и сортировать его

Я пытаюсь использовать C ++ для простых частей моего пакета R, используя Rcpp пакет. Я новичок в C ++ (но хочу учиться!). Я реализовал несколько простых программ cpp, используя превосходный Rcpp — На самом деле этот пакет побудил меня изучать C ++ …

Во всяком случае, я застрял с простой проблемой, которая, если я могу исправить, очень поможет. у меня есть NumericVector Я хочу подмножество, а затем сортировать. Код ниже сортирует весь вектор (и также будет иметь дело с NA, что мне и нужно).

Мой вопрос, скажем, я хочу извлечь часть этого вектора, отсортировать и сделать его доступным для другой обработки — как я могу это сделать? Например, для вектора длины 10, как извлечь и отсортировать элементы 5:10?

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
RcppExport SEXP rollP(SEXP x) {
NumericVector A(x); // the data
A = sort_unique(A);
return A;
}

который я звоню из R:

sourceCpp( "rollP.cpp")
rollP(10:1)
# [1]  1  2  3  4  5  6  7  8  9 10

10

Решение

Вот 3 варианта:

include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
NumericVector rollP(NumericVector A, int start, int end) {
NumericVector B(end-start+1) ;
std::copy( A.begin() + start-1, A.begin() + end, B.begin() ) ;
return B.sort() ;
}

// [[Rcpp::export]]
NumericVector rollP2(NumericVector A, int start, int end) {
NumericVector B( A.begin() + start-1, A.begin() + end ) ;
return B.sort() ;
}

// [[Rcpp::export]]
NumericVector rollP3(NumericVector A, int start, int end) {
NumericVector B = A[seq(start-1, end-1)] ;
return B.sort() ;
}

start а также end подразумеваются как индексы на основе 1, как если бы вы проходили A[start:end] от R,

12

Другие решения

Вы должны посмотреть на индексирование C ++, итераторы и все остальное. Как минимум, вам нужно изменить свой интерфейс (vector, fromInd, toInd) и выяснить, что вы хотите вернуть.

Одна из интерпретаций вашего вопроса будет копировать подмножество из [fromInd, toInd) в новый вектор, сортируйте его и возвращайте. Все это стандартная плата за C ++ и хороший текст, как отличный (и бесплатный!) C ++ Аннотации будет полезен Он также имеет довольно сильный раздел STL.

4

Ты можешь использовать std::slice на std::valarray. Но если вы хотите использовать std::vector именно тогда вы можете использовать std::copy чтобы извлечь часть вектора, а затем использовать std::sort отсортировать извлеченный фрагмент вектора.

2

Вы можете сделать это довольно легко, используя std::sort реализация, которая получает два итератора:

#include <vector>
#include <cinttypes>
#include <algorithm>

template <typename SeqContainer>
SeqContainer slicesort(SeqContainer const& sq, size_t begin, size_t end) {
auto const b = std::begin(sq)+begin;
auto const e = std::begin(sq)+end;
if (b <= std::end(sq) && e <= std::end(sq)) {
SeqContainer copy(b,e);
std::sort(copy.begin(),copy.end());
return copy;
}
return SeqContainer();
}

Который может быть вызван как

  std::vector<int> v = {3,1,7,3,6,-2,-8,-7,-1,-4,2,3,9};
std::vector<int> v2 = slicesort(v,5,10);
2
По вопросам рекламы [email protected]