leetcode 2302. 统计得分小于 K 的子数组数目 困难

一个数组的 分数 定义为数组之和 乘以 数组的长度。

  • 比方说,[1, 2, 3, 4, 5] 的分数为 (1 + 2 + 3 + 4 + 5) * 5 = 75

给你一个正整数数组 nums 和一个整数 k ,请你返回 nums 中分数 严格小于 k非空整数子数组数目

子数组 是数组中的一个连续元素序列。

示例 1:

复制代码
输入:nums = [2,1,4,3,5], k = 10
输出:6
解释:
有 6 个子数组的分数小于 10 :
- [2] 分数为 2 * 1 = 2 。
- [1] 分数为 1 * 1 = 1 。
- [4] 分数为 4 * 1 = 4 。
- [3] 分数为 3 * 1 = 3 。 
- [5] 分数为 5 * 1 = 5 。
- [2,1] 分数为 (2 + 1) * 2 = 6 。
注意,子数组 [1,4] 和 [4,3,5] 不符合要求,因为它们的分数分别为 10 和 36,但我们要求子数组的分数严格小于 10 。

示例 2:

复制代码
输入:nums = [1,1,1], k = 5
输出:5
解释:
除了 [1,1,1] 以外每个子数组分数都小于 5 。
[1,1,1] 分数为 (1 + 1 + 1) * 3 = 9 ,大于 5 。
所以总共有 5 个子数组得分小于 5 。

提示:

  • 1 <= nums.length <= 10^5
  • 1 <= nums[i] <= 10^5
  • 1 <= k <= 10^15

分析:根据题目对数组分数的定义,以及 nums 是正整数数组这一条件,对于子数组 [i,j],当右端点 j 固定时,随着左端点 i 增加,子数组的和会减少,长度也会缩短,因此子数组的分数会单调递减。如果子数组 [i,j] 的分数小于 k,由于分数单调递减,那么子数组 [p,j],i<p≤j 的分数也小于 k。

基于以上性质,我们可以使用滑动窗口的方法求解。从 j=0 开始枚举子数组的右端点,并维护一个左端点 i(初始值为 0)。对于每个 j:

扩展窗口:将 nums[j] 加入当前窗口对应的子数组和 total。

收缩窗口:如果当前窗口对应的子数组分数 total×(j−i+1) 大于等于 k,说明子数组不符合要求,因此需要向右移动左端点 i,直到分数小于 k 为止。

计算子数组数量:此时,以 j 为右端点且分数小于 k 的子数组的数量为 j−i+1,累加到最终结果 ans。

枚举结束后,返回最终结果 ans。

cpp 复制代码
long long countSubarrays(int* nums, int numsSize, long long k) {
    long long ans=0,cnt=0;
    for(int l=0,r=0;r<numsSize;++r)
    {
        cnt+=nums[r];
        if(cnt*(r-l+1)<k)ans+=r-l+1;
        else
        {
            while(l<=r&&cnt*(r-l+1)>=k)
            {
                cnt-=nums[l],l++;
                if(cnt*(r-l+1)<k)
                    ans+=r-l+1;
            }
        }
    }
    return ans;
}
相关推荐
许愿与你永世安宁1 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子1 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
YuTaoShao2 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵
杰克尼3 小时前
1. 两数之和 (leetcode)
数据结构·算法·leetcode
YuTaoShao4 小时前
【LeetCode 热题 100】56. 合并区间——排序+遍历
java·算法·leetcode·职场和发展
JoJo_Way9 小时前
LeetCode三数之和-js题解
javascript·算法·leetcode
凌肖战11 小时前
力扣网C语言编程题:在数组中查找目标值位置之二分查找法
c语言·算法·leetcode
GEEK零零七13 小时前
Leetcode 1103. 分糖果 II
数学·算法·leetcode·等差数列
重庆小透明14 小时前
力扣刷题记录【1】146.LRU缓存
java·后端·学习·算法·leetcode·缓存
desssq15 小时前
力扣:70. 爬楼梯
算法·leetcode·职场和发展