1978D - Elections - CodeForces Solution


data structures greedy implementation

Please click on ads to support us..

Python Code:

t = int(input())
for tc in range (t):
    n,c = map(int,input().split(' '))
    a = list(map(int,input().split(' ')))
    maxvote = a[0]
    index = 0
    for i in range (0,n):
        if (a[i] > maxvote):
            maxvote = a[i]
            index = i
    if (index == 0):
        maxvote += c
    else:
        if (a[0] + c >= maxvote):
            maxvote = a[0]+c
            index = 0
        preficSum = []
    sum = 0
    for i in range (0,n):
        sum += a[i]
        preficSum.append(sum)

    l = []
    for i in range (0,n):
        if (i== 0):
            if (a[i] + c >= maxvote):
                l.append(0)
            else:
                l.append(1)
        else:
            if (i == index):
                l.append(0)
            elif (a[i] < maxvote):
                if (preficSum[i] + c >= maxvote):
                    l.append(i)
                else:
                    l.append(i+1)
            else:
                if (i <= index):
                    l.append(0)
                else:
                    l.append(i)
    print(' '.join(map(str,l)))


Comments

Submit
0 Comments
More Questions

415. Add Strings
22. Generate Parentheses
13. Roman to Integer
2. Add Two Numbers
515. Find Largest Value in Each Tree Row
345. Reverse Vowels of a String
628. Maximum Product of Three Numbers
1526A - Mean Inequality
1526B - I Hate 1111
1881. Maximum Value after Insertion
237. Delete Node in a Linked List
27. Remove Element
39. Combination Sum
378. Kth Smallest Element in a Sorted Matrix
162. Find Peak Element
1529A - Eshag Loves Big Arrays
19. Remove Nth Node From End of List
925. Long Pressed Name
1051. Height Checker
695. Max Area of Island
402. Remove K Digits
97. Interleaving String
543. Diameter of Binary Tree
124. Binary Tree Maximum Path Sum
1465. Maximum Area of a Piece of Cake After Horizontal and Vertical Cuts
501A - Contest
160A- Twins
752. Open the Lock
1535A - Fair Playoff
1538F - Interesting Function