LeeCode热题 100打卡第八天
第一道题
和为k的子数组(LeeCode第560题):
给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数.子数组是数组中元素的连续非空序列。
主要利用前缀和与HashMap,用HashMap记录相同前缀和的值,通过前缀和的差值找到和为k的一段子序列,每次都找[0,j-1]这个区间里面满足条件的前缀和,此时hashmap里面存的值,可以计算出当前的j所对应的解有多少。
class Solution {
public int subarraySum(int[] nums, int k) {
if(nums == null || nums.length ==0) return 0;
HashMap<Integer, Integer> map = new HashMap<>();
map.put(0,1);
int count = 0;
int preSum = 0;
for(int j = 0; j< nums.length; j++){
preSum += nums[j];
int diff = preSum - k;
count += map.getOrDefault(diff, 0);
map.put(preSum, map.getOrDefault(preSum, 0) + 1);
}
return count;
}
}
第二道题
滑动窗口最大值(LeeCode第239题):
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。返回 滑动窗口中的最大值 。
解法一:暴力求解
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
List<Integer> result = new ArrayList<>();
if (nums == null || nums.length == 0 || k <= 0 || k > nums.length) return new int[0];
if (k == 1) return nums;
int max;
for (int i = 0; i <= nums.length - k; i++) {
max = nums[i];
for (int j = i; j < i + k; j++) {
max = Math.max(max, nums[j]);
}
result.add(max);
}
int[] resultArray = new int[result.size()];
for(int i = 0;i < result.size(); i++){
resultArray[i] = result.get(i);
}
return resultArray;
}
}
解法二:双端队列
每次都只维护窗口内最大的值,首先让第一个窗口内的值入队列,找到最大的值,将最大值前面的元素出栈,每次都只维护最大的值(这是最主要的思想)
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
ArrayDeque<Integer> deque = new ArrayDeque<>();
int n = nums.length;
int [] res = new int[n - k + 1];
int idx= 0;
for(int i = 0; i < n; i++){
while(!deque.isEmpty() && deque.peek() < i - k + 1){
deque.poll();
}
while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]){
deque.pollLast();
}
deque.offer(i);
if(i >= k - 1){
res[idx++] = nums[deque.peek()];
}
}
return res;
}
}