Згладжуйте лінійний графік


13

За умови введення цілого числа n та списку натуральних чисел m 1 , m 2 , ..., виведіть список цілих чисел m 1 ' , m 2 ' , ... де 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

Щоб було зрозуміло, це те саме, що попередньо додавати нулі та додавати нулі, а потім обчислювати засоби кожного суміжного подання довжиною 3?
El'endia Starman

@ El'endiaStarman. Ні. avg(0,1,2)Це не те, що avg(1,2). Для "крайових випадків" (га) не слід усереднювати стільки елементів списку вводу.
Дверна ручка

Ага, правда, я зараз бачу.
El'endia Starman


Це було б кращим викликом без обмеження, що m_iпозитивні.
Пітер Тейлор

Відповіді:


1

Pyth, 20 байт

m.O:vzeS,0-dQh+dQUvz

Тестовий набір

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


@ThomasKwa Я спробував це, але, схоже, вам доведеться зберігати Eза межами карти, оскільки воно буде намагатися читати нове значення, якщо воно знаходиться всередині. Це змушує зайняти однакову кількість байтів.
FryAmTheEggman

3

MATL , 30 28 26 24 байт

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

Випробували на Матлабі та на Октаві. Використовується поточна версія (9.1.0) мови / компілятора.

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

EDIT (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

2

CJam, 31 30 байт

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

Формат вводу є n [m1 m2 ... mx].

Запустити всі тестові справи. (Автоматично перетворює тестовий набір у потрібний формат введення.)

Це працює за допомогою попереднього та додавання nпробілів, а потім взяття всіх підрядків довжини 2n+1та видалення пробілів ще раз перед обчисленням їхніх засобів.


1

Юлія, 57 байт

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

Це функція, яка приймає два цілих числа і повертає масив плавців.

Підхід тут дуже простий. Ми будуємо новий масив, беручи середнє значення розділів вхідного масиву, обрізаючи передню і задню частини.


0

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

0

Піт, 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

Спробуйте тут .


0

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))

Безумовно

(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>


0

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>');

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.