Split Into Chunks
Python
Medium
2 views
Problem Description
Read n integers and chunk size k. Write function chunks(arr,k) that prints each chunk sum in new line.
Input Format
Line1: n k. Line2: n integers.
Output Format
Lines of chunk sums.
Official Solution
import sys
p=sys.stdin.read().strip().split()
if len(p)<2: sys.exit(0)
n=int(p[0]); k=int(p[1])
a=list(map(int,p[2:2+n]))
def chunk_sums(arr,sz):
out=[]
i=0
while i<len(arr):
s=0
for j in range(i,min(i+sz,len(arr))):
s+=arr[j]
out.append(s)
i+=sz
return out
res=chunk_sums(a,k)
sys.stdout.write('\
'.join(map(str,res)))
Solutions (0)
No solutions submitted yet. Be the first!
No comments yet. Start the discussion!