TopCoder SRM 680 Div 2

srm680-a-BearPair.py

import itertools
class BearPair:
    def bigDistance(self, s): 
        n = len(s)
        res = -1
        for e in itertools.combinations(xrange(n), 2):
            i = e[0]
            j = e[1]
            if s[i] != s[j]:
                res = max(res, abs(i - j))
        return res

srm680-b-BearChairs.py

import bisect
class BearChairs:
    def findPositions(self, atLeast, d): 
        res = []
        occupied = []
        for i in atLeast:
            tmp = i
            for j in occupied:
                if j - d < tmp < j + d:
                    tmp = j + d
            res.append(tmp)
            bisect.insort_left(occupied, tmp)
        return tuple(res)