题目地址:
https://www.lintcode.com/problem/number-of-airplanes-in-the-sky/description
给定若干个长度为 2 2 2的数组,表示每个飞机的起降时间。问同一时间天空最多有多少架飞机。如果同一时间有多架飞机起降,要求降落优先。
思路是扫描线。先开一个pair类,存两个变量,一个存时间,另一个存”起还是降“。然后将所有时间点封装成这个类,加入一个列表后排序。接着遍历这些pair,如果是起,计数就加一,否则就减一,同时更新最大同时在天空中的飞机数量即可。注意,由于如果同一个时间有多个飞机同时起降,要把降落的pair排在前面,这样遍历到这个时间点的时候,会先把飞着的飞机减去,再加上新起飞的飞机。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
// time表示起降时间,flag表示起还是降,起取1,降取0
class Pair {
int time, flag;
public Pair(int time, int flag) {
this.time = time;
this.flag = flag;
}
}
/**
* @param airplanes: An interval array
* @return: Count of airplanes are in the sky.
*/
public int countOfAirplanes(List<Interval> airplanes) {
// write your code here
List<Pair> list = new ArrayList<>();
// 将所有起降时间加入list
for (Interval interval : airplanes) {
list.add(new Pair(interval.start, 1));
list.add(new Pair(interval.end, 0));
}
// 按照时间排序,如果时间相同,降者优先
list.sort((p1, p2) -> p1.time != p2.time ? Integer.compare(p1.time, p2.time) : Integer.compare(p1.flag, p2.flag));
// count统计当前有多少架,res记录最多有多少架
int res = 0, count = 0;
for (Pair pair : list) {
if (pair.flag == 1) {
count++;
} else {
count--;
}
res = Math.max(res, count);
}
return res;
}
}
class Interval {
int start, end;
public Interval(int start, int end) {
this.start = start;
this.end = end;
}
}
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间 O ( n ) O(n) O(n)。