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和-1之间最大的连续0的个数。
因此我们只需要使用一个变量last来记录上一个非0数是1还是-1,再使用一个变量cnt来记录当前连续0的个数。
接着遍历地图数组:
-
如果当前元素非零:
- 就看是否为 “1遇到-1或-1遇到1”,如果是则更新答案最大值
- 更新cnt和last
-
否则(当前元素为0):c n t + + cnt++cnt++ 即可。
时间复杂度O ( l e n ( f o r t s ) ) O(len(forts))O(len(forts))
空间复杂度O ( 1 ) O(1)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 { // 0cnt++;}}return ans;}
};
GO
func captureForts(forts []int) (ans int) {n := len(forts)i := 0for i < n {j := i + 1if forts[i] != 0 {for j < n && forts[j] == 0 {j++}if j < n && forts[i]+forts[j] == 0 {ans = max(ans, j-i-1)}}i = j}return
}func max(a, b int) int {if a > b {return a}return b
}