题目地址:
https://www.lintcode.com/problem/aerial-movie/description
一段航线的时间是 t t t,为了让旅客不无聊,飞机提供若干电影,时长为数组 A A A。旅客想挑出两个电影,要求它们的时长之和不大于 t − 30 t-30 t−30,并且时长之和尽可能大。如果有若干个同时满足条件的解,则返回那个较长电影最长的那个组合。最终返回的是长度为 2 2 2的数组,存的是两个电影的时长,按两个电影时长递增排列。
先对 A A A排序,然后对撞双指针,开两个指针 l l l和 r r r分别从数组左右两端开始,如果 A [ l ] + A [ r ] > t − 30 A[l]+A[r]>t-30 A[l]+A[r]>t−30,则左移 r r r,否则说明找到了一组解,看该解是否更优(即和是否更大),如果更优则更新答案,否则不更新,然后右移 l l l。指针相遇后退出循环。代码如下:
public class Solution {
/**
* @param t: the length of the flight
* @param dur: the length of movies
* @return: output the lengths of two movies
*/
public int[] aerialMovie(int t, int[] dur) {
// Write your code here
int[] res = new int[2];
t -= 30;
Arrays.sort(dur);
int l = 0, r = dur.length - 1;
while (l < r) {
if (dur[l] + dur[r] > t) {
r--;
} else {
// 如果找到了更优解,则更新答案
if (dur[l] + dur[r] > res[0] + res[1]) {
res[0] = dur[l];
res[1] = dur[r];
}
l++;
}
}
return res;
}
}
时间复杂度 O ( l A log l A ) O(l_A\log l_A) O(lAloglA),空间 O ( 1 ) O(1) O(1)。