class Solution {
public:
int countSubarrays(vector<int>& nums) {
}
};
3392. 统计符合条件长度为 3 的子数组数目
给你一个整数数组 nums ，请你返回长度为 3 的子数组，满足第一个数和第三个数的和恰好为第二个数的一半。
子数组 指的是一个数组中连续 非空 的元素序列。
示例 1：
输入：nums = [1,2,1,4,1]
输出：1
解释：
只有子数组 [1,4,1] 包含 3 个元素且第一个和第三个数字之和是中间数字的一半。number.
示例 2：
输入：nums = [1,1,1]
输出：0
解释：
[1,1,1] 是唯一长度为 3 的子数组，但第一个数和第三个数的和不是第二个数的一半。
提示：
3 <= nums.length <= 100-100 <= nums[i] <= 100原站题解
golang 解法, 执行用时: 0 ms, 内存消耗: 6.3 MB, 提交时间: 2024-12-25 17:38:57
func countSubarrays(nums []int) (ans int) {
for i := 2; i < len(nums); i++ {
if (nums[i-2]+nums[i])*2 == nums[i-1] {
ans++
}
}
return
}
cpp 解法, 执行用时: 0 ms, 内存消耗: 47.5 MB, 提交时间: 2024-12-25 17:38:42
class Solution {
public:
int countSubarrays(vector<int>& nums) {
int ans = 0;
for (int i = 2; i < nums.size(); i++) {
if ((nums[i - 2] + nums[i]) * 2 == nums[i - 1]) {
ans++;
}
}
return ans;
}
};
java 解法, 执行用时: 1 ms, 内存消耗: 44.1 MB, 提交时间: 2024-12-25 17:38:31
class Solution {
public int countSubarrays(int[] nums) {
int ans = 0;
for (int i = 2; i < nums.length; i++) {
if ((nums[i - 2] + nums[i]) * 2 == nums[i - 1]) {
ans++;
}
}
return ans;
}
}
python3 解法, 执行用时: 4 ms, 内存消耗: 17.4 MB, 提交时间: 2024-12-25 17:38:17
# 遍历统计
class Solution:
def countSubarrays(self, nums: List[int]) -> int:
return sum((nums[i - 2] + nums[i]) * 2 == nums[i - 1]
for i in range(2, len(nums)))