LeetCode 双周赛：渐入佳境！

闲谈

2023-01-13 13:47:05

LeetCode 双周赛 108：大展身手，突破自我

前言

LeetCode 双周赛第 108 场已经落幕，让我们回顾这场精彩的算法挑战，看看各位算法高手是如何大展身手的。

T1. 最长交替子序列

题目解析

给定一个由 0 和 1 构成的数组，要求找出最长的交替子序列，即元素依次交替出现 0 和 1。例如，[0, 1, 0] 是交替子序列，而 [0, 0, 1] 不是。

解题思路

使用两个指针 i 和 j 分别指向数组头和尾。
指针同时向数组中心移动，检查是否符合交替子序列条件。
当指针之间的元素满足交替子序列条件时，更新最长交替子序列长度。
重复以上步骤，直至指针相遇或超出数组长度。

代码示例

def longest_alternating_subarray(A):
    if not A:
        return 0

    i = 0
    j = 1
    max_length = 1

    while j < len(A):
        if A[j] != A[j - 1]:
            max_length = max(max_length, j - i + 1)
            j += 1
        else:
            i = j
            j += 1

    return max_length

T2. 重新放置石块

题目解析

有 N 块石块，重量为 A[i]。任务是将石块重新放置到 K 个袋子中，使得每个袋子中的石块总重量相等。

解题思路

计算石块总重量。
对石块按重量从小到大排序。
使用贪心算法逐个将石块放入袋子。
若当前袋子重量加当前石块重量大于或等于总重量的 K 分之一，则将石块放入下一个袋子。
重复上述步骤，直至所有石块放入袋子。
检查每个袋子重量是否相等。

代码示例

def can_place_stones(N, K, A):
    if N > K:
        return False

    A.sort()

    total_weight = sum(A)

    if total_weight % K != 0:
        return False

    target_weight = total_weight // K

    bags = [0] * K

    for i in range(N):
        j = 0
        while j < K and bags[j] + A[i] <= target_weight:
            bags[j] += A[i]
            j += 1

    return bags[K - 1] == target_weight