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

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

Во всяком случае, я застрял в простой проблеме, которая, если я могу исправить, поможет многим. У меня есть 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

Ответы

Ответ 1

Вот 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.

Ответ 2

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

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

Ответ 3

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

Ответ 4

Вы можете сделать это довольно легко, используя 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);