sample_single_discrete.c

来自「贝叶斯网络matlab源程序,可用于分类,欢迎大家下载测试」· C语言 代码 · 共 23 行

#include "mex.h"

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray
		 *prhs[]) 
{
  double *y, *pr, *dist;
  int k, distSize;
  double r, cumSum;
  
  plhs[0] = mxCreateDoubleMatrix(1,1, mxREAL);
  y = mxGetPr (plhs[0]);

  pr = mxGetPr (prhs[0]);
  r = pr[0];

  dist = mxGetPr (prhs[1]);
  distSize = mxGetNumberOfElements (prhs[1]);

  for (k = 0, cumSum = 0; (k < distSize) && (r >= cumSum); cumSum += dist[k], k++);

  y[0] = k;
}