Suavizar um gráfico de linhas

13

Dada a entrada de um número inteiro n e uma lista de números inteiros positivos m 1 , m 2 , ..., imprima uma lista de números inteiros m 1 ' , m 2 ' , ... onde m x ' é definido como a média de m xn a m x + n .

Ao calcular essas médias, ignore os índices que estão fora dos limites (e ajuste o que você está dividindo a soma de acordo). n sempre será ≥ 1, mas nunca a metade do comprimento de m (arredondado para baixo) ou mais. Isso significa que o comprimento mínimo de m é 4. Os elementos em m serão números inteiros positivos, mas a saída deve ser precisa com pelo menos três casas decimais.

Os elementos de entrada / saída que são listas podem ser cadeias separadas por espaço em branco / vírgula ou matrizes / listas / etc. Para entrada, se sua solução for uma função, você poderá adicionalmente usar um primeiro argumento de n e argumentos adicionais como m x (isso também se aplica a argumentos de linha de comando).

Aqui está uma representação visual de 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

Como esse é o , o código mais curto em bytes vence.

Casos de teste ( estes foram feitos manualmente; notifique-me de quaisquer erros ):

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
Maçaneta da porta
fonte
Para ser claro, isso é o mesmo que acrescentar e acrescentar zeros e depois calcular as médias de cada subsequência contígua de comprimento 3?
El'endia Starman
@ El'endiaStarman No. avg(0,1,2)não é o mesmo que avg(1,2). Para os "casos extremos" (ha), você não deve calcular a média de tantos elementos da lista de entrada.
Maçaneta
Ah, certo, eu vejo agora.
El'endia Starman 11/01
Este teria sido um desafio melhor sem a restrição de que eles m_isão positivos.
Peter Taylor

Respostas:

1

Pitão, 20 bytes

m.O:vzeS,0-dQh+dQUvz

Suíte de teste

Bem simples, basta cortar a seção apropriada da lista e depois a média.

isaacg
fonte
@ThomasKwa Eu tentei isso, mas parece que você precisa armazenar Efora do mapa, porque ele continuará tentando ler um novo valor se estiver dentro. Isso leva o mesmo número de bytes.
FryAmTheEggman
3

MATL , 30 28 26 24 bytes

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

Testado no Matlab e no Octave. Usa a versão atual (9.1.0) do idioma / compilador.

A entrada é: primeiro o número que controla o comprimento da janela, depois a matriz com o formato [1 4 5 7 10].

EDIT (20 de maio de 2016): Experimente online! O código no link foi X+substituído por Y+para estar em conformidade com a versão 18.0.0 do idioma.

Exemplo

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

Explicação

O código Matlab equivalente seria

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

O código MATL utiliza os recursos recentemente adicionados da entrada implícita e da área de transferência automática de entrada de função:

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
Luis Mendo
fonte
2

CJam, 31 30 bytes

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

O formato de entrada é n [m1 m2 ... mx].

Execute todos os casos de teste. (Converte automaticamente o conjunto de testes no formato de entrada necessário.)

Isso funciona pré e anexando nespaços, pegando todas as substrings de comprimento 2n+1e removendo os espaços novamente antes de calcular seus meios.

Martin Ender
fonte
1

Julia, 57 bytes

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

Esta é uma função que aceita dois números inteiros e retorna uma matriz de flutuadores.

A abordagem aqui é muito direta. Construímos uma nova matriz tomando a média das seções da matriz de entrada, truncando na frente e atrás.

Alex A.
fonte
0

Haskell, 97 95 bytes

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]

Exemplo de uso: 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].

Como funciona:

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
nimi
fonte
0

Pitão, 22 bytes

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

Explicação:

.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

Experimente aqui .

lirtosiast
fonte
0

JavaScript (ES6), 104

Tamanho total / em execução da amostra em execução. Em Javascript, a leitura de um valor fora dos limites de uma matriz fornece indefinidos, que podem ser convertidos para 0 usando ~~

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

Teste

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>

edc65
fonte
0

JavaScript (ES6), 82 bytes

código:

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)

teste:

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

core1024
fonte