这是我的思路。。。可惜二分查找的代码写的很不熟练。。。最后还是看了别人的代码
class Solution {
public:
long long find_out(vector<int> v, vector<int> nums, int k) {
int l = 0;
int r = v.size() - 1;
long long cnt = 0;
while(l <= r) {
int l1 = v[l];
while(nums[l1] < k && l1 <= v[r]) {
l1 ++;
}
int t1 = l;
while(v[l] < l1){
l++;
}
int skip1 = l - t1;
cnt += skip1 * (skip1 + 1) / 2;
int r1 = v[r];
while(nums[r1] < k && r1 >= v[l]) {
r1 --;
}
int t2 = r;
while(v[r] > r1){
r --;
}
int skip2 = l - t1;
cnt += skip2 * (skip2 + 1) / 2;
}
return cnt;
}
long long numberOfSubarrays(vector<int>& nums) {
// 思路:统计一个数组中的各个数字的个数,假如说只有一个就是1,2个则看之间有没有更大的数字,3个则是分别看两段之间有没有,这样还能合在一起看一整段有没有
// (类似二分查找!)n个则是,放置一个左指针和一个右指针,对两边独立操作;假说左边碰到比k大的数字了,此时则停下,数一下左指针的左边经过了多少个==k的点,假说是3个,则我们可以得到4*3/2个符合题目条件的字符串
long long res = 0;
unordered_map<int, vector<int>> cnt;
for(int i = 0; i < nums.size(); i ++) {
cnt[nums[i]].push_back(i);
}
for(int i = 0; i <= 9; i ++) {
auto it = cnt.find(i);
// cout << "i: " << i << endl;
if (it != cnt.end()) {
vector<int> tmp;
for(auto x: it->second) {
// cout << x << " ";
tmp.push_back(x);
}
// cout << endl;
long long s = it->second.size();
if(s == 1) res += s;
else {
long long t = find_out(tmp, nums, i);
res += t;
}
}
// cout << endl;
}
return res;
}
};