2025-07-04:统计符合条件长度为 3 的子数组数目。用go语言,给定一个整数数组 nums,请你计算有多少个长度恰好为 3 的连续子数组满足这样的条件:子数组的第一个元素与第三个元素的和,正好是第二个元素的一半。请返回这个满足条件的子数组数量。
3 <= nums.length <= 100。
-100 <= nums[i] <= 100。
输入:nums = [1,2,1,4,1]。
输出:1。
解释:
只有子数组 [1,4,1] 包含 3 个元素且第一个和第三个数字之和是中间数字的一半。
题目来自力扣3392。
分步骤描述过程:
-
初始化变量:
- 首先,函数
countSubarrays
接收一个整数数组nums
作为输入。 - 计算数组的长度
n
,即nums
的元素个数。 - 初始化计数器
ans
为 0,用于记录满足条件的子数组数量。
- 首先,函数
-
遍历数组:
- 使用一个
for
循环,从索引i = 1
开始遍历到i = n-2
(即从第二个元素遍历到倒数第二个元素)。这是因为我们需要检查以i
为中心的三个连续元素(nums[i-1]
、nums[i]
、nums[i+1]
)。 - 对于每个
i
,检查是否满足条件:nums[i] == (nums[i-1] + nums[i+1]) * 2
。这个条件的意思是:中间元素的值是否等于第一个和第三个元素之和的两倍(即第一个和第三个元素之和是中间元素的一半)。
- 使用一个
-
条件检查:
- 如果条件满足,则将计数器
ans
加 1。 - 如果不满足,继续检查下一个
i
。
- 如果条件满足,则将计数器
-
返回结果:
- 遍历完成后,返回
ans
,即满足条件的子数组数量。
- 遍历完成后,返回
-
示例运行:
- 对于输入
nums = [1, 2, 1, 4, 1]
:i = 1
:检查nums[0] = 1
、nums[1] = 2
、nums[2] = 1
。条件是2 == (1 + 1) * 2
,即2 == 4
,不满足。i = 2
:检查nums[1] = 2
、nums[2] = 1
、nums[3] = 4
。条件是1 == (2 + 4) * 2
,即1 == 12
,不满足。i = 3
:检查nums[2] = 1
、nums[3] = 4
、nums[4] = 1
。条件是4 == (1 + 1) * 2
,即4 == 4
,满足,ans
加 1。
- 最终
ans = 1
,与题目描述一致。
- 对于输入
时间复杂度:
- 遍历数组时,
for
循环的迭代次数是n-2
(因为i
从 1 到n-2
)。 - 每次迭代的操作是常数时间(比较和加法)。
- 因此,总的时间复杂度是 O(n) ,其中
n
是数组的长度。
额外空间复杂度:
- 只使用了固定数量的额外变量(
n
、ans
、i
),与输入规模无关。 - 因此,总的额外空间复杂度是 O(1)。
Go完整代码如下:
go
package main
import (
"fmt"
)
func countSubarrays(nums []int) int {
n := len(nums)
ans := 0
for i := 1; i < n-1; i++ {
if nums[i] == (nums[i-1]+nums[i+1])*2 {
ans++
}
}
return ans
}
func main() {
nums := []int{1, 2, 1, 4, 1}
result := countSubarrays(nums)
fmt.Println(result)
}

Python完整代码如下:
python
# -*-coding:utf-8-*-
def count_subarrays(nums):
n = len(nums)
ans = 0
for i in range(1, n - 1):
if nums[i] == (nums[i - 1] + nums[i + 1]) * 2:
ans += 1
return ans
if __name__ == "__main__":
nums = [1, 2, 1, 4, 1]
result = count_subarrays(nums)
print(result)
