心情
今天又在简书上被安利了一部剧《大明王朝1566》
这才是可怕的历史,陈芊芊遇到这部剧,真活不过3集。
读题
leetcode: 15. 三数之和
描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
测试用例:
输入:[-1, 0, 1, 2, -1, -4]
输出:[ [-1, 0, 1], [-1, -1, 2]]
答案中不可以包含重复的三元组。
乍一想,三层for循环?仔细一想,三层for循环……
思路
前提:数组先排过序,这样可以控制循环的次数
1:第一层循环,将第一个数固定
2:第二层循环,第二个数比第一个数大
3:第三层循环,找到唯一的解能满足三数之和为0
小技巧:第一个数固定的时候,第二个和第三个可以分别从两端向中间靠拢,直到两个数相遇。这样就算是3层循环,也是O(n2)的时间复杂度
实现
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>();
//对数组升序排序
Arrays.sort(nums);
//用于标记最后一个数,j++的时候,从index处k--比较迅速
int index,temp;
for (int i = 0; i < nums.length - 2 && nums[i] <= 0; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
index = nums.length - 1;
//j从i+1到index,若nums[i] + nums[j] + nums[index] < 0则让j++
for (int j = i + 1; j < index && (temp = nums[i] + nums[j]) <= 0; j++) {
if (j > i + 1 && nums[j] == nums[j - 1] || temp + nums[index] < 0) {
continue;
}
//k从index到j,当nums[i] + nums[j] + nums[k] >= 0则让k--
for (int k = index; k > j && (temp = nums[i] + nums[j] + nums[k]) >= 0; k--) {
if (temp == 0) {
result.add(Arrays.asList(nums[i], nums[j], nums[k]));
//标记当i固定时,k的最大下标,等待j再一次++
index = k;
break;
}
}
}
}
return result;
}
为了让代码尽可能地行数少,写的比较冗长。只要注释足够就不影响理解。
提交
这已经是我优化了很多次的结果了。去掉了Set集合(用了相邻数不重复),去掉了存放结果的list(用了Arrays.asList())。执行效率也提升了很多,就是名次没有提升……
第一次通过的代码如下:
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>();
List<Integer> element;
Set<Integer> trimSet;
//对数组升序排序
Arrays.sort(nums);
//先找出所有的解
int temp = 0,index;
//锚定第一个数
for (int i = 0; i < nums.length - 2; i++) {
if(nums[i] >0){
break;
}else if(i>0 && nums[i]==nums[i-1]){
continue;
}
trimSet = new HashSet<>();
index = nums.length - 1;
//第二个数从i+1开始
for (int j = i + 1; j < nums.length - 1; j++) {
temp = nums[i] + nums[j];
if(temp > 0){
break;
}else if(!trimSet.add(nums[j])){
continue;
}
element = new ArrayList<>();
element.add(nums[i]);
element.add(nums[j]);
//第三个数从后向前遍历
for (int k = index; k > j; k--) {
temp = nums[i] + nums[j] + nums[k];
if (temp == 0) {
element.add(nums[k]);
result.add(element);
index = k;
break;
}else if(temp < 0){
break;
}
}
}
}
return result;
}
标答
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> lists = new ArrayList<>();
//排序
Arrays.sort(nums);
//双指针
int len = nums.length;
for(int i = 0;i < len;++i) {
if(nums[i] > 0) return lists;
if(i > 0 && nums[i] == nums[i-1]) continue;
int curr = nums[i];
int L = i+1, R = len-1;
while (L < R) {
int tmp = curr + nums[L] + nums[R];
if(tmp == 0) {
List<Integer> list = new ArrayList<>();
list.add(curr);
list.add(nums[L]);
list.add(nums[R]);
lists.add(list);
//两个指针向中间靠拢
while(L < R && nums[L+1] == nums[L]) ++L;
while (L < R && nums[R-1] == nums[R]) --R;
++L;
--R;
} else if(tmp < 0) {
++L;
} else {
--R;
}
}
}
return lists;
}
标答取自评论区某大神
感觉差不多的思路啊,咋就差了呢?