@LeetCode刷题
文章目录
题目
题目链接:https://leetcode-cn.com/problems/merge-intervals/
一、实现思路
- 将二维数组按照第一列进行排序;
- 逐一判断当前的
start 和 end
是否被包含在上一个中,如果是,合并;不是,添加。 - 因为二维数组已经是有序的数组,所以每次比较均以最后一个为准。
二、总体代码
Java:
class Solution {
public int[][] merge(int[][] intervals) {
if( intervals.length<=1 ){
return intervals;
}
Arrays.sort(intervals, new Comparator<int[]>(){
public int compare(int[] o1, int[] o2){
return o1[0]-o2[0];
}
});
List<int[]> merge = new ArrayList<>();
for( int i=0; i<intervals.length; ++i ){
int L = intervals[i][0], R = intervals[i][1];
if( merge.size()==0 || merge.get(merge.size()-1)[1]<L ){
merge.add(new int[]{L, R});
}else{
merge.get(merge.size()-1)[1] = Math.max(merge.get(merge.size()-1)[1], R);
}
}
return merge.toArray(new int[merge.size()-1][]);
}
}
pyrhon3:
class Solution:
def merge(self, intervals: List[List[int]]) -> List[List[int]]:
intervals.sort(key=lambda x: x[0])
merged = []
for interval in intervals:
if not merged or merged[-1][1]<interval[0]:
merged.append(interval)
else:
merged[-1][1] = max(merged[-1][1], interval[1])
return merged
总结
我认为,重点在于对数组进行按列排序 。
1. java:
//按照第一列排序
Arrays.sort(ts_id,new Comparator<int[]>() {
public int compare(int[] o1, int[] o2) {
return o1[0]-o2[0];
}
});
//当第一列相同时,按照第二列排序
Arrays.sort(ts_id,new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if (o1[0] == o2[0]) {
return o1[1] - o2[1];
}
return o1[0]-o2[0];
}
});
2. python3:
1)sort()只为list使用,可以使用lambda函数
来定义排序的标准
arr = [(1, 2, 3), (3, 2, 1), (2, 4, 6)]
arr.sort(key = lambda a: a[2])
print(arr)
# 在现有arr的基础上再进行一次排序
arr.sort(key = lambda a: a[0])
print(arr)
得到结果如下:
[(3, 2, 1), (1, 2, 3), (2, 4, 6)]
[(1, 2, 3), (2, 4, 6), (3, 2, 1)]
2)sorted可以接收任何的iterable类型
dic = {3:'c', 2:'f', 7:'a'}
print(dic)
# 返回key排序后的值
s1 = sorted(dic)
print(s1)
s2 = sorted(dic, key=lambda s: -s)
print(s2)
得到结果如下:
{3: ‘c’, 2: ‘f’, 7: ‘a’}
[2, 3, 7]
[7, 3, 2]
如果想排序字典的values
,可以将dic
变为dic.values()