Если задано целое число n и список натуральных чисел m ₁ , m ₂ , ..., выведите список целых чисел m ₁ ' , m ₂ ' , ... где m _x ' определяется как среднее значение m _{от xn} до m _{x + n} .

При расчете этих средних значений игнорируйте индексы, выходящие за пределы (и соответственно измените то, на что вы делите сумму). n всегда будет ≥ 1, но никогда не будет меньше половины длины m (округлено в меньшую сторону) или больше. Это означает, что минимальная длина m равна 4. Элементы в m будут целыми положительными числами, но выходные данные должны быть с точностью не менее 3 десятичных знаков.

Элементами ввода / вывода, которые являются списками, могут быть строки с пробелами / запятыми или массивы / списки / и т. Д. Для ввода, если ваше решение является функцией, вы можете дополнительно принять первый аргумент n и дополнительные аргументы как m _x (это также относится и к аргументам командной строки).

Вот визуальное представление n=1:

1 4 5 7 10
__/ | | |
L avg(1,4) = 2.5
    | | |
\___/ | |
  L avg(1,4,5) = 3.333
      | |
  \___/ |
    L avg(4,5,7) = 5.333
        |
    \___/
      L avg(5,7,10) = 7.333

      \___
        L avg(7,10) = 8.5

Final output: 2.5 3.333 5.333 7.333 8.5

Поскольку это код-гольф , выигрывает самый короткий код в байтах.

Тестовые случаи ( они были сделаны вручную; пожалуйста, сообщите мне о любых ошибках ):

In                             Out
----------------------------------------------------------------------
n=1, m=12 6 3 9                9 7 6 6
n=1, m=1 4 5 7 10              2.5 3.333 5.333 7.333 8.5
n=1, m=1 3 3 7 4 2 4 2         2 2.333 4.333 4.666 4.333 3.333 2.666 3
n=2, m=1 3 5 9 10 14 15 16 23  3 4.5 5.6 8.2 10.6 12.8 15.6 17 18
n=3, m=1 1 1 1 1 1 1 1         1 1 1 1 1 1 1 1
n=3, m=1 2 3 4 5 6 7 8         2.5 3 3.5 4 5 5.5 6 6.5

Pyth, 20 байтов

m.O:vzeS,0-dQh+dQUvz

Тестирование

Довольно просто, просто вырежьте соответствующий раздел из списка, затем усредните.

MATL , 30 28 26 24 байта

2*1+:g2X53$X+1Mbgbb3$X+/

Проверено на Matlab и на Октаве. Использует текущую версию (9.1.0) языка / компилятора.

Входные данные: сначала число, управляющее длиной окна, затем массив с форматом [1 4 5 7 10].

РЕДАКТИРОВАТЬ (20 мая 2016 г.): попробуйте онлайн! Код в ссылке X+заменен Y+на соответствующий языковой версии 18.0.0.

пример

>> matl
 > 2*1+:g2X53$X+1Mbgbb3$X+/
 >
> 1
> [1 4 5 7 10]
2.5 3.333333333333333 5.333333333333333 7.333333333333333               8.5

>> matl
 > 2*1+:g2X53$X+1Mbgbb3$X+/
 >
> 2
> [1 3 5 9 10 14 15 16 23]
3               4.5               5.6 8.199999999999999              10.6               2.8              15.6        17                18

объяснение

Эквивалентный код Matlab будет

n = 1; %// first input: number controlling window length
x = [1 4 5 7 10]; %// second input: array
result = conv(x,ones(1,2*n+1),'same')./conv(ones(size(x)),ones(1,2*n+1),'same');

В коде MATL используются недавно добавленные функции неявного ввода и автоматического буфера обмена с функцией ввода:

2*1+          % get implicit input "n". Multipliy by 2 and add 1
:g            % create a vector of 2*n+1 "true" values (will act as "one" values)
2X5           % 'same' string literal
3$X+          % get implicit input "x" (array). Convolution using three inputs
1M            % push all three inputs from last function
bgbb          % convert first input to "true" values. Will act as "one" values
3$X+          % convolution using three inputs
/             % divide element-wise. Implicitly print

CJam, 31 30 байт

