c++ - 子集 vector 并对其进行排序

Question

我正在研究使用 Rcpp 包将一些 C++ 用于我的 R 包的简单部分。我是 C++ 新手(但热衷于学习!)。我已经使用优秀的 Rcpp 实现了一些简单的 cpp 程序 - 事实上，这个包激发了我学习 C++...

无论如何，我遇到了一个简单的问题，如果我能解决这个问题将会有很大帮助。我有一个 NumericVector 我想要子集然后排序。下面的代码对整个 vector 进行排序(并且还会处理 NA，这正是我所需要的)。

我的问题是，假设我想提取此 vector 的一部分，对其进行排序并使其可用于其他处理 - 我该怎么做？例如，对于一个长度为10的 vector ，如何提取和排序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。