Вступ

Давайте подивимося за цей масив: [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.

Пітон, 40 байт

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

Перевірте це на Ideone .

Excel, 21 байт

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

Відкрийте нову електронну таблицю, покладіть тестові значення у стовпчик А. Введіть формулу в 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 через індекс плюс значення цього element та reduceing додаючи.

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

Функції вищого порядку Erlang для списків не отримують індекс поточного елемента. При цьому використовується словник процесу для встановлення індексу поточного елемента.

Пайк, 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 байт

Рекурсивне рішення, яке не використовує жоден із методів масиву:

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

Діалог APL, 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 зберігає початкові значення, $ враховується для кожного індексу, $ 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/Eumerable

Пояснення коду: _.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 [], який представляє вхід, який слід вирішити.