欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

算法三数之和 Java

程序员文章站 2022-07-27 20:26:08
给你一个包含 n 个整数的数组?nums,判断?nums?中是否存在三个元素 a,b,c ,使得?a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。注意:答案中不可以包含重复的三元组。给定数组 nums = [-1, 0, 1, 2, -1, -4],满足要求的三元组集合为:[ [-1, 0, 1], [-1, -1, 2]]package algorithm;import java.util.ArrayList;import java.util.Arr...
给你一个包含 n 个整数的数组?nums,判断?nums?中是否存在三个元素 a,b,c ,
使得?a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

package algorithm;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.*;

public class 三数之和 {

	public static void main(String[] args) {
		int[] nums = {-1,0,1,2,-1,-4}; 
		
		//先给数组排序从小到大
		Arrays.sort(nums);
		//排序后是 -4 -1 -1 0 1 2
		
		//创建一个集合存放list
		List<List<Integer>> resurt = new ArrayList<>();
		
		for(int i=0;i<nums.length;i++) {
		
			//当i大于零时后面出现重复的数字就运用不上,
			//就要对i指针所指的数字进行查重排除数组中的重复的数字
			if(i>0&&nums[i]==nums[i-1])continue;
			
			//设置一个数字方便后面进行比较
			int sum = -nums[i];
			
			//现在此指针指向 -1
			int j = i+1;
			//现在此指针指向尾部 2
			int k = nums.length-1;
			
			//循环开始
			while(j<k) {
				//如果此两数相加等于sum就存储这三个数字到list集合
				//并进行j向前移,同时k向后移
				if(nums[j]+nums[k] == sum) {
					List<Integer> list = new ArrayList();
					list.add(nums[i]);
					list.add(nums[j]);
					list.add(nums[k]);
					resurt.add(list);
					j++;k--;
					
					//进行 j 和 k的查重并进行限制j最多移到尾部,k最多移到j的后一位
					while(j<nums.length&&nums[j]==nums[j-1])j++;
					while(k>j&&nums[k]==nums[k+1])k--;
				//当两数相加大于是k-- 使数字变小
				}else if(nums[j]+nums[k]>sum) {
					k--;
				//两数小的情况下 j++ 使数字变大
				}else {
					j++;
				}
			}
		}
		System.out.println(resurt);
	}
}

本文地址:https://blog.csdn.net/kaszxc/article/details/109829786