package com.myproject.demo.算法;
import com.sun.tools.javac.util.ArrayUtils;
import org.apache.lucene.util.CollectionUtil;
import org.elasticsearch.common.util.CollectionUtils;
import java.util.*;
/**
* 给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
* <p>
* 注意:答案中不可以包含重复的三元组。
* <p>
* <p>
* <p>
* 示例:
* <p>
* 给定数组 nums = [-1, 0, 1, 2, -1, -4],
* <p>
* 满足要求的三元组集合为:
* [
* [-1, 0, 1],
* [-1, -1, 2]
* ]
*
* @author yuJie
* @since 2020/5/12
*/
public class 三数之和 {
//吊问题思路很简单,但是bugfree还是有很多坑的
/**
* 思路: 先排序nums = [-1, 0, 1, 2, -1, -4]
* 为[-1,-1,0,1,2,4]
* 对于每个节点i 从i+1到 length-1开始遍历 求三者相加=0的集
* 每一次遍历,如果相加<0说明左边的数l小了,需要l++
* 如果相加>0说明右边的数r大了,需要r--
* 如果=0说明是结果,继续执行l++,r--继续走
*
*
*
*
* @param nums
* @return
*/
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>();
Arrays.sort(nums);
if (nums.length <= 0 || nums[0] > 0 || nums[nums.length - 1] < 0) {
return new ArrayList<>();
}
//记录结果去重
Map memory = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
//如果存在nums[i]和nums[i-1]相等就没必要跑了 (去重一)
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int l = i + 1;
int r = nums.length - 1;
while (l < r) {
if (nums[l] + nums[i] + nums[r] == 0) {
//如果已经计算过了直接去重
if (memory.get(nums[l] + "-" + nums[i] + "-" + nums[r]) != null && (Boolean) memory.get(nums[l] + "-" + nums[i] + "-" + nums[r])) {
l++;
r--;
continue;
}
memory.put(nums[l] + "-" + nums[i] + "-" + nums[r], true);
List<Integer> res = new ArrayList<>();
res.add(nums[i]);
res.add(nums[l]);
res.add(nums[r]);
result.add(res);
l++;
r--;
} else if (nums[l] + nums[i] + nums[r] < 0) {
l++;
} else {
r--;
}
}
}
return result;
}
public static void main(String[] args) {
int[] nums = {-2, 0, 0, 2, 2};
System.out.println(threeSum(nums));
}
}