整数のシーケンス

前の演習で作成した関数は計算を行い、単一の数値を返していました。ベクトルを返す関数を書くこともできます。

ベクトル型を作成する構文は、ベクトルの型、変数名、そしてベクトルの要素数を丸括弧で指定します。たとえば、10 個の要素を含む数値ベクトル numbers を作成するには、次のように書きます。

NumericVector numbers(10);

この演習はコースの一部です

Rcpp で R コードを最適化する

コースを見る

演習の手順

2つの整数 lo と hi を受け取り、その範囲の数を含む IntegerVector を返す関数 seq_cpp() の定義を完成させてください。
- 返り値の型を IntegerVector に設定します。
- サイズが n の整数ベクトル sequence を作成します。
- for ループの中で、sequence の i 番目の要素を lo に i を足した値に設定します。
- sequence を返します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

#include 
using namespace Rcpp;

// Set the return type to IntegerVector
// [[Rcpp::export]]
___ seq_cpp(int lo, int hi) {
  int n = hi - lo + 1;
    
  // Create a new integer vector, sequence, of size n
  ___;
    
  for(int i = 0; i < n; i++) {
    // Set the ith element of sequence to lo plus i
    ___;
  }
  
  return ___;
}

/*** R
lo <- -2
hi <- 5
seq_cpp(lo, hi)
# Does it give the same answer as R's seq() function?
all.equal(seq_cpp(lo, hi), seq(lo, hi))
*/

コードを編集して実行