[leetcode日记]1248.统计优美子数组

题目描述：

给你一个整数数组 nums 和一个整数 k。
如果某个 连续 子数组中恰好有 k 个奇数数字，我们就认为这个子数组是「优美子数组」。
请返回这个数组中「优美子数组」的数目。

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：参考了题解，使用双指针。因为奇数的数量k是固定的，所以如果超出数量左指针右移，如果少于数量右指针右移，如果数量相等统计左指针左侧0数量的个数，右指针右侧连续0数量的个数。

可行的优化方法：可以记录每个奇数左右相邻的连续0的个数，甚至可以记下他们对应的下标，在取数和判断的时候能大大加快运算速度。时间复杂度O（n），空间复杂度O（n）。

基本代码如下，没有考虑上述优化方法，空间复杂度是O（1）：

#define MMAX(a, b)        ((a) > (b)? (a) : (b))
#define MMIN(a, b)        ((a) < (b)? (a) : (b))

#define IS_ODD(a)           ((a) % 2 == 1)

//【算法思路】双指针。
int numberOfSubarrays(int* nums, int numsSize, int k){
    int ll, rr;
    //找到第一个奇数
    for(int i = 0; i < numsSize; i++) {
        if(IS_ODD(nums[i])) {
            ll = i;
            rr = i;
            break;
        }
    }

    int odd_cnt = 0;
    int ret = 0;
    //由rr增加当前奇数个数，ll移动将个数降低
    while(rr < numsSize) {
        if(IS_ODD(nums[rr])) {
            odd_cnt++;
        }

        if(odd_cnt < k) {
            rr++;
            continue;
        }
        //找到k个奇数，开始统计个数
        //寻找左边偶数的个数
        int lcnt = 0;
        for(int i = ll - 1; i >= 0; i--) {
            if(IS_ODD(nums[i])) {
                break;
            }
            lcnt++;
        }
        //寻找右边偶数的个数
        int rcnt = 0;
        for(int i = rr + 1; i < numsSize; i++) {
            if(IS_ODD(nums[i])) {
                break;
            }
            rcnt++;
        }
        ret += (lcnt + 1) * (rcnt + 1);
        //更新下一个奇数
        bool find = false;
        for(int i = ll + 1; i < numsSize; i++) {
            if(IS_ODD(nums[i])) {
                ll = i;
                find = true;
                break;
            }
        }
        if(find == false) {
            break;
        }
        //启动下一次寻找
        rr++;
        odd_cnt--;
    }
    return ret;
}

测试结果如下：

我还编写了考虑优化的代码：

#define MMAX(a, b)        ((a) > (b)? (a) : (b))
#define MMIN(a, b)        ((a) < (b)? (a) : (b))

#define IS_ODD(a)           ((a) % 2 == 1)

//【算法思路】双指针。
int numberOfSubarrays(int* nums, int numsSize, int k){	//默认k>0否则还需要些边界处理条件 

	int index[numsSize + 2]; 	//记录第i个奇数的下标，两侧都设有哨兵，+2防止栈溢出 
	int count = 0,sum = 0;
	index[0] = -1;	//作为边界 
    for (int i = 0;i < numsSize; i++){
    	if (IS_ODD(nums[i])){
    		index[++count] = i;	//记录第count+1个奇数的位置 
		} 
	}
	if (count < k ){
		return 0;
	} 
	index[count + 1] = numsSize;  
	for (int i = 1; i <= count - k + 1; i++){
		sum += (index[i]-index[i-1]) * (index[i+k] - index[i+k-1]); 
	}

    return sum;
}

以下是运行结果，可以看到速度有所提升但不明显，排名提升不少可惜还是没有到达90%。

由于leetcode平台运行时间和机器有关，甚至相同代码提交多次处理时间也不同；我参考了评论区其他的C语言做法，没有发现明显比我第二种优化下小的。所以我决定不再继续优化了。

这是我发的第一篇与leetcode有关的博文，以后多尝试尝试嘛hhh。