一、子数组最大平均数 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;
}