1375. 二进制字符串前缀一致的次数
1375. 二进制字符串前缀一致的次数 (Medium)
给你一个长度为 n
、下标从 1 开始的二进制字符串,所有位最开始都是 0
。我们会按步翻转该二进制字符串的所有位(即,将 0
变为 1
)。
给你一个下标从 1 开始的整数数组 flips
,其中 flips[i]
表示对应下标 i
的位将会在第 i
步翻转。
二进制字符串 前缀一致 需满足:在第 i
步之后,在 闭 区间 [1, i]
内的所有位都是 1 ,而其他位都是 0 。
返回二进制字符串在翻转过程中 前缀一致 的次数。
示例 1:
输入:flips = [3,2,4,1,5]
输出:2
解释:二进制字符串最开始是 "00000" 。
执行第 1 步:字符串变为 "00100" ,不属于前缀一致的情况。
执行第 2 步:字符串变为 "01100" ,不属于前缀一致的情况。
执行第 3 步:字符串变为 "01110" ,不属于前缀一致的情况。
执行第 4 步:字符串变为 "11110" ,属于前缀一致的情况。
执行第 5 步:字符串变为 "11111" ,属于前缀一致的情况。
在翻转过程中,前缀一致的次数为 2 ,所以返回 2 。
示例 2:
输入:flips = [4,1,2,3]
输出:1
解释:二进制字符串最开始是 "0000" 。
执行第 1 步:字符串变为 "0001" ,不属于前缀一致的情况。
执行第 2 步:字符串变为 "1001" ,不属于前缀一致的情况。
执行第 3 步:字符串变为 "1101" ,不属于前缀一致的情况。
执行第 4 步:字符串变为 "1111" ,属于前缀一致的情况。
在翻转过程中,前缀一致的次数为 1 ,所以返回 1 。
提示:
n == flips.length
1 <= n <= 5 * 10⁴
flips
是范围[1, n]
中所有整数构成的一个排列
分析
维护前缀1的尾部和最右侧1的位置,记为 lo 和 hi; 同时维护区间 [lo, hi] 之间的和 sum。
当 sum == hi-lo 时,得到了一个一致前缀,结果 +1,更新 lo 为 hi, sum 为 0。
时间复杂度 O(n),空间复杂度 O(1)。
func numTimesAllBlue(flips []int) int {
lo := 0
hi := 0
sum := 0
res := 0
for _, v := range flips {
sum++
if v > hi {
hi = v
}
if v == lo+1 {
lo++
sum--
}
if sum == hi-lo {
res++
lo = hi
sum = 0
}
}
return res
}
还可以更简单。仅维护 hi 和置为 1 的总数 sum,每次考虑 hi 和 sum 是否相等,相等则结果加 1.
复杂度同上。
func numTimesAllBlue(flips []int) int {
hi := 0
res := 0
for i, v := range flips {
if v > hi {
hi = v
}
if i+1 == hi { // i+1 is exactly sum
res++
}
}
return res
}
Local tests:
func Test_numTimesAllBlue(t *testing.T) {
tests := []struct {
name string
flips []int
want int
}{
{
flips: []int{3, 2, 4, 1, 5},
want: 2,
},
{
flips: []int{4, 1, 2, 3},
want: 1,
},
}
for _, tt := range tests {
if got := numTimesAllBlue(tt.flips); got != tt.want {
t.Errorf("%v. numTimesAllBlue() = %v, want %v", tt.flips, got, tt.want)
}
}
}