算法实战:力扣数组篇(1)

题目:1343. 大小为 K 且平均值大于等于阈值的子数组数目

给你一个整数数组 arr 和两个整数 k 和 threshold 。

请你返回长度为 k 且平均值大于等于 threshold 的子数组数目。

示例 :

输入:arr = [2,2,2,2,5,5,5,8], k = 3, threshold = 4
输出:3
解释:子数组 [2,5,5],[5,5,5] 和 [5,5,8] 的平均值分别为 4,5 和 6 。其他长度为 3 的子数组的平均值都小于 4
(threshold 的值)。

解题思路1:题目很简洁,也很容易懂,一开始思路很简单,直接暴力求解,先从数组的第一个元素开始,求出k个元素的平均值,然后判断平均值是否大于等于threshold,如果符合条件,数组个数加一,继续进行判断,最终返回符合条件的数组长度。代码如下:

int numOfSubarrays(int* arr, int arrSize, int k, int threshold){
	int count = 0;						             //记录符合条件的子数组长度
    float result = 0;								//用来保存每次计算的平均值
    for(int i = 0;i <= arrSize - k;i++)		//这里i小于等于arrSize - k,防止下一个循环的k + i导致arr[j]数组下标越界
    {
        result = 0;
        for(int j = i;j < k + i;j++)
        {
            result += arr[j];
        }
        if((result / k) >= threshold)
        {
            count++;
        }
    }
    return count;
}

提交过后不出所料,超出时间限制…

解题思路2:首先将k * threshold,然后将数组中前k个数求和再减去k * threshold,如果结果大于等于0,说明该子数组符合条件,符合条件的子数组数count加一.
指针后移一位,用后移一位的值减去移动之前的第一位的值,再加上上次减法的结果,如果大于0,说明符合要求.(因为每次后移一位在求和的话中间必有一段数据是重复的,只有移动前的的第一位和移动后的最后一位是改变的数,只需考虑这一部分的数据即可)上代码:

int numOfSubarrays(int* arr, int arrSize, int k, int threshold){
    int count = 0;
    int sum= 0;
    threshold *= k;
    for(int i = 0;i < k;i++){
        sum += arr[i];
    }
    int temp = sum - threshold;
    if(temp >= 0){
        count++;
    }
    for(int i = 1;i <= arrSize - k;i++){
        temp += (arr[i + (k - 1)] - arr[i - 1]);
        if(temp >= 0){
            count++;
        }
    }
    return count;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值