问题描述:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
分析:
对于这个问题,我用的是set来定义的,因为他是不允许数组重复的,然后判断三个数合为0这个问题我是用三个循环来解决的,定义三个变量i,j,k。让i从零开始,j为i+1;而k 是从数组末尾向前找,当k 碰到j的时候 ,k的这一层循环终止。如果找到了三个数的合为0的话,就将这三个数依次加入到set中,当所有的数都找完了之后,将set在转换为list。
public class ThreeSum {
public static void main(String[] args) {
int[] nums = {-1,0,1,2,1,4};
System.out.println(threeSum(nums));
}
public static List<List<Integer>> threeSum(int[] nums){
Set<List<Integer>> lead = new HashSet<>();
Arrays.sort(nums); //排序
for (int i = 0; i < nums.length - 2; i++) {
for (int j =i + 1; j < nums.length-1; j++) {
for (int k = nums.length-1; k > j; k--) {
if(nums[i] + nums[j] + nums[k] == 0){
List<Integer> e = new ArrayList<>();
e.add(nums[i]);
e.add(nums[j]);
e.add(nums[k]);
lead.add(e);
}
}
}
}
List<List<Integer>> list = new ArrayList<>(lead);
return list;
}
}