Categorical sampling (for Rcpp)

rcate.cpp

#include <RcppArmadillo.h>
using namespace Rcpp;
using namespace arma;
// [[Rcpp::depends(RcppArmadillo)]]

// [[Rcpp::export]]
mat rcate(const int N, const rowvec & p){
  int K = p.n_cols;
  rowvec cump = cumsum(p);
  mat x(N, K);
  x.fill(0);
  for(int n = 0; n<N; n++){
    double U = R::runif(0,1);
    if(U<=cump(0)){
      x(n,0) += 1;
    }else{
      for(int k=1; k<K; k++){
        if(cump[k-1]<U & U<=cump[k]){
          x(n,k) += 1;
        }
      }
    }
  }
  return(x);
}