leetcode花期内花的数目(困难,差分数组,周赛)

leetcode花期内花的数目(困难,差分数组,周赛)_第1张图片
leetcode花期内花的数目(困难,差分数组,周赛)_第2张图片
思路:差分数组
难点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];}); 
这里的‘&’要写上,捕获列表中使用到函数里的变量隐式为'引用',而不是‘值捕获’,写成'='的话(‘值捕获’)会造成调用大量的拷贝构造函数,特别耗时。

你可能感兴趣的:(周赛题,周赛)