优美的子数组
给你一个整数数组 nums 和一个整数 k。
如果某个 连续 子数组中恰好有 k 个奇数数字,我们就认为这个子数组是「优美子数组」。
请返回这个数组中「优美子数组」的数目。
输入:nums = [1,1,2,1,1], k = 3
输出:2
解释:包含 3 个奇数的子数组是 [1,1,2,1] 和 [1,2,1,1] 。
思路:满脑子只有暴力循环,害。。。滑动窗口
果然超出时间限制了,我感觉我做数组类的题目只会暴力循环。
运行结果是正确的,但是超时了。
class Solution {
public:
//连续+奇数
int numberOfSubarrays(vector<int>& nums, int k) {
int count=0;//初始数字为0
int wlen=k;//滑动窗口的最小值为k,因为必须要保证子数组中至少有k个数 最大值为nums.size()
//滑动窗口的长度依次增加
while(wlen<=nums.size()){
for(int i=0;i<=nums.size()-wlen;i++){
int temp=0;
for(int j=i;j<i+wlen;j++){
cout<<nums[j]<<" ";
if(nums[j]%2==1){
temp++;//记录当前窗口奇数的数目
}
}
cout<<endl;
if(temp==k){
count++;
}
}
wlen++;
}
return count;
}
};
ac代码
找到第i个奇数,再找到第i+k个奇数,看他们的活动范围。
记录奇数的数组下标
odd数组: -1 ……………… nums.size() 末尾的下标+1
int numberOfSubarrays(vector<int>& nums, int k) {
vector<int> odd;//存放数组奇数的下标
//如果第一个数就为奇数
odd.push_back(-1);
for(int i=0;i<nums.size();i++){
if(nums[i]%2==1){
odd.push_back(i);//存入下标
}
}
odd.push_back(nums.size());//如果数组的最后一个数为奇数
for(int i=0;i<odd.size();i++){
cout<<odd[i]<<" ";
}
//找到第i个奇数,再找到第i+k个奇数,看他们的活动范围 活动范围 (odd[i]-odd[i-1])*(odd[i+k]-odd[i+k-1])
//循环odd数组
int count=0;
for(int i=1;i+k<odd.size();i++){
count+=(odd[i]-odd[i-1])*(odd[i+k]-odd[i+k-1]);
}
return count;
}
和为k的子数组
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
参考上面的思路,可以创建一个新数组,存放以该下标数组值结束的前面数组元素之和(包括自己)=>即前缀和。
绝了!!哭了!!继续写!!
for(int i=0;i<res.size()-1;i++){
for(int j=i+1;j<res.size();j++){
if(res[j]-res[i]==k){
count++;
}
}
}
估计是这里的双重循环出了问题,那就不用数组了
int subarraySum(vector<int>& nums, int k) {
int count=0;
unordered_map<int,int> mp;//哈希表的键值分别为前缀和 及数目
mp[0]=1; //如果前缀和就满足了k
int sum=0;//记录前缀和
for(int i=0;i<nums.size();i++){
sum+=nums[i];
count+=mp[sum-k];
++mp[sum];
}
return count;
}
之前想过把mp全部计算出来,再循环,由于加法的交换律,会重复计算。