Ваше завдання - написати програму або функцію, яка виводить n випадкові числа з інтервалу [0,1] з фіксованою сумою s.

Вхідні дані

n, n≥1, кількість випадкових чисел для генерації

s, s>=0, s<=n, сума чисел, які потрібно генерувати

Вихід

Випадкову nмножину чисел з плаваючою комою з усіма елементами з інтервалу [0,1] і сумою всіх елементів дорівнює s, вивести будь-яким зручним однозначним способом. Усі дійсні nпара-параметри повинні бути однаково вірогідними в межах обмежень чисел з плаваючою комою.

Це дорівнює рівномірному відбору проб від перетину точок всередині nкуба-розмірної одиниці та n-1-вимірної гіперплани, яка проходить через (s/n, s/n, …, s/n)і перпендикулярна вектору (1, 1, …, 1)(див. Червону зону на малюнку 1 на трьох прикладах).

Малюнок 1: Площина дійсних виходів з n = 3 і сумами 0,75, 1,75 і 2,75

Приклади

n=1, s=0.8 → [0.8]
n=3, s=3.0 → [1.0, 1.0, 1.0]
n=2, s=0.0 → [0.0, 0.0]
n=4, s=2.0 → [0.2509075946818119, 0.14887693388076845, 0.9449661625992032, 0.6552493088382167]
n=10, s=9.999999999999 → [0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999,0.9999999999999]

Правила

• Ваша програма повинна закінчитися за секунду на вашій машині, принаймні з n≤10будь-якими дійсними s
• Якщо ви цього хочете, ваша програма може бути ексклюзивною у верхньому кінці, тобто s<nчисла виходів з напіввідкритого інтервалу [0,1) (порушуючи другий приклад)
• Якщо ваша мова не підтримує числа з плаваючою комою, ви можете підробити висновок принаймні десятьма десятковими цифрами після десяткової крапки.
• Стандартні лазівки заборонені, а стандартні методи введення / виводу дозволені.
• Це код-гольф , тому найкоротший запис, виміряний в байтах, виграє.

Мова Вольфрама (Mathematica) , 92 90 байт

If[2#2>#,1-#0[#,#-#2],While[Max[v=Differences@Sort@Join[{0,#2},RandomReal[#2,#-1]]]>1];v]&

Спробуйте в Інтернеті!

Код без гольфу:

R[n_, s_] := Module[{v},
  If[s <= n/2,             (* rejection sampling for s <= n/2:                        *)
    While[
      v = Differences[Sort[
            Join[{0},RandomReal[s,n-1],{s}]]];         (* trial randoms that sum to s *)
      Max[v] > 1           (* loop until good solution found                          *)
    ];
    v,                     (* return the good solution                                *)
    1 - R[n, n - s]]]      (* for s > n/2, invert the cube and rejection-sample       *)

Ось рішення, яке працює в 55 байтах, але поки що (версія Mathematica 12) обмежена тим, n=1,2,3що RandomPointвідмовляється малювати точки з гіперпланів більш великих розмірів (у версії 11.3 TIO це також не вдається n=1). Можливо, nу майбутньому це може працювати і вище :

RandomPoint[1&~Array~#~Hyperplane~#2,1,{0,1}&~Array~#]&

Спробуйте в Інтернеті!

Код без гольфу:

R[n_, s_] :=
  RandomPoint[                           (* draw a random point from *)
    Hyperplane[ConstantArray[1, n], s],  (* the hyperplane where the sum of coordinates is s *)
    1,                                   (* draw only one point *)
    ConstantArray[{0, 1}, n]]            (* restrict each coordinate to [0,1] *)

JavaScript (Node.js) , 122 115 байт

N=>W=S=>2*S>N?W(N-S).map(t=>1-t):(t=(Q=s=>n?[r=s-s*Math.random()**(1/--n),...r>1?[++Q]:Q(s-r)]:[])(S,n=N),Q?t:W(S))

Спробуйте в Інтернеті!

Python 2 , 144 128 119 байт

from random import*
def f(n,s):
 r=min(s,1);x=uniform(max(0,r-(r-s/n)*2),r);return n<2and[s]or sample([x]+f(n-1,s-x),n)

Спробуйте в Інтернеті!

• -20 байт, завдяки Кевіну Крейсейну

Java 8, 194 188 196 237 236 байт

n->s->{double r[]=new double[n+1],d[]=new double[n],t;int f=0,i=n,x=2*s>n?1:0;for(r[n]=s=x>0?n-s:s;f<1;){for(f=1;i-->1;)r[i]=Math.random()*s;for(java.util.Arrays.sort(r);i<n;d[i++]=x>0?1-t:t)f=(t=Math.abs(r[i]-r[i+1]))>1?0:f;}return d;}

+49 байт (188 → 196 і 196 → 237), щоб встановити швидкість тестових випадків, що наближаються до 1, а також загалом виправити алгоритм.

Спробуйте в Інтернеті

Пояснення:

Використовується підхід у цій відповіді StackoverFlow всередині циклу, якщо один із елементів ще більший за 1.
Також, якщо 2*s>n, sбуде змінено на n-s, і встановлено прапор, який вказує, що ми повинні використовувати 1-diffзамість diffмасиву результатів (дякую за пораду @soktinpk та @ l4m2 ).

n->s->{              // Method with integer & double parameters and Object return-type
  double r[]=new double[n+1]
                     //  Double-array of random values of size `n+1`
         d[]=new double[n],
                     //  Resulting double-array of size `n`
         t;          //  Temp double
  int f=0,           //  Integer-flag (every item below 1), starting at 0
      i=n,           //  Index-integer, starting at `n`
      x=             //  Integer-flag (average below 0.5), starting at:
        2*s>n?       //   If two times `s` is larger than `n`:
         1           //    Set this flag to 1
        :            //   Else:
         0;          //    Set this flag to 0
  for(r[n]=s=        //  Set both the last item of `r` and `s` to:
       x>0?          //   If the flag `x` is 1:
        n-s          //    Set both to `n-s`
       :             //   Else:
        s;           //    Set both to `s`
      f<1;){         //  Loop as long as flag `f` is still 0
    for(f=1;         //   Reset the flag `f` to 1
        i-->1;)      //   Inner loop `i` in range (n,1] (skipping the first item)
      r[i]=Math.random()*s;
                     //    Set the i'th item in `r` to a random value in the range [0,s)
    for(java.util.Arrays.sort(r);
                     //   Sort the array `r` from lowest to highest
        i<n;         //   Inner loop `i` in the range [1,n)
        ;d[i++]=     //     After every iteration: Set the i'th item in `d` to:
          x>0?       //      If the flag `x` is 1:
           1-t       //       Set it to `1-t`
          :          //      Else:
           t)        //       Set it to `t`
      f=(t=Math.abs( //    Set `t` to the absolute difference of:
            r[i]-r[i+1])) 
                     //     The i'th & (i+1)'th items in `r`
        >1?          //    And if `t` is larger than 1 (out of the [0,1] boundary)
         0           //     Set the flag `f` to 0
        :            //    Else:
         f;}         //     Leave the flag `f` unchanged
  return d;}         //  Return the array `d` as result

JavaScript (Node.js) , 153 байти

(n,s)=>s+s>n?g(n,n-s).map(r=>1-r):g(n,s)
g=(n,s)=>{do(a=[...Array(n-1)].map(_=>Math.random()*(s<1?s:1))).map(r=>t-=r,t=s);while(t<0|t>=1);return[...a,t]}

Спробуйте в Інтернеті!

C ++ 11, 284 267 байт

-17 байт завдяки
випадковій бібліотеці Zacharý Uses C ++, що виводиться на стандартний вихід

#include<iostream>
#include<random>
typedef float z;template<int N>void g(z s){z a[N],d=s/N;int i=N;for(;i;)a[--i]=d;std::uniform_real_distribution<z>u(.0,d<.5?d:1-d);std::default_random_engine e;for(;i<N;){z c=u(e);a[i]+=c;a[++i]-=c;}for(;i;)std::cout<<a[--i]<<' ';}

Щоб зателефонувати, вам просто потрібно зробити це:

g<2>(0.0);

Де параметр шаблону (тут, 2) - N, а фактичний параметр (тут, 0,0) - S

Чисто , 221 201 байт

Чисті, кодові-гольфні чи випадкові числа. Виберіть два.

import StdEnv,Math.Random,System._Unsafe,System.Time
g l n s#k=toReal n
|s/k>0.5=[s/k-e\\e<-g l n(k-s)]
#r=take n l
#r=[e*s/sum r\\e<-r]
|all((>)1.0)r=r=g(tl l)n s

g(genRandReal(toInt(accUnsafe time)))

Спробуйте в Інтернеті!

Часткова функція буквальна :: (Int Real -> [Real]). Дасть нові результати лише раз на секунду.
Вказуйте не менше 10 знаків після коми.

R , 99 байт (з gtoolsпакетом)

f=function(n,s){if(s>n/2)return(1-f(n,n-s))
while(any(T>=1)){T=gtools::rdirichlet(1,rep(1,n))*s}
T}

Спробуйте в Інтернеті!

$\tilde{\mathcal A}=\{w_1,\ldots,w_n: \forall i, 0<w_i<1; \sum w_i=s\}$$w_i$$s$$\mathcal A=\{w_1,\ldots,w_n: \forall i, 0<w_i<\frac1s; \sum w_i=1\}$

$s=1$$Dirichlet(1, 1, \ldots, 1)$ $s\neq 1$$<1/s$$s$

$s>n/2$

С, 132 127 125 118 110 107 байт

-2 байти завдяки @ceilingcat

i;f(s,n,o,d)float*o,s,d;{for(i=n;i;o[--i]=d=s/n);for(;i<n;o[++i%n]-=s)o[i]+=s=(d<.5?d:1-d)*rand()/(1<<31);}

Спробуйте в Інтернеті!

Haskell , 122 217 208 байт

import System.Random
r p=randomR p
(n#s)g|n<1=[]|(x,q)<-r(max 0$s-n+1,min 1 s)g=x:((n-1)#(s-x)$q)
g![]=[]
g!a|(i,q)<-r(0,length a-1)g=a!!i:q![x|(j,x)<-zip[0..]a,i/=j]
n%s=uncurry(!).(n#s<$>).split<$>newStdGen

Спробуйте в Інтернеті!

Іноді відповіді злегка відхиляються через, я припускаю, помилки з плаваючою комою. Якщо це проблема, я можу виправити її вартістю 1 байт. Я також не впевнений, наскільки це рівномірно (я впевнений, що це добре, але я не все так добре в такій справі), тому опишу свій алгоритм.

Основна ідея полягає в тому, щоб генерувати число, xа потім віднімати його sі повторювати, поки у нас nелементи не перетасуються. Я генерую xз верхньою межею або 1, або s(залежно від того, що менше), і нижньої, s-n+1або 0 (залежно від того, що більше). Ця нижня межа існує так, що при наступній ітерації sвсе одно буде менше або дорівнює n(деривація: s-x<=n-1-> s<=n-1+x-> s-(n-1)<=x-> s-n+1<=x).

EDIT: Дякуємо @ michi7x7, що вказав на недолік моєї рівномірності. Я думаю, що я вирішив це з перетасуванням, але повідомте мені, чи є інша проблема

EDIT2: покращена кількість байтів плюс обмеження фіксованого типу

Haskell , 188 байт

import System.Random
import Data.List
n!s|s>n/2=map(1-)<$>n!(n-s)|1>0=(zipWith(-)=<<tail).sort.map(*s).(++[0,1::Double])<$>mapM(\a->randomIO)[2..n]>>= \a->if all(<=1)a then pure a else n!s

Безголівки:

n!s
 |s>n/2       = map (1-) <$> n!(n-s)       --If total more than half the # of numbers, mirror calculation 
 |otherwise   = (zipWith(-)=<<tail)        --Calculate interval lengths between consecutive numbers
              . sort                       --Sort
              . map(*s)                    --Scale
              . (++[0,1::Double])          --Add endpoints
              <$> mapM(\a->randomIO)[2..n] --Calculate n-1 random numbers
              >>= \a->if all(<=1)a then pure a else n!s   --Retry if a number was too large

Спробуйте в Інтернеті!