【教3妹学算法】统计得分小于 K 的子数组数目

手机软件开发 2024-10-4 10:46:26 114 0 来自 中国
标题:

一个数字的 分数 界说为数组之和 乘以 数组的长度。
比方说,[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 <= 10^5
1 <= k <= 10^15
java代码:

class Solution {    public long countSubarrays(int[] nums, long k) {        long ans = 0L, sum = 0L;        for (int left = 0, right = 0; right < nums.length; right++) {            sum += nums[right];            while (sum * (right - left + 1) >= k)                sum -= nums[left++];            ans += right - left + 1;        }        return ans;    }}
您需要登录后才可以回帖 登录 | 立即注册

Powered by CangBaoKu v1.0 小黑屋藏宝库It社区( 冀ICP备14008649号 )

GMT+8, 2024-11-21 20:45, Processed in 0.135403 second(s), 32 queries.© 2003-2025 cbk Team.

快速回复 返回顶部 返回列表