560. 和为 K 的子数组

刷题找工作!


题目链接

. - 力扣(LeetCode)


自己的思路

我没思路(哭)。

模拟面试的时候遇到这个题,很熟悉,但是做不出来(哭)。

我用滑动窗口做的,但是我个人觉得很不合理,因为左右滑的条件找不到。

后面看了题解,大致思路猜测是用前缀和与map。

开干。


官方题解

思路是对的,但我做不出。

class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;
        HashMap<Integer, Integer> mp = new HashMap<>();
        mp.put(0, 1);//用put放,为什么放(0, 1)
        for(int i = 0 ; i < nums.length; i++){
            pre += nums[i];
            if(mp.containsKey(pre - k)){//哇哦天才,直接看是否有已经存在的前缀和使我们的值为k
                count += mp.get(pre - k);//存在, 所以把所有能这样的前缀和都加起来
                
            }
            //getOrDefault是如果不存在则创建, 括号里写的是创建的默认数据
            mp.put(pre, mp.getOrDefault(pre, 0) + 1);//将这个前缀和放入
            
        }
        return count;
    }
}

妙啊。

### 解题思路 LeetCode 2090 题目要求计算一个数组中每个元素的半径为 `k` 的子数组的平均值。如果某个元素的位置 `i` 满足其左右各有 `k` 个元素,即从 `i-k` 到 `i+k` 的子数组存在,则计算该子数组的平均值;否则结果为 `-1`。 为了高效地解决这个问题,可以使用 **滑动窗口** 技术。通过维护一个滑动窗口的,可以在 `O(n)` 时间复杂度内完成计算。 #### 关键点 - 初始化窗口的大小为 `2*k + 1`,如果数组长度小于这个值,则某些位置无法形成完整的子数组。 - 首先计算初始窗口的总,然后逐个移动窗口,并更新总(减去退出窗口的元素,加上进入窗口的元素)。 - 对于无法形成完整窗口的位置,直接标记为 `-1`。 --- ### C++ 实现代码 ```cpp #include <vector> using namespace std; class Solution { public: vector<int> getAverages(vector<int>& nums, int k) { long long n = nums.size(); long long windowSize = 2 * k + 1; vector<int> avg(n, -1); if (windowSize > n) return avg; // 如果窗口比数组还大,全部返回-1 long long sum = 0; for (int i = 0; i < windowSize; ++i) { sum += nums[i]; } avg[k] = sum / windowSize; // 第一个窗口的平均值 for (int i = windowSize; i < n; ++i) { sum += nums[i] - nums[i - windowSize]; // 滑动窗口更新 avg[i - k] = sum / windowSize; // 计算当前窗口的平均值 } return avg; } }; ``` --- ### 示例说明 假设输入如下: ```cpp vector<int> nums = {1, 2, 3, 4, 5, 6, 7, 8, 9}; int k = 3; ``` 输出将是: ```cpp {-1, -1, -1, 4, 5, 6, -1, -1, -1} ``` 其中只有中间部分能够形成完整的窗口,其他位置由于缺少足够的元素而返回 `-1`。 --- ### 复杂度分析 - **时间复杂度**:`O(n)`,其中 `n` 是数组的长度。整个算法只遍历数组一次。 - **空间复杂度**:`O(n)`,用于存储最终的结果数组。 --- ### 性能优化建议 - 使用 `long long` 类型确保累加过程中不会溢出。 - 提前处理窗口大小大于数组长度的情况,避免无效计算。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值