Введение

Давайте понаблюдаем за этот массив: [3, 2, 4, 1, 1, 5, 1, 2].

Каждый элемент отображает длину подстроки, которую необходимо суммировать. Давайте посмотрим на первый элемент вышеуказанного массива:

[3, 2, 4, 1, 1, 5, 1, 2]
 ^

Элемент в первом индексе равен 3 , поэтому теперь мы берем подстроку длины три с тем же индексом, что и в начальной позиции:

[3, 2, 4]

При суммировании это приводит к 9 , поэтому первый элемент из набора суммы подстрок равен 9.

Мы делаем это для всех элементов в массиве:

3 -> [3, 2, 4]
2 -> [2, 4]
4 -> [4, 1, 1, 5]
1 -> [1]
1 -> [1]
5 -> [5, 1, 2]
1 -> [1]
2 -> [2]

Вы можете видеть, что число 5 немного странно. Это число превышает длину массива:

[3, 2, 4, 1, 1, 5, 1, 2]
                ^  ^  ^  ^  ^

Мы будем игнорировать все, что превышает массив, поэтому мы просто используем [5, 1, 2].

Последний шаг - подвести итог:

[3, 2, 4]     -> 9
[2, 4]        -> 6
[4, 1, 1, 5]  -> 11
[1]           -> 1
[1]           -> 1
[5, 1, 2]     -> 8
[1]           -> 1
[2]           -> 2

И это массив, который должен быть выведен:

[9, 6, 11, 1, 1, 8, 1, 2]

Задание

Если дан непустой массив с положительными (ненулевыми) целыми числами, выведите набор суммы подстрок . Это код-гольф , поэтому выигрывает представление с наименьшим количеством байтов!

Контрольные примеры

[1, 2, 3, 4, 5] -> [1, 5, 12, 9, 5]
[3, 3, 3, 3, 3, 3, 3, 3] -> [9, 9, 9, 9, 9, 9, 6, 3]
[5, 1, 2, 4, 1] -> [13, 1, 6, 5, 1]
[1] -> [1]

Желе , 6 байт

ṫJḣ"ḅ1

Попробуйте онлайн! или проверьте все контрольные примеры .

Как это работает

ṫJḣ"ḅ1  Main link. Argument: A (array)

 J      Index; yield the 1-based indices of A.
ṫ       Tail; map k to the postfix of A that begins with the k-th element.
  ḣ"    Vectorized head; for each k in A, truncate the corr. postfix to length k.
    ḅ1  Convert the resulting slices from base 1 to integer.

Python, 40 байт

f=lambda x:x and[sum(x[:x[0]])]+f(x[1:])

Проверьте это на Ideone .

Excel, 21 байт

=SUM(OFFSET(A1,,,A1))

Откройте новую электронную таблицу, поместите тестовые значения в столбец A. Введите формулу в B1 и дважды щелкните маркер ячейки, чтобы перейти к диапазону.

Python 3, 47 байт

lambda X:[sum(X[i:i+k])for i,k in enumerate(X)]

Довольно простая реализация. Поведение Python по умолчанию для срезов, которые выходят за пределы списка, было очень удобным здесь.

Haskell, 34 , 33 байта

f l@(x:y)=sum(take x l):f y
f x=x

Один байт, сохраненный Ними.

JavaScript ES6, 50 байт

a=>a.map((e,i)=>a.slice(i,i+e).reduce((a,b)=>a+b))

Довольно понятно. Он распространяется mapна каждый элемент массива, получая sliceиз этого index индекс, плюс eзначение этого элемента и reduceдобавляя.

f=
  a=>a.map((e,i)=>a.slice(i,i+e).reduce((a,b)=>a+b))

;[
  [3, 2, 4, 1, 1, 5, 1, 2],
  [1, 2, 3, 4, 5],
  [3, 3, 3, 3, 3, 3, 3, 3,],
  [5, 1, 2, 4, 1],
  [1]
].forEach(function(test){
  document.getElementById('p').textContent += test + ' => ' + f(test) + '\n';
});

<pre id="p"></pre>

J 11 байт

