2025-07-04:统计符合条件长度为 3 的子数组数目。用go语言,给定一个整数数组 nums,请你计算有多少个长度恰好为 3 的连续子数组满足这样的条件:

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。

分步骤描述过程:

  1. 初始化变量

    • 首先,函数 countSubarrays 接收一个整数数组 nums 作为输入。
    • 计算数组的长度 n,即 nums 的元素个数。
    • 初始化计数器 ans 为 0,用于记录满足条件的子数组数量。
  2. 遍历数组

    • 使用一个 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。这个条件的意思是:中间元素的值是否等于第一个和第三个元素之和的两倍(即第一个和第三个元素之和是中间元素的一半)。
  3. 条件检查

    • 如果条件满足,则将计数器 ans 加 1。
    • 如果不满足,继续检查下一个 i
  4. 返回结果

    • 遍历完成后,返回 ans,即满足条件的子数组数量。
  5. 示例运行

    • 对于输入 nums = [1, 2, 1, 4, 1]
      • i = 1:检查 nums[0] = 1nums[1] = 2nums[2] = 1。条件是 2 == (1 + 1) * 2,即 2 == 4,不满足。
      • i = 2:检查 nums[1] = 2nums[2] = 1nums[3] = 4。条件是 1 == (2 + 4) * 2,即 1 == 12,不满足。
      • i = 3:检查 nums[2] = 1nums[3] = 4nums[4] = 1。条件是 4 == (1 + 1) * 2,即 4 == 4,满足,ans 加 1。
    • 最终 ans = 1,与题目描述一致。

时间复杂度:

  • 遍历数组时,for 循环的迭代次数是 n-2(因为 i 从 1 到 n-2)。
  • 每次迭代的操作是常数时间(比较和加法)。
  • 因此,总的时间复杂度是 O(n) ,其中 n 是数组的长度。

额外空间复杂度:

  • 只使用了固定数量的额外变量(nansi),与输入规模无关。
  • 因此,总的额外空间复杂度是 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)
相关推荐
SimonKing几秒前
拯救大文件上传:一文彻底彻底搞懂秒传、断点续传以及分片上传
java·后端·架构
深栈解码几秒前
JUC并发编程 内存布局和对象头
java·后端
37手游后端团队3 分钟前
巧妙利用装饰器模式给WebSocket连接新增持久化
后端
编程乐趣6 分钟前
C#版本LINQ增强开源库
后端
tonydf6 分钟前
记一次近6万多个文件的备份过程
windows·后端
前端付豪6 分钟前
13、你还在 print 调试🧾?教你写出自己的日志系统
后端·python
加瓦点灯7 分钟前
Spring AI + Milvus 实现 RAG 智能问答系统实战
后端
JohnYan8 分钟前
Bun技术评估 - 07 S3
javascript·后端·bun
vivo互联网技术10 分钟前
号码生成系统的创新实践:游戏周周乐幸运码设计
redis·后端·架构
这里有鱼汤11 分钟前
hvPlot:用你熟悉的 Pandas,画出你没见过的炫图
后端·python