multinomial.js 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.createMultinomial = void 0;
  6. var _collection = require("../../utils/collection.js");
  7. var _factory = require("../../utils/factory.js");
  8. var name = 'multinomial';
  9. var dependencies = ['typed', 'add', 'divide', 'multiply', 'factorial', 'isInteger', 'isPositive'];
  10. var createMultinomial = /* #__PURE__ */(0, _factory.factory)(name, dependencies, function (_ref) {
  11. var typed = _ref.typed,
  12. add = _ref.add,
  13. divide = _ref.divide,
  14. multiply = _ref.multiply,
  15. factorial = _ref.factorial,
  16. isInteger = _ref.isInteger,
  17. isPositive = _ref.isPositive;
  18. /**
  19. * Multinomial Coefficients compute the number of ways of picking a1, a2, ..., ai unordered outcomes from `n` possibilities.
  20. *
  21. * multinomial takes one array of integers as an argument.
  22. * The following condition must be enforced: every ai <= 0
  23. *
  24. * Syntax:
  25. *
  26. * math.multinomial(a) // a is an array type
  27. *
  28. * Examples:
  29. *
  30. * math.multinomial([1,2,1]) // returns 12
  31. *
  32. * See also:
  33. *
  34. * combinations, factorial
  35. *
  36. * @param {number[] | BigNumber[]} a Integer numbers of objects in the subset
  37. * @return {Number | BigNumber} Multinomial coefficient.
  38. */
  39. return typed(name, {
  40. 'Array | Matrix': function ArrayMatrix(a) {
  41. var sum = 0;
  42. var denom = 1;
  43. (0, _collection.deepForEach)(a, function (ai) {
  44. if (!isInteger(ai) || !isPositive(ai)) {
  45. throw new TypeError('Positive integer value expected in function multinomial');
  46. }
  47. sum = add(sum, ai);
  48. denom = multiply(denom, factorial(ai));
  49. });
  50. return divide(factorial(sum), denom);
  51. }
  52. });
  53. });
  54. exports.createMultinomial = createMultinomial;