Python 1-06 练习

2057. 值相等的最小索引

class Solution:
    def smallestEqual(self, nums: List[int]) -> int:
        for i, x in enumerate(nums):
            if i % 10 == x: return i
        return -1 

2591. 将钱分给最多的儿童

class Solution:
    def distMoney(self, money: int, children: int) -> int:
        if money < children: return -1
        money -= children # 每人至少 1 美元
        res = min(money // 7, children) # 初步分配，让尽量多的人分到 8 美元
        money -= res * 7
        children -= res
        if (children == 0 and money > 0) or (children == 1 and money == 3):
            res -= 1
        return res

2706. 购买两块巧克力

class Solution:
    def buyChoco(self, prices: List[int], money: int) -> int:
        a = b = inf
        for x in prices:
            if x < a:
                b, a = a, x
            elif x < b:
                b = x
        return money - a - b if a + b <= money else money 

2455. 可被三整除的偶数的平均值

class Solution:
    def averageValue(self, nums: List[int]) -> int:
        sum = count = 0
        for x in nums:
            if x % 6 == 0:
                sum += x
                count += 1
        return 0 if count == 0 else sum // count

2078. 两栋颜色不同且距离最远的房子

class Solution:
    def maxDistance(self, colors: List[int]) -> int:
        n = len(colors)
        res = 0   # 两栋颜色不同房子的最远距离
        # 遍历两栋房子下标并维护最远距离
        for i in range(n):
            for j in range(i + 1, n):
                if colors[i] != colors[j]:
                    res = max(res, j - i)
        return res

优化

class Solution:
    def maxDistance(self, colors: List[int]) -> int:
        n = len(colors)
        c = colors[0]
        if c != colors[-1]: return n - 1
        i, j = 1, n - 2
        while colors[i] == c: i += 1
        while colors[j] == c: j -= 1
        return max(j, n - i - 1)

2073. 买票需要的时间

class Solution:
    def timeRequiredToBuy(self, tickets: List[int], k: int) -> int:
        n = len(tickets)
        res = 0
        for i in range(n):
            # 遍历计算每个人所需时间
            if i <= k:
                res += min(tickets[i], tickets[k])
            else:
                res += min(tickets[i], tickets[k] - 1)
        return res

1848. 到目标元素的最小距离

class Solution:
    def getMinDistance(self, nums: List[int], target: int, start: int) -> int:
        res = inf
        for i, x in enumerate(nums):
            if x == target:
                res = min(res, abs(i - start))
        return res

1732. 找到最高海拔

class Solution:
    def largestAltitude(self, gain: List[int]) -> int:
        # 差分逆运算前缀和
        m = pre = 0
        for x in gain:
            pre += x
            if pre > m: m = pre
        return m

1413. 逐步求和得到正数的最小值

class Solution:
    def minStartValue(self, nums: List[int]) -> int:
        ans = 1
        acc = 0
        #mn = inf
        for i, x in enumerate(nums):
            acc += x
            if acc < 0:
                ans = max(ans, 1 - acc)
                #mn = min(mn, acc)
        return ans
        #return 1 if mn > 0 else 1 - mn