历年蓝桥杯青少组Python中/高级选拔赛（STEMA）真题解析 | 2025年3月编程题（10）

​欢迎大家订阅我的专栏：算法题解：C++与Python实现！
本专栏旨在帮助大家从基础到进阶 ，逐步提升编程能力，助力信息学竞赛备战！

专栏特色
1.经典算法练习：根据信息学竞赛大纲，精心挑选经典算法题目，提供清晰的代码实现与详细指导，帮助您夯实算法基础。
2.系统化学习路径：按照算法类别和难度分级，从基础到进阶，循序渐进，帮助您全面提升编程能力与算法思维。

适合人群：

• 准备参加蓝桥杯、GESP、CSP-J、CSP-S等信息学竞赛的学生
• 希望系统学习C++/Python编程的初学者
• 想要提升算法与编程能力的编程爱好者

附上汇总贴：历年蓝桥杯青少组Python中/高级选拔赛（STEMA）真题解析 | 汇总

---

【题目描述】

有 n 颗棋子排成一排，每颗棋子的颜色为白色（用 1 表示）或黑色（用 0 表示）。每次可以选择从最左端或者最右端移除一颗棋子，目标是使剩余的棋子中白色棋子数量为 m。

给定两个整数 n 和 m，及 n 颗棋子的颜色排列。请计算最少要移除多少颗棋子，才能使剩余的棋子中白色棋子数量为 m；如果无法实现该目标，输出 -1。

例1：n = 8，m = 2；8 颗棋子的颜色分别是 0 1 0 1 1 0 0 1，要使剩余的棋子中白色棋子数量为 2，最少需要移除 3 颗棋子，移除方案如下：

第一次，移除最右端的白色棋子，移除后剩余棋子的颜色分别是 0 1 0 1 1 0 0；

第二次，移除最左端的黑色棋子，移除后剩余棋子的颜色分别是 1 0 1 1 0 0；

第三次，移除最左端的白色棋子，移除后剩余棋子的颜色分别是 0 1 1 0 0；

此时，剩余棋子中白色棋子数量为 2。

例2：n=5，m=3；5 颗棋子的颜色分别是 1 0 0 1 0，无论如何移除棋子都不能使剩余棋子中白色棋子数量为 3，则输出 -1。

【输入】

第一行输入两个整数 n，m((1≤n，m≤1000)，分别表示初始棋子数量和目标白色棋子数量，整数之间以一个空格隔开；

第二行输入 n 个整数（整数为 1 或 0，1 表示白色棋子，0 表示黑色棋子），表示从左到右每颗棋子的颜色，整数之间以一个空格隔开。

【输出】

输出一个整数，表示最少要移除多少颗棋子，才能使剩余棋子中白色棋子的数量为 m；如果无论如何移除棋子都不能使剩余棋子中白色棋子数量为 m，则输出 -1。

【输入样例】

8 2
0 1 0 1 1 0 0 1

【输出样例】

3

【代码详解】

n, m = map(int, input().split())  # 输入n和m，n表示数组长度，m表示需要保留的白色方块数量
colors = list(map(int, input().split()))  # 输入颜色数组，1表示白色，0表示黑色
total_white = sum(colors)  # 计算总共有多少个白色方块

# 如果总白色方块数量不足m，直接输出-1
if total_white < m:
    print(-1)
else:
    left = 0  # 左指针
    right = n - 1  # 右指针
    removed = 0  # 记录移除的方块数量
  
    # 使用双指针法移除方块
    while left <= right and total_white > m:
        # 优先从左侧移除白色方块
        if colors[left] == 1:
            total_white -= 1
            left += 1
            removed += 1
        # 其次从右侧移除白色方块
        elif colors[right] == 1:
            total_white -= 1
            right -= 1
            removed += 1
        # 如果两侧都不是白色方块，则从左侧移除黑色方块
        else:
            left += 1
            removed += 1
  
    print(removed)  # 输出需要移除的方块总数

【运行结果】

8 2
0 1 0 1 1 0 0 1
3