【Leetcode】1943. Describe the Painting

题目地址：

https://leetcode.com/problems/describe-the-painting/

给定一个数轴，再给定若干三元数组 [ l , r , c ] [l,r,c] [l,r,c]，代表 [ l , r ] [l,r] [l,r]这一段要涂成颜色 c c c，不同三元数组的 c c c都不同。要求返回一个三元数组的序列，每个三元数组 [ l , r , c ] [l,r,c] [l,r,c]表示最后 [ l , r ] [l,r] [l,r]这一段涂过的所有颜色的和，顺序按从左到右，相同颜色组合的区间要合并（ { 1 , 4 } , { 2 , 3 } {1,4},{2,3} {1,4},{2,3}这两种颜色组合虽然加起来和一样，但算不同的颜色组合）。

相当于要快速将某个线段加上某个值，可以用差分数组来做。要记录不同颜色组合，只需要开一个bool数组，记录一下每次涂色的时候的端点即可。代码如下：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    public List<List<Long>> splitPainting(int[][] segments) {
        List<List<Long>> res = new ArrayList<>();
        int max = 0;
        for (int[] seg : segments) {
            max = Math.max(max, seg[1]);
        }
        
        long[] diff = new long[max + 1];
        boolean[] mark = new boolean[max + 1];
        for (int[] seg : segments) {
            diff[seg[0]] += seg[2];
            diff[seg[1]] -= seg[2];
            // 标记一下当前线段涂了一个不同颜色
            mark[seg[0]] = mark[seg[1]] = true;
        }
        
        // 还原成原数组
        for (int i = 1; i < diff.length; i++) {
            diff[i] += diff[i - 1];
        }
        
        for (int i = 1; i < diff.length - 1; i++) {
            if (mark[i] && diff[i] > 0) {
                int j = i + 1;
                while (j < diff.length - 1 && !mark[j]) {
                    j++;
                }
                
                res.add(new ArrayList<>(Arrays.asList((long) i, (long) j, diff[i])));
                i = j - 1;
            }
        }
        
        return res;
    }
}

时间复杂度 O ( n ) O(n) O(n)， n n n是输入的三元组个数，空间 O ( N ) O(N) O(N)， N N N是最大右端点的值。