算法 子数组平均数(滑动窗口)


一、子数组最大平均数 I

1.1 题意

题目链接

给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。

请你找出平均数最大且 长度为 k 的连续子数组,并输出该最大平均数。

任何误差小于 10-5 的答案都将被视为正确答案。

示例 1:

输入:nums = [1,12,-5,-6,50,3], k = 4
输出:12.75
解释:最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
示例 2:

输入:nums = [5], k = 1
输出:5.00000

1.2 代码

double findMaxAverage(int* nums, int numsSize, int k)
{
	int sum = 0;
	double res = 0;

	int min_range = numsSize < k ? numsSize : k;

	for(int i = 0; i < min_range; i++)
	{
		sum += nums[i];
	}

	res = sum;

	for(int i = k; i < numsSize; i++)
	{
		sum = sum - nums[i-k] + nums[i];

		res = res < sum ? sum : res;
	}

	return res / k;
}

二、半径为 k 的子数组平均值

2.1 题意

题目链接

给你一个下标从 0 开始的数组 nums ,数组中有 n 个整数,另给你一个整数 k 。

半径为 k 的子数组平均值 是指:nums 中一个以下标 i 为 中心 且 半径 为 k 的子数组中所有元素的平均值,即下标在 i - k 和 i + k 范围(含 i - k 和 i + k)内所有元素的平均值。如果在下标 i 前或后不足 k 个元素,那么 半径为 k 的子数组平均值 是 -1 。

构建并返回一个长度为 n 的数组 avgs ,其中 avgs[i] 是以下标 i 为中心的子数组的 半径为 k 的子数组平均值 。

x 个元素的 平均值 是 x 个元素相加之和除以 x ,此时使用截断式 整数除法 ,即需要去掉结果的小数部分。

例如,四个元素 2、3、1 和 5 的平均值是 (2 + 3 + 1 + 5) / 4 = 11 / 4 = 2.75,截断后得到 2 。


2.2 代码

int* getAverages(int* nums, int numsSize, int k, int* returnSize)
{
	int* result_arr = NULL;
	*returnSize = numsSize;
    result_arr = malloc(sizeof(int) * numsSize);

	if(numsSize <= 2*k)
	{
		for(int i = 0; i < numsSize; i++)
		{
			result_arr[i] = -1;
		}

		return result_arr;
	}

	long sum = 0, diameter = 2*k + 1;

	for(int i = 0; i < diameter; i++)
	{
		sum += nums[i];

		if(i < k)
		{
			result_arr[i] = -1;
		}
	}


	result_arr[k] = sum / diameter;


	for(int i = k + 1; i < numsSize - k; i++)
	{
		sum = sum - nums[i - k - 1] + nums[i + k];

		result_arr[i] = sum / diameter;
	}

	for(int i = numsSize - k; i < numsSize; i++)
	{
		result_arr[i] = -1;
	}

	return result_arr;

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值