题目:leecode 5457. 和为奇数的子数组数目
题目描述:
给你一个整数数组 arr 。请你返回和为 奇数 的子数组数目。
由于答案可能会很大,请你将结果对 10^9 + 7 取余后返回。
示例 1:
输入:arr = [1,3,5]
输出:4
解释:所有的子数组为 [[1],[1,3],[1,3,5],[3],[3,5],[5]] 。
所有子数组的和为 [1,4,9,3,8,5].
奇数和包括 [1,9,3,5] ,所以答案为 4 。
示例 2 :
输入:arr = [2,4,6]
输出:0
解释:所有子数组为 [[2],[2,4],[2,4,6],[4],[4,6],[6]] 。
所有子数组和为 [2,6,12,4,10,6] 。
所有子数组和都是偶数,所以答案为 0 。
示例 3:
输入:arr = [1,2,3,4,5,6,7]
输出:16
示例 4:
输入:arr = [100,100,99,99]
输出:4
示例 5:
输入:arr = [7]
输出:1
提示:
*1 <= arr.length <= 10^5
1 <= arr[i] <= 100*
解题思路(本人)一般思路:
思路讲解:
因为和为奇数的条件:奇数个奇数相加(偶数只能作为变动子数组的个数有价值)
奇数的个数 i:1,3,5,7,9…以次类推;
当前 (i个奇数)的符合要求的子数组的个数;
当i = 3时;
以 “00000(101001)000” :
(i个奇数)前的偶数为 5 ;
(i个奇数)后的偶数为 3 ;
故: 子数组的个数 5 * 3 + 5 + 3 + 1 = 4 * 6;
代码实现:
class Solution {
public:
int numOfSubarrays(vector<int>& arr) {
int i , j ;
vector<int> sub ;
sub.push_back(-1) ;
//记录奇数的位置信息
for (i = 0 ; i < arr.size() ; i ++)
{
if (arr[i] & 1)
sub.push_back(i) ;
}
sub.push_back(arr.size()) ;
long long ret = 0 ;
int cha , cha2 ;
//第一层是表示奇数的个数:1,3,5,7,......
for(i = 1 ; i <= sub.size() - 2 ; i = i + 2)
{
//计算i个奇数,前后的偶数的个数, (偶数)(i个奇数)(偶数)
for (j = 1 ; j + i <= sub.size() - 1 ; j ++)
{
cha = sub[j] - sub[j - 1] - 1 ;
cha2 = sub[j + i] - sub[j + i - 1] - 1 ;
// (cha * cha2 + cha + cha2 + 1)为当前的(i个奇数)存在的符合要求的子串的个数。
ret += (cha * cha2 + cha + cha2 + 1);
}
//cout << j << endl ;
}
return ret % 1000000007 ;
}
};
复杂度计算:
时间复杂度:O(n * n / 2);
空间复杂度:O(n); vector sub的size()最大为n;
解题思路(借鉴):
借鉴的作者:xxyyttxx
链接:https://leetcode-cn.com/problems/number-of-sub-arrays-with-odd-sum/solution/duan-ma-di-tui-shi-jian-on-kong-jian-o1-tjua-by-xx/
利用动态规划的思想,记录当前(i)位置的 符合要求的子数组数目;
例如:
偶数用‘0’表示,奇数用‘1’表示;
str = “010001000101010000”
old[i]:(i) 表示 在字符串中的下标,从0开始;old[0] = 0 ;
i = 0 : if(‘0’)old[i] = 0 ; else(‘1’) old[0] = 1 ;
循环:
i = 1 : old[i] = i + 1 - old[i -1] ;
i = 2 : old[i] = old[i -1] ;
…
…
循环结束条件:i < str .size() ;
代码实现:
class Solution {
public:
int numOfSubarrays(int* a, int n){
intmax_t ans = 0, old = 0;
for (int i=0; i<n; i++)
ans += old = a[i]&1 ? i + 1 - old : old;
return ans % 1000000007;
}
};
复杂度计算:
时间复杂度:O(n);
空间复杂度:O(1);