+/@{."_1]\.

использование

   f =: +/@{."_1]\.
   f 3 2 4 1 1 5 1 2
9 6 11 1 1 8 1 2
   f 1 2 3 4 5
1 5 12 9 5

объяснение

+/@{."_1]\.  Input: A
        ]\.  Get each suffix of A from longest to shortest
   {."_1     For each value in A, take that many values from its corresponding suffix
+/@          Sum that group of values taken from that suffix
             Return the sums

JavaScript (ES6), 45

reduce избили снова!

a=>a.map((v,i)=>eval(a.slice(i,v+i).join`+`))

F=
a=>a.map((v,i)=>eval(a.slice(i,v+i).join`+`))

;[[3, 2, 4, 1, 1, 5, 1, 2],
  [1, 2, 3, 4, 5],
  [3, 3, 3, 3, 3, 3, 3, 3,],
  [5, 1, 2, 4, 1],
  [1]].forEach(t=>console.log(t+' -> '+F(t)))

Сетчатка , 38 байт

Количество байтов предполагает кодировку ISO 8859-1.

\d+
$*
M!&`\b1(1)*(?<-1>,1+)*
M%`1
¶
,

Вход и выход - разделенные запятыми списки.

Попробуйте онлайн! (Первая строка включает набор тестов, разделенных переводом строки.)

Mathematica 60 55 байт

Tr@Take[#,UpTo@#&@@#]&/@Drop[#,t-1]~Table~{t,Length@#}&

например

f = %; f /@ {{1, 2, 3, 4, 5}, {3, 3, 3, 3, 3, 3, 3, 3}, {5, 1, 2, 4, 1}, {1}}

(*    {{1, 5, 12, 9, 5}, {9, 9, 9, 9, 9, 9, 6, 3}, {13, 1, 6, 5, 1}, {1}}    *)

Спасибо @MartinEnder за то, что сбрил 5 байтов :)

05AB1E, 11 8 байт

[D¬£Oˆ¦Ž

объяснение

[         # infinite loop
 D        # duplicate current list
  ¬       # get head of list
   £      # get that many elements from list
    O     # sum
     ˆ    # add to global array
      ¦   # remove first element of list
       Ž  # break if stack is empty
          # implicitly push and print global array

Попробуйте онлайн

Pyth , 8 байт

.es:Qk+b

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

Перевод ответа Эля на Python .

Эрланг, 69 байт

f(A)->put(1,1),L=lists,[L:sum(L:sublist(A,put(1,get(1)+1),X))||X<-A].

Функции высшего порядка Эрланга для списков не получают индекс текущего элемента. Это использует словарь процесса, чтобы установить индекс текущего элемента.

Пайк, 12 7 байт

FKo>i<s

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

        - o = 0
F       - for i in input:
  o     -    o+=1
   >    -    input[o:]
    i<  -   ^[:i]
      s -  sum(^)

VBA, 160 байт

Function e(g())
Dim h()
k=LBound(g)
l=UBound(g)
ReDim h(k To l)
On Error Resume Next
For i=k To l
For j=i To i+g(i)-1
h(i)=h(i)+g(j)
Next
Next
e=h
End Function

Pyth, 6 байт

ms<~tQ

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

Это отличное решение от любых других. Он зацикливается на вводе, разделяя суммирование начальных значений, затем удаляя первый элемент сохраненного ввода, и повторяется.

Объяснение:

ms<~tQ
ms<~tQdQ    Implicit variable introduction
            Implicit: Q = eval(input())
m      Q    Map d over the input, Q
  <  Qd     Take the first d elements of Q
 s          Sum them
   ~tQ      Afterwards, set Q to the tail of Q, removing the first element.

Юлия, 39 байт

!x=x!=[]?[take(x,x[])|>sum;!x[2:end]]:x

Попробуйте онлайн!

F #, 84 82 байта

let f(A:int[])=[for i in 0..A.Length-1->Seq.skip i A|>Seq.truncate A.[i]|>Seq.sum]

JavaScript (ES6) - 79 байт

Рекурсивное решение, которое не использует ни один из методов Array:

f=([a,...t],n)=>a&&n?a+f(t,n-1):0;g=([a,...t],r=[])=>a?g(t,[...r,a+f(t,a-1)]):r

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

f=([a,...t],n)=>a&&n?a+f(t,n-1):0;
g=([a,...t],r=[])=>a?g(t,[...r,a+f(t,a-1)]):r;

[
  [3, 2, 4, 1, 1, 5, 1, 2],
  [1, 2, 3, 4, 5],
  [3, 3, 3, 3, 3, 3, 3, 3,],
  [5, 1, 2, 4, 1],
  [1]
].forEach(a=>console.log(''+g(a)));

C #, 89 байт

int[]s(List<int>a)=>a.Select((n,i)=>a.GetRange(i,Math.Min(n,a.Count-i)).Sum()).ToArray();

довольно прямо

идеи улучшения приветствуются

Брахилог , 27 байт

.v|h~l(A:Tc?;A?)b:0&~b.h~+A

Попробуйте онлайн! или проверьте все контрольные примеры .

объяснение

  .v           Input = Output = []
|            Or
  h~l          A is a list, its length is the value of the first element of the Input
  (
    A:Tc?        The concatenation of A with another list T results in the Input
  ;            Or
    A?           A = Input
  )
  b:0&         Call recursively on Input minus the first element
  ~b.          Output is the output of that call with an extra element at the beginning
  h~+A         That extra element is the sum of the elements of A

Дьялог АПЛ, 15 байт

{+/¨⍵↑∘⌽¨⌽,\⌽⍵}

или

{⌽+/¨(-↑¨,\)⌽⍵}

PHP программа, 72 байта

<?foreach($a=$_GET[a]as$i=>$v)echo array_sum(array_slice($a,$i,$v)),"
";

позвонить с php-cgi -f <filename> 'a[]=3&a[]=2&a[]=4...

+11 как функция:

function f($a){foreach($a as$i=>$v)$r[]=array_sum(array_slice($a,$i,$v));return$r;}

+9 без встроенных функций:

function p($a){foreach($c=$r=$a as$i=>$v)for($k=$i;$k--;)if(--$a[$k]>0)$r[$k]+=$v;return$r;}

($ c сохраняет исходные значения, $ a ведет обратный отсчет для каждого индекса, $ r получает суммы)

-3 как программа:

<?foreach($a=$r=$c=$_GET[a]as$i=>$v)for($k=$i;$k--;)if(--$c[$k]>0)$r[$k]+=$v;print_r($r);

q (37 байт)

{sum each(til[count x],'x)sublist\:x}

Пример:

q){sum each(til[count x],'x)sublist\:x}3 2 4 1 1 5 1 2
9 6 11 1 1 8 1 2

Матрицы , 25 байтов

Да, наконец, вызов, для которого мне не нужны новые функции!

md{z:l-g:c;+c;q:c;};:1:l;

Бежать с: python matricks.py substring.txt [[<input>]] 0

Объяснение:

m                  :1:l;   #loop over entire input
                           #set each value to...
 d{               }        #the sum of...
   z:l-g:c:+c;q:c;         #the input cropped to
                           #the length of the value in the cell

Javascript (с использованием внешней библиотеки) (66 байт)

n=>_.From(n).Select((v,i)=>_.From(n).Slice(i,i+v).Sum()).ToArray()

Ссылка на lib: https://github.com/mvegh1/Enumerable

Объяснение кода: _.From загружает входной массив в библиотеку, которая в основном является LINQ для js. Затем каждый элемент в массиве отображается в соответствии со следующим предикатом: взять входные данные, вырезать их из индекса текущего элемента и взять этот индекс плюс значение текущего элемента. Затем подведите итог этой подпоследовательности. Преобразовать результат в собственный массив JS и вернуть его

Clojure, 63 байта

(defn f[[b & r]](concat[(apply + b(take(dec b)r))](if r(f r))))

Использует сопоставление с образцом, чтобы разложить входной аргумент на первый и остальные аргументы.

MATL , 17 14 13 байт

fGy+!-R0<G*!s

объяснение

Попробуйте онлайн! Или проверьте все контрольные примеры (код изменен для обработки нескольких входных данных).

f     % Take input implicitly. Indices of nonzero elements: this gives [1 2 ... n]
      % where n is input size
G     % Push input again
y     % Push a copy of [1 2 ... n]
+     % Add. Gives [a+1 b+2...] where [a b...] is the input
!     % Transpose into a column vector
-     % Subtraction with broadcast. Gives 2D array
R     % Keep upper triangular part, making the rest of entries 0
0<    % True for negative entries. Each row corresponds to a substring sum.
      % For each row, this gives true for the entries of the input that make up
      % that substring sum. Each row is thus a mask to select entries of the input
G     % Push input again
*     % Multiply with broadcast. This multiplies the input times each row
!s    % Sum of each row. Implicitly display

C #, 94 байта

Console.Write(String.Join(",",a.Select((v,i)=>a.Skip(i).Take(v).Sum().ToString()).ToArray()));

Где a - это int [], представляющий входные данные для решения.