文章目录
一、题目
1、题目描述
给你一个整数数组 n u m s nums nums 和一个整数 k k k。如果某个连续子数组中恰好有 k k k 个奇数数字,我们就认为这个子数组是「优美子数组」。请返回这个数组中「优美子数组」的数目。
样例输入:nums = [1,1,2,1,1], k = 3
样例输出:2
2、基础框架
- C语言 版本给出的基础框架代码如下:
int numberOfSubarrays(int* nums, int numsSize, int k) {
}
3、原题链接
二、解题报告
1、思路分析
首先,用
s
u
m
sum
sum 统计奇数个数。
然后,假设某个子数组为
n
u
m
s
[
x
:
i
]
nums[x:i]
nums[x:i],两种情况:
1)如果
x
=
0
x = 0
x=0 时,
s
u
m
[
i
]
=
=
k
sum[i] == k
sum[i]==k,则
[
0
,
i
]
[0,i]
[0,i] 为一种方案,计数器 加一;
2)如果
0
<
x
≤
i
0 < x \le i
0<x≤i 时, 就是看有多少个
x
x
x 满足
s
u
m
[
i
]
−
s
u
m
[
x
−
1
]
=
=
k
sum[i] - sum[x-1] == k
sum[i]−sum[x−1]==k于是,就是要找到 满足
s
u
m
[
x
−
1
]
=
=
s
u
m
[
i
]
−
k
sum[x-1] == sum[i] - k
sum[x−1]==sum[i]−k 的最小的
x
x
x 和 最大的
x
x
x,相减+1 就是答案了。
2、时间复杂度
- 前缀和初始化,只需要一次遍历,时间复杂度为 O ( n ) O(n) O(n)。
- 统计过程,需要先枚举 i i i 作为右端点,左端点通过二分得到一个区间,时间复杂度 O ( n l o g 2 n ) O(nlog_2n) O(nlog2n)。
3、代码详解
1、辅助函数 minIndex
- 首先,我们实现一个辅助函数,它的作用是在一个可重复数组中,找到一个数 v v v ,且下标最小,如果找不到,返回 − 1 -1 −1;实现方法就是二分。
int minIndex(int *sum, int l, int r, int v) {
int mid, ans = -1;
while(l <= r) {
mid = (l + r) >> 1;
if( v <= sum[mid-1] ) {
r = mid - 1; // (1)
ans = mid; // (2)
}else {
l = mid + 1; // (3)
}
}
if(ans != -1 && sum[ans-1] != v) {
ans = -1; // (4)
}
return ans; // (5)
}
-
(
1
)
(1)
(1) 当满足
v <= sum[mid-1]
,满足条件的下标一定在区间 [ l , m i d ] [l, mid] [l,mid],并且 m i d mid mid 这个位置是一个可行解(最优解待定),所以将右区间缩小为 m i d − 1 mid-1 mid−1; - ( 2 ) (2) (2) 记录下可行解 m i d mid mid 到 a n s ans ans 中;
-
(
3
)
(3)
(3) 否则,
v > sum[mid-1]
,满足条件的下标一定在区间 [ m i d + 1 , r ] [mid+1, r] [mid+1,r],将左区间缩小为 m i d + 1 mid+1 mid+1; -
(
4
)
(4)
(4) 目前找到的位置只是一个
v <= sum[ans-1]
的位置,所以当v != sum[ans-1]
时,实际上这个位置是无效的,所以需要变回 − 1 -1 −1; - ( 5 ) (5) (5) 这时候的 a n s ans ans 为全局最优解,直接返回即可。
2、辅助函数 maxIndex
- 然后,我们依样画葫芦,再实现一个辅助函数,它的作用是在一个可重复数组中,找到一个数 v v v ,且下标最大,如果找不到,返回 − 1 -1 −1;实现方法也是二分。
int maxIndex(int *sum, int l, int r, int v) {
int mid, ans = -1;
while(l <= r) {
mid = (l + r) >> 1;
if( v >= sum[mid-1] ) {
l = mid + 1; // (1)
ans = mid; // (2)
}else {
r = mid - 1; // (3)
}
}
if(ans != -1 && sum[ans-1] != v) {
ans = -1; // (4)
}
return ans; // (5)
}
-
(
1
)
(1)
(1) 当满足
v >= sum[mid-1]
,满足条件的下标一定在区间 [ m i d , r ] [mid, r] [mid,r],并且 m i d mid mid 这个位置是一个可行解(最优解待定),所以将左区间缩小为 m i d + 1 mid+1 mid+1; - ( 2 ) (2) (2) 记录下可行解 m i d mid mid 到 a n s ans ans 中;
-
(
3
)
(3)
(3) 否则,
v < sum[mid-1]
,满足条件的下标一定在区间 [ l , m i d − 1 ] [l, mid-1] [l,mid−1],将右区间缩小为 m i d − 1 mid-1 mid−1; -
(
4
)
(4)
(4) 目前找到的位置只是一个
v >= sum[ans-1]
的位置,所以当v != sum[ans-1]
时,实际上这个位置是无效的,所以需要变回 − 1 -1 −1; - ( 5 ) (5) (5) 这时候的 a n s ans ans 为全局最优解,直接返回即可。
3、前缀和 + 线性枚举 + 二分统计求解
int numberOfSubarrays(int* nums, int numsSize, int k) {
int i, ans;
int l, r, mid;
for(i = 0; i < numsSize; ++i) {
sum[i] = (nums[i]&1) ? 1 : 0; // (1)
if(i) {
sum[i] += sum[i-1]; // (2)
}
}
ans = 0;
for(i = 0; i < numsSize; ++i) {
if(sum[i] == k) { // (3)
++ans;
}
l = minIndex(sum, 1, i, sum[i]-k); // (4)
r = maxIndex(sum, 1, i, sum[i]-k); // (5)
if(l != -1) {
ans += r - l + 1; // (6)
}
}
return ans;
}
-
(
1
)
(1)
(1)
x & 1
为 1 1 1 代表是奇数,否则是偶数; - ( 2 ) (2) (2) 计算前缀有多少个奇数;
-
(
3
)
(3)
(3)
nums[0:i]
这一段有 k k k 个奇数的情况需要统计进去; -
(
4
)
(4)
(4) 在开区间
(
0
,
i
)
(0, i)
(0,i) 中找到
sum[x-1]
等于sum[i]-k
的最小 x x x; -
(
5
)
(5)
(5) 在开区间
(
0
,
i
)
(0, i)
(0,i) 中找到
sum[x-1]
等于sum[i]-k
的最大 x x x; - ( 6 ) (6) (6) 最大 x x x 减去 最小 x x x 再加一,就是可行的 x x x 的个数;
三、本题小知识
这一章,我们学会了寻找某个 重复有序数组 中的一个数字的 最小下标 和 最大下标。
四、加群须知
相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」。
那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:
如果链接被屏蔽,或者有权限问题,可以私聊作者解决。
大致题集一览:
为了让这件事情变得有趣,以及「 照顾初学者 」,目前题目只开放最简单的算法 「 枚举系列 」 (包括:线性枚举、双指针、前缀和、二分枚举、三分枚举),当有 一半成员刷完 「 枚举系列 」 的所有题以后,会开放下个章节,等这套题全部刷完,你还在群里,那么你就会成为「 夜深人静写算法 」专家团 的一员。
不要小看这个专家团,三年之后,你将会是别人 望尘莫及 的存在。如果要加入,可以联系我,考虑到大家都是学生, 没有「 主要经济来源 」,在你成为神的路上,「 不会索取任何 」。
🔥联系作者,或者扫作者主页二维码加群,加入刷题行列吧🔥
🔥让天下没有难学的算法🔥
C语言免费动漫教程,和我一起打卡! 🌞《光天化日学C语言》🌞
入门级C语言真题汇总 🧡《C语言入门100例》🧡
几张动图学会一种数据结构 🌳《画解数据结构》🌳
组团学习,抱团生长 🌌《算法入门指引》🌌
竞赛选手金典图文教程 💜《夜深人静写算法》💜