子集一个矢量并对其进行排序

Question

我正在考虑使用一些C++来使用我的R包的简单部分 Rcpp包中的.我是C++新手(但很想学习!).我已经使用优秀的实现了一些简单的cpp程序Rcpp- 实际上这个程序包激励我学习C++ ...

无论如何,我遇到了一个简单的问题,如果我可以修复它会帮助很多.我有一个NumericVector我想要子集然后排序.下面的代码对整个向量进行排序(并且还将处理NAs,这是我需要的).

我的问题是,我想要提取这个向量的一部分,排序并让它可用于其他处理 - 我该怎么做？例如,对于长度为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

Answer 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.