【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 。

示例 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

解答

源代码

class Solution {
    public int captureForts(int[] forts) {
        int pre = -1, res = 0;
 
        for (int i = 0; i < forts.length; i++) {
            if (forts[i] == 1 || forts[i] == -1) {
                // 下标 i 之前有一个相邻（不考虑0）的和自己不相等的1或者-1
                if (pre >= 0 && forts[i] != forts[pre]) {
                    res = Math.max(res, i - pre - 1);
                }
 
                // 有新的1/-1，就更新pre
                pre = i;
            }
        }
 
        return res;
    }
}

总结

理解题意，归根结底我们需要找的其实是两个相邻（不考虑 0）的 1 和 -1 之间的 0 最多有几个。

依次遍历数组 forts 中的每个元素，我们用pre记录数组中前一个为 1 或 -1 的位置；如果当前forts[i]为 1 或 -1 ，那么判断它前面是否有 1 或者 -1 而且值与它不同，如果有，那就更新结果 res ，如果没有就不做操作，最后再将 pre 更新为当前坐标。