题目地址:
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是最大右端点的值。