[LeetCode] 1248. 统计「优美子数组」

滑动窗口的方法太牛了,学习学习到了

https://leetcode-cn.com/problems/count-number-of-nice-subarrays/solution/hua-dong-chuang-kou-qian-zhui-he-bi-xu-miao-dong-b/

package leetcode;

/**
 * @author doyinana
 * @create 2020-04-21 23:42
 */
public class L1248 {
    public int numberOfSubarrays(int[] nums, int k) {
        int left=0,right=0,oddCnt=0,res=0;
        while(right<nums.length){
            if((nums[right++]&1)==1){
                oddCnt++;
            }
            if(oddCnt==k){
                int tmp=right;
                while(right<nums.length&&(nums[right]&1)==0)
                    right++;
                int rightEvenCnt=right-tmp;
                int leftEvenCnt=0;
                while((nums[left]&1)==0){
                    leftEvenCnt++;
                    left++;
                }
                res+=(leftEvenCnt+1)*(rightEvenCnt+1);

                left++;
                oddCnt--;
            }
        }
        return res;
    }
}

 这个方法也太牛了!!赞赞赞 

class Solution {
    public int numberOfSubarrays(int[] nums, int k) {
        // 数组 prefixCnt 的下标是前缀和(即当前奇数的个数),值是前缀和的个数。
        int[] prefixCnt = new int[nums.length + 1];
        prefixCnt[0] = 1;
        // 遍历原数组,计算当前的前缀和,统计到 prefixCnt 数组中,
        // 并且在 res 中累加上与当前前缀和差值为 k 的前缀和的个数。
        int res = 0, sum = 0;
        for (int num: nums) {
            sum += num & 1;
            prefixCnt[sum]++;
            if (sum >= k) {
                res += prefixCnt[sum - k];
            }       
        }
        return res;
    }
}

 

posted @ 2020-04-22 00:00  doyi  阅读(192)  评论(0编辑  收藏  举报