ri_S*l~1$++\2*)ew{S-_:+\,d/}%`

Формат ввода есть n [m1 m2 ... mx].

Запустите все тестовые случаи. (Автоматически преобразует набор тестов в требуемый формат ввода.)

Это работает путем добавления и добавления nпробелов, затем взятия всех подстрок длины 2n+1и повторного удаления пробелов перед вычислением их средних значений .

Юлия, 57 байт

f(n,m)=[mean(m[max(1,i-n):min(end,i+1)])for i=1:endof(m)]

Это функция, которая принимает два целых числа и возвращает массив чисел с плавающей точкой.

Подход здесь очень прост. Мы создаем новый массив, беря среднее значение секций входного массива, обрезая их спереди и сзади.

Haskell, 97 95 байт

import Data.List
n#x|t<-2*n+1=[sum a/sum(1<$a)|a<-take t<$>take t(inits x)++tails x,length a>n]

Пример использования: 2 # [1,3,5,9,10,14,15,16,23]-> [3.0,4.5,5.6,8.2,10.6,12.8,15.6,17.0,18.0].

Как это устроено:

t<-2*n+1                      -- assign t to the maximum number of elements of a
                              -- of sublist
     take t(inits x)          -- build the sublists from t elements of the inits
                ++tails x     -- of the input and the tails of the input,
                              -- e.g. x=[1,2,3,4], t=3:
                              -- [[],[1],[1,2]] ++ [[1,2,3,4],[2,3,4],[3,4],[4],[]]
  a<-take t<$>                -- take at most t elements from every sublist
                ,length a>n   -- keep those with a minimum length of n+1
sum a/sum(1<$a)               -- calculate average, "sum(1<$a)" is the length of a

Pyth, 22 байта

.OMsM.:++J*]YKE]MQJhyK

Объяснение:

.OM sM .: +               Means of flattens of sublists of length 2K+1 of
            + J *         
                  ] Y     J is E copies of [].
                  K E     Save E to a variable to use it later.
               ]MQ        the input
            J             Put J on both sides of ]MQ.
          h y K           2K+1

Попробуй это здесь .

JavaScript (ES6), 104

Текущий общий / текущий размер выборки. В Javascript чтение значения за пределами массива дает неопределенное значение, которое можно преобразовать в 0, используя ~~

(l,n,p=0)=>l.map((v,i)=>(p+=~l[i-n-1]-~l[i+n])/(n+1+Math.min(n,l.length-1-i,i)),l.map((v,i)=>p+=i<n&&v))

Ungolfed

(l, n) => {
    p = 0;
    for (i = 0; i < n; i++) p += v;
    r = [];
    for (i = 0; i < l.length; i++) {
        p += (l[i + n] || 0) - (i > n ? l[i - n - 1] : 0);
        r.push(p / Math.min(n, l.length - i - 1, i);
    }
    return r;
}

Тестовое задание

f=(n,l,p=0)=>l.map((v,i)=>(p+=~l[i-n-1]-~l[i+n])/(n+1+Math.min(n,l.length-1-i,i)),l.map((v,i)=>p+=i<n&&v))

console.log=x=>O.textContent+=x+'\n';


;[
  [1,[12,6,3,9],[9,7,6,6]]
, [1,[1,4,5,7,10],[2.5,3.333,5.333,7.333,8.5]]
, [1,[1,3,3,7,4,2,4,2],[2,2.333,4.333,4.667,4.333,3.333,2.667,3]]
, [2,[1,3,5,9,10,14,15,16,23],[3,4.5,5.6,8.2,10.6,12.8,15.6,17,18]]
, [3,[1,1,1,1,1,1,1,1],[1,1,1,1,1,1,1,1]]
, [3,[1,2,3,4,5,6,7,8],[2.5,3,3.5,4,5,5.5,6,6.5]]
].forEach(t=>{
  var n=t[0],l=t[1],x=t[2],r=f(n,l)
  // verify (limited to 3 decimals)
  var ok = r.every((v,i)=>v.toFixed(3)==x[i].toFixed(3))
  
  console.log((ok?'OK   ':'Fail ')+n+' '+l+' -> '+r+' ('+x+')')
})

<pre id=O></pre>

JavaScript (ES6), 82 байта

код:

F=(n,a)=>a.map((e,i)=>(s=a.slice(i<n?0:i-n,i+n+1))&&s.reduce((l,c)=>l+c)/s.length)

тестовое задание:

F=(n,a)=>a.map((e,i)=>(s=a.slice(i<n?0:i-n,i+n+1))&&s.reduce((l,c)=>l+c)/s.length)

document.write('<pre>' +
  [
    [1, [12, 6, 3, 9], [9, 7, 6, 6] ],
    [1, [1, 4, 5, 7, 10], [2.5, 3.333, 5.333, 7.333, 8.5] ],
    [1, [1, 3, 3, 7, 4, 2, 4, 2], [2, 2.333, 4.333, 4.667, 4.333, 3.333, 2.667, 3] ],
    [2, [1, 3, 5, 9, 10, 14, 15, 16, 23], [3, 4.5, 5.6, 8.2, 10.6, 12.8, 15.6, 17, 18] ],
    [3, [1, 1, 1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1, 1, 1] ],
    [3, [1, 2, 3, 4, 5, 6, 7, 8], [2.5, 3, 3.5, 4, 5, 5.5, 6, 6.5] ]
  ].map(t => {
    var [n, m, e] = t;
    var r = F(n, m);
    // verify to precision of 3 decimals
    var test = r.every((v, i) => v.toPrecision(3) === e[i].toPrecision(3));

    return 'F(' + n + ', [' + m + '])\t' + (test ? 'Pass' : 'Fail') +
      '\n\t{' + r + '} ' + (test ? '=' : '≠') + ' {' + e + '}';
  }).join('\n\n') +
  '</pre>');