Dima and To-do List (Codeforces 366B)
Tags: Implement, brute force

You helped Dima to have a great weekend, but it's time to work. Naturally, Dima, as all other men who have girlfriends, does everything wrong.Inna and Dima are now in one room. Inna tells Dima off for everything he does in her presence. After Inna tells him off for something, she goes to another room, walks there in circles muttering about how useless her sweetheart is. During that time Dima has time to peacefully complete k - 1 tasks. Then Inna returns and tells Dima off for the next task he does in her presence and goes to another room again. It continues until Dima is through with his tasks.Overall, Dima has n tasks to do, each task has a unique number from 1 to n. Dima loves order, so he does tasks consecutively, starting from some task. For example, if Dima has 6 tasks to do in total, then, if he starts from the 5-th task, the order is like that: first Dima does the 5-th task, then the 6-th one, then the 1-st one, then the 2-nd one, then the 3-rd one, then the 4-th one.Inna tells Dima off (only lovingly and appropriately!) so often and systematically that he's very well learned the power with which she tells him off for each task. Help Dima choose the first task so that in total he gets told off with as little power as possible.

Input: The first line of the input contains two integers n, k (1 ≤ k ≤ n ≤ 105). The second line contains n integers a1, a2, ..., an (1 ≤ ai ≤ 103), where ai is the power Inna tells Dima off with if she is present in the room while he is doing the i-th task.It is guaranteed that n is divisible by k.

Output: In a single line print the number of the task Dima should start with to get told off with as little power as possible. If there are multiple solutions, print the one with the minimum number of the first task to do.

Note: Explanation of the first example.If Dima starts from the first task, Inna tells him off with power 3, then Dima can do one more task (as k = 2), then Inna tells him off for the third task with power 1, then she tells him off for the fifth task with power 5. Thus, Dima gets told off with total power 3 + 1 + 5 = 9. If Dima started from the second task, for example, then Inna would tell him off for tasks 2, 4 and 6 with power 2 + 6 + 4 = 12. Explanation of the second example.In the second example k = 5, thus, Dima manages to complete 4 tasks in-between the telling off sessions. Thus, Inna tells Dima off for tasks number 1 and 6 (if he starts from 1 or 6), 2 and 7 (if he starts from 2 or 7) and so on. The optimal answer is to start from task 3 or 8, 3 has a smaller number, so the answer is 3.

Sample Inputs:
6 2
3 2 1 6 5 4

10 5
1 3 5 7 9 9 4 1 8 5

Sample Outputs:
1

3

Solution (Python):
def solve():
    rems = {}
    for i in range(k):
        rems[i] = 0
    for i in range(n):
        rems[i % k] += arr[i]
    mn = 10000000000
    ans = 0
    for i in rems:
        if mn > rems[i]:
            mn = rems[i]
            ans = i + 1
    return ans
from sys import stdin
input = stdin.readline
(n, k) = [int(x) for x in input().split()]
arr = [int(x) for x in input().split()]
print(solve())

Solution (Python):
(n, k) = map(int, input().split())
power = list(map(int, input().split()))
a = sum(power)
b = 0
for i in range(k):
    s = 0
    for j in range(i, len(power), k):
        s += power[j]
    if s < a:
        a = s
        b = i
print(b + 1)

Solution (Python):
(n, k) = map(int, input().split())
l = list(map(int, input().split()))
r = 100000000000
for i in range(n):
    if l[i] == 0:
        break
    j = i
    sum = 0
    while j < n:
        sum = sum + l[j]
        l[j] = 0
        j = j + k
    if r > sum:
        idx = i
        r = sum
print(idx + 1)