几何分布函数

接口定义

int viRngGeometric (const int method, VSLStreamStatePtr stream, const SizeType n, int *r, const double p);

概率密度函数

参数

依赖

#include "krng.h"

示例

#include <stdio.h>
#include <stdlib.h>
#include "krng.h"
int main()
{
   /* initialize stream with given BRNG type and seed */
    VSLStreamStatePtr stream;
    unsigned seed = 42;
    int errcode = vslNewStream(&stream, VSL_BRNG_MCG59, seed);
    if (errcode != VSL_STATUS_OK) {
        fprintf(stderr, "Failure in newstream\n");
        return 0;
    }

    Sizetype n = 10;
    float *r = (float *)malloc(sizeof(float) * n);
    if (r == NULL) {
        fprintf(stderr, "Failure in malloc\n");
        return 0;
    }

    if (viRngGeometric(VSL_RNG_METHOD_GEOMETRIC_ICDF, stream, n, r, 0.5)) {
        fprintf(stderr, "Failure in viRngGeometric\n");
        goto out;
    }

    /* deinitialize the stream */
    errcode = vslDeleteStream(&stream);
    if (errcode != VSL_STATUS_OK) {
        fprintf(stderr, "Failure in deleting stream\n");
        goto out;
    }

out:
    free(r);
    return 0;
}