LeetCode刷题日记（2511. 最多可以摧毁的敌人城堡数目）

'''
给你一个长度为 n ，下标从 0 开始的整数数组 forts ，表示一些城堡。forts[i] 可以是 -1 ，0 或者 1 ，其中：
-1 表示第 i 个位置 没有 城堡。
0 表示第 i 个位置有一个 敌人 的城堡。
1 表示第 i 个位置有一个你控制的城堡。
现在，你需要决定，将你的军队从某个你控制的城堡位置 i 移动到一个空的位置 j ，满足：
0 <= i, j <= n - 1
军队经过的位置 只有 敌人的城堡。正式的，对于所有 min(i,j) < k < max(i,j) 的 k ，都满足 forts[k] == 0 。
当军队移动时，所有途中经过的敌人城堡都会被 摧毁 。
请你返回最多可以摧毁的敌人城堡数目。如果无法移动你的军队，或者没有你控制的城堡，请返回 0 。
'''
from typing import List
class Solution:
    def captureForts(self, forts: List[int]) -> int:
        n = len(forts)  # 获取城堡数组的长度
        ret = 0
        last_not_zero = -1  # 用于记录上一个非零元素的索引
        for i in range(n):
            if forts[i] != 0:
                if last_not_zero != -1 and forts[last_not_zero] != forts[i]:
                    ret = max(ret, i - last_not_zero - 1)
                last_not_zero = i
        return ret