思路:差分数组
难点1:start 和 end都是10的9次方,定义这么大的数组会超时,怎么解决?
解决方案:只维护边界(l和r+1),找某一点的值为:小于等于该值的nums[i]的和。
具体细节:如果只是遍历persons数组,依次得到每个值的话,每次都需要一个二分查找(n==10*10^4),复杂度很高,因此可以先将persons排序,从小到大找(实际不能在persons上面排序,而是只能按照persons由小到大排序,不能修改persons数组),因为找j>i时,前i也算在累加里面,所以可以遍历一遍找到所有ans。
难点 2:怎么实现persons不排序,而且按照persons由小到大访问得到ans呢?
int n = persons.size();
vector<int> v(n); //定义一个n,因为要放到each的位置上而不是push_back
vector<int> ans(n);
iota(v.begin(), v.end(), 0);
sort(v.begin(), v.end(), [&](int i, int j) {return persons[i] < persons[j];});
for (auto& each : v) { //注意each为persons的下标
比persons[each]大
在ans[each]的位置更新
}
代码如下:
class Solution {
public:
vector<int> fullBloomFlowers(vector<vector<int>>& flowers, vector<int>& persons) {
map<int, int> mp;
for (auto& each : flowers) {
mp[each[0]]++;
mp[each[1] + 1]--;
}
int n = persons.size();
vector<int> v(n);
iota(v.begin(), v.end(), 0);
sort(v.begin(), v.end(), [&](int i, int j) {return persons[i] < persons[j];});
int cnt = 0;
vector<int> ans(n);
auto it = mp.begin();
for (auto& each : v) {
while (it != mp.end() && it->first <= persons[each]) {
cnt += it->second;
it++;
}
ans[each] = cnt;
}
return ans;
}
};
代码优化:
1:初始化的一个函数
iota(v.begin(), v.end(), 0); //第一个位置为0,后面依次为前面值+1。即:0 1 2 3...n-1
2:统计到hash里面后,需要按照first升序,并在遍历persons时需要升序遍历pair<>,如果再转换到vector,再排序,麻烦。
做法:直接用map,遍历的时候从 begin()到end()就是升序的。
3:
sort(v.begin(), v.end(), [&](int i, int j) {return persons[i] < persons[j];});
这里的‘&’要写上,捕获列表中使用到函数里的变量隐式为'引用',而不是‘值捕获’,写成'='的话(‘值捕获’)会造成调用大量的拷贝构造函数,特别耗时。