[M差分] lc5806. 描述绘画结果(离散化+差分+双周赛57_3)

Posted 2021-08-27 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：5806. 描述绘画结果

2. 题目解析

很明显的差分，将区间端点离散化即可，做一遍前缀和就行了，注意边界问题。

注意区间 sum=0 时说明其未被涂色，不计入答案。答案要去的是左开右闭区间，还需要注意下差分数组的处理即可。

---

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

---

代码：

我的代码。

思路差不多，关于两边端点的问题，可以使用标记数组处理，即对差分过的端点位置打个标记，最后直接求就行了。大同小异吧。

class Solution {
public:
    vector<vector<long long>> splitPainting(vector<vector<int>>& seg) {
        typedef long long LL;
        vector<LL> diff(1e5 + 5);
        int n = seg.size();
        for (int i = 0; i < n; i ++ ) 
            diff[seg[i][0]] += seg[i][2], diff[seg[i][1]] -= seg[i][2];   // 左闭右开区间
        for (int i = 1; i <= 1e5; i ++ ) diff[i] += diff[i - 1];
        
        vector<int> R;      
        for (int i = 0; i < n; i ++ ) R.push_back(seg[i][0]), R.push_back(seg[i][1]);
        sort(R.begin(), R.end()); R.erase(unique(R.begin(), R.end()), R.end());
        
        
        vector<vector<LL>> res;
        for (int i = 1; i < R.size(); i ++ ) {
            LL l = R[i - 1], r = R[i], sum = diff[l];
            if (l == r || !sum) continue;                                // 有l==r这个判断，也就不必去重了
            res.push_back({l, r, sum});
        }
        
        return res;
    }
};