TopCoder SRM 670

SRM670_Div2_Easy_Cdgame.py

class Cdgame():
  def rescount(self, a, b):
    sum_a = sum(a)
    sum_b = sum(b)
    res = []
    for i in a:
        for j in b:
            res.append((sum_a-i+j)*(sum_b-j+i))
    return(len(set(res)))

SRM670_Div2_Med_Drbalance.py

class Drbalance():
    def lesscng(self, s, k):
        l = list(s)
        res = 0
        while True:
            costs = [l.count("+") - l.count("-")]
            for i in range(len(l)):
                ll = l[:-i]
                costs.append(ll.count("+") - ll.count("-"))
            costs_sum = len([i for i in costs if i < 0])
            if costs_sum > k:
                i = l.index("-")
                l[i] = "+"
                res += 1
            else:
                return res