LeetCode 2511. 最多可以摧毁的敌人城堡数目-Toy模板网

这篇具有很好参考价值的文章主要介绍了LeetCode 2511. 最多可以摧毁的敌人城堡数目。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

【LetMeFly】2511.最多可以摧毁的敌人城堡数目

力扣题目链接：https://leetcode.cn/problems/maximum-enemy-forts-that-can-be-captured/

给你一个长度为 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 。

示例 1：

输入：forts = [1,0,0,-1,0,0,0,0,1]
输出：4
解释：
- 将军队从位置 0 移动到位置 3 ，摧毁 2 个敌人城堡，位置分别在 1 和 2 。
- 将军队从位置 8 移动到位置 3 ，摧毁 4 个敌人城堡。
4 是最多可以摧毁的敌人城堡数目，所以我们返回 4 。

示例 2：

输入：forts = [0,0,1,-1]
输出：0
解释：由于无法摧毁敌人的城堡，所以返回 0 。

提示：

1 <= forts.length <= 1000
-1 <= forts[i] <= 1

方法一：遍历

这道题说白了就是问你1和-1之间最大的连续0的个数。

因此我们只需要使用一个变量last来记录上一个非0数是1还是-1，再使用一个变量cnt来记录当前连续0的个数。

接着遍历地图数组：

如果当前元素非零：
- 就看是否为 “1遇到-1或-1遇到1”，如果是则更新答案最大值
- 更新cnt和last
否则（当前元素为0）： $c n t + +$

即可。

时间复杂度 $O (l e n (f or t s))$
空间复杂度 $O (1)$

AC代码

C++

class Solution {
public:
    int captureForts(vector<int>& forts) {  // 1和-1之间最多连续0的个数
        int ans = 0;
        int last = 2, cnt = 0;
        for (int i = 0; i < forts.size(); i++) {
            if (forts[i]) {
                if ( last != forts[i] && last != 2) {
                    ans = max(ans, cnt);
                }
                last = forts[i];
                cnt = 0;
            }
            else {  // 0
                cnt++;
            }
        }
        return ans;
    }
};

Python

# from typing import List

class Solution:
    def captureForts(self, forts: List[int]) -> int:
        ans = 0
        last, cnt = 2, 0
        for fort in forts:
            if fort:
                if fort != last and last != 2:
                    ans = max(ans, cnt)
                cnt = 0
                last = fort
            else:
                cnt += 1
        return ans