10

我正在考虑使用一些 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
4

4 回答 4

12

这里有 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.

于 2012-11-23T07:39:07.300 回答
4

您需要研究 C++ 索引、迭代器和全部内容。至少，您需要更改您的界面（vector、fromInd、toInd）并找出您想要返回的内容。

对您的问题的一种解释是将子集复制[fromInd, toInd)到一个新向量中，对其进行排序并返回它。所有这些都是标准的 C++ 票价，以及像优秀的（而且免费的！！）C++ 注释这样的好文本将会有所帮助。它也有一个非常强大的 STL 部分。

于 2012-11-22T20:33:50.823 回答
3

您可以std::slice在std::valarray. 但是如果你想std::vector具体使用，那么你可以使用std::copy提取向量的一部分，然后使用std::sort对提取的向量切片进行排序。

于 2012-11-22T20:23:39.307 回答
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);
于 2012-11-23T08:00:39.443 回答