Rcpp函数填充不同值的矩阵

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
NumericMatrix fillMatrix(int rows, int cols) {
  NumericMatrix mat(rows, cols); // 创建一个指定行数和列数的矩阵

  for (int i = 0; i < rows; i++) {
    for (int j = 0; j < cols; j++) {
      if (i % 2 == 0) {
        mat(i, j) = i + j; // 偶数行填充 i+j
      } else {
        mat(i, j) = i * j; // 奇数行填充 i*j
      }
    }
  }

  return mat;
}

library(Rcpp)
sourceCpp("path/to/cpp/file.cpp")

mat <- fillMatrix(3, 3)
print(mat)

     [,1] [,2] [,3]
[1,]    0    1    2
[2,]    1    0    1
[3,]    2    1    2