牛客网刷题&华为OD2024年真题

""" 题目描述 近些年来，我国防沙治沙取得显著成果。某沙漠新种植N棵胡杨（编号1-N），排成一排。 一个月后，有M棵胡杨未能成活。 现可补种胡杨K棵，请问如何补种（只能补种，不能新种），可以得到最多的连续胡杨树？ 输入描述 1、第一行包含一个整数 N，表示总种植数量。N 总种植数量，1 <= N <= 100000 2、第二行包含一个整数 M，表示未成活胡杨数量。1 <= M <= N 3、第三行包含M 个用空格分隔的整数，表示未成活胡杨的编号（按从小到大排列）。 4、第四行包含一个整数 K，表示最多可以补种的数量。，0 <= K <= M 输出描述 最多的连续胡杨棵树 考点：滑动窗口+双指针算法 具体步骤如下： 1.初始化窗口：定义一个窗口的初始大小。 2.滑动窗口：通过改变左右指针来调整窗口的大小。 3.检查窗口条件：在每次滑动窗口时，检查窗口内的元素是否满足某个条件。 4.记录结果：如果窗口内的元素满足条件，则记录结果，并调整窗口大小继续检查。 """ """ 解题思路如下： 首先，将胡杨树的状态表示为一个长度为 N 的数组，其中 1 表示存活的树，0 表示未成活的树。 使用滑动窗口（双指针）技术来解决这个问题： 1.用两个指针 left 和 right 表示当前考虑的区间。 2.移动 right 指针扩大窗口，直到窗口内未成活的树木数量超过 K。 3.记录当前窗口大小，这可能是一个潜在的最大连续序列。 4.如果窗口内未成活的树木数量超过了 K，移动 left 指针缩小窗口，直到未成活的树木数量再次不超过 K。 5.在整个过程中，持续更新最大连续树木数量。 """ def max_continuous_trees(n, m, d, k): """ :param n: 种植的胡杨树总数 :param m: 未成活的胡杨树总数，这个参数实际并没有什么用！ :param d: 未成活胡杨数的编号 :param k: 最多可以补种的数量 :return: 最多连续胡杨树 """ # 初始化数组，所有树都是成活的，0表示成活，1表示未成活 trees = [1] * (n + 1) # 这里使用的n+1用来保证数组下标和树的编号对应 # 标记死亡的树 for tree in d: trees[tree] = 0 left = right = 1 # 滑动窗口的左右指针 zeros = 0 # 当前窗口内 0 的数量 max_length = 0 # 最大连续树木数量，也就是最终要输出的结果 # 遍历数组，统计当前窗口中0的数量 for right in range(1, n + 1) : if trees[right] == 0: zeros += 1 # 如果窗口内的 0 超过了可补种数量，移动左指针 while zeros > k: if trees[left] == 0: zeros -= 1 left += 1 # 持续更新最大连续树木数量 max_length = max(max_length, right - left + 1) return max_length # 读取输入 N = int(input()) M = int(input()) dead_trees = list(map(int, input().split())) K = int(input()) # 计算并输出结果 print(max_continuous_trees(N, M, dead_trees, K))