Video Posts (Codeforces 1070K)
Tags: Implement, implementation

Polycarp took $$$n$$$ videos, the duration of the $$$i$$$-th video is $$$a_i$$$ seconds. The videos are listed in the chronological order, i.e. the $$$1$$$-st video is the earliest, the $$$2$$$-nd video is the next, ..., the $$$n$$$-th video is the last.Now Polycarp wants to publish exactly $$$k$$$ ($$$1 \le k \le n$$$) posts in Instabram. Each video should be a part of a single post. The posts should preserve the chronological order, it means that the first post should contain one or more of the earliest videos, the second post should contain a block (one or more videos) going next and so on. In other words, if the number of videos in the $$$j$$$-th post is $$$s_j$$$ then:  $$$s_1+s_2+\dots+s_k=n$$$ ($$$s_i>0$$$),  the first post contains the videos: $$$1, 2, \dots, s_1$$$;  the second post contains the videos: $$$s_1+1, s_1+2, \dots, s_1+s_2$$$;  the third post contains the videos: $$$s_1+s_2+1, s_1+s_2+2, \dots, s_1+s_2+s_3$$$;  ...  the $$$k$$$-th post contains videos: $$$n-s_k+1,n-s_k+2,\dots,n$$$. Polycarp is a perfectionist, he wants the total duration of videos in each post to be the same.Help Polycarp to find such positive integer values $$$s_1, s_2, \dots, s_k$$$ that satisfy all the conditions above.

Input: The first line contains two integers $$$n$$$ and $$$k$$$ ($$$1 \le k \le n \le 10^5$$$). The next line contains $$$n$$$ positive integer numbers $$$a_1, a_2, \dots, a_n$$$ ($$$1 \le a_i \le 10^4$$$), where $$$a_i$$$ is the duration of the $$$i$$$-th video.

Output: If solution exists, print "Yes" in the first line. Print $$$k$$$ positive integers $$$s_1, s_2, \dots, s_k$$$ ($$$s_1+s_2+\dots+s_k=n$$$) in the second line. The total duration of videos in each post should be the same. It can be easily proven that the answer is unique (if it exists).If there is no solution, print a single line "No".


Sample Inputs:
6 3
3 3 1 4 1 6

3 3
1 1 1

3 3
1 1 2

3 1
1 10 100

Sample Outputs:
Yes
2 3 1

Yes
1 1 1

No

Yes
3

Solution (Python):
(n, k) = map(int, input().split())
arr = list(map(int, input().split()))
tot = sum(arr)
if tot % k != 0:
    print('No')
else:
    tot //= k
    (idx, cur, ans) = (0, 0, [])
    for i in range(n):
        cur += arr[i]
        idx += 1
        if cur == tot:
            ans.append(idx)
            (idx, cur) = (0, 0)
        elif cur > tot:
            print('No')
            break
    else:
        if sum(ans) == n:
            print('Yes')
            print(' '.join(map(str, ans)))
        else:
            print('No')

Solution (Python):
(n, k) = map(int, input().split())
durations = [int(i) for i in input().split()]
total_duration = sum(durations)
if total_duration % k != 0:
    print('No')
else:
    post_duration = total_duration // k
    (count, temp_duration) = (0, 0)
    (ans, impossible) = ([], False)
    for duration in durations:
        temp_duration += duration
        count += 1
        if temp_duration > post_duration:
            impossible = True
            break
        if temp_duration == post_duration:
            ans.append(count)
            (count, temp_duration) = (0, 0)
    print('No') if impossible else print('Yes')
    if not impossible:
        print(*ans)

Solution (Python):
import sys
input = sys.stdin.readline
(n, k) = map(int, input().split())
w = list(map(int, input().split()))
s = sum(w)
if s % k != 0:
    print('No')
else:
    s = s // k
    (c, d, e) = (0, 0, [])
    for i in w:
        c += 1
        d += i
        if d == s:
            e.append(c)
            d = 0
            c = 0
        elif d > s:
            print('No')
            break
    else:
        print('Yes')
        print(' '.join(map(str, e)))