求子串满足某一条件的问题一般可以转化为先求以第i(0 < i <=n)个元素结尾满足该条件的解,然后将所有解加起来即可得到原问题的解。正是因为我们将以每个元素作为结尾来定义解,所以可以包含所有解,并且不会重复。
例子
求一个数组中连续子数组中任何数字出现次数不大于k的所有子数组
1 2 2 4
1
答案是6
1、2、2、4、1 2、2 4
要求出所有满足条件的子数组,只要求出以nums[i]结尾的满足条件的所有数组长度的和。用上面的例子说明一下:
当 i = 0 时以1结尾的满足条件的数组长度是1,满足条件的子数组是[1]
当 i = 1 时以2结尾的满足条件的数组长度是2 即[1, 2],满足条件的子数组是[2]、[1, 2]
当 i = 2 时以2结尾的满足条件的数组长度是1 即[2],满足条件的子数组是[2](这个2和i=1时的2不是一个2,没有重复)
当 i = 3 时以4结尾的满足条件的数组长度是2 即[2,4],满足条件的子数组是[4]、[2, 4]
实现代码
#include <iostream>
#include <unordered_map>
#include <vector>
#include <algorithm>
using namespace std;
int main(int argc, char const