LeetCode和为奇数的子数组数目

Posted 2022-12-12 头号理想

题目

给你一个整数数组 arr 。请你返回和为 奇数 的子数组数目。

由于答案可能会很大，请你将结果对 10^9 + 7 取余后返回。

这个题看上去也不难吧 充其量也就是几层循环就搞定了

但是那就难在时间复杂度上
我们用三层循环可以很轻松的解决
但是用脚都能想到会超时

然后我又尝试了两层循环 不出意外 超时

 int numOfSubarrays(vector<int>& arr) 
  vector<int> nums;
	int ans = 0;
	for (int i = 0; i < arr.size(); i++) 
		int cur = 0;
		for (int j = i; j < arr.size(); j++) 
			cur += arr[j];
			if (cur % 2 == 1) 
				ans++;
			
		
	
	return ans;
    

所以只能想别的方法
这个题我们尝试使用前缀和的方法 并且我们思考
奇数=奇数-偶数
奇数=偶数-奇数
想要得到奇数只能是上边两种情况

于是我们就维护两个值 一个是当前位置之前奇数的个数，一个偶数的个数
如果当前位置是奇数 就让结果加上之前的偶数的个数
如果当前是偶数 就让结果加上之前的奇数的个数

**注意：默认偶数个数是1 因为我们不要忽略0的存在 **

比如一个数组为[1,3,5]
当 当前位置是1的时候 如果没有初始的0 那么这个用例将不会写入到结果中
但是事实上1是符合题目要求的

代码

int numOfSubarrays(vector<int>& arr) 
 int cur = 0;
	int evencount=1;//之前偶数的个数 (别忘了有一位0)
	int oddcount=0;//之前奇数的个数
	
	int ans = 0;
	int num = pow(10, 9);
	num += 7;
	for (int i = 0; i < arr.size(); i++) 
		cur += arr[i];
		if (cur % 2 == 1) 
			ans += evencount;
			oddcount++;//这个是奇数 加一
		
		else 
			ans += oddcount;
			evencount++; //上边相反
		
		ans %= num;
	
	return ans;
    

这个题是好久之前一道LeetCode双周赛的第二道中等难度的题
希望上边我所写的对大家有帮助