15.threeSum

版权声明:非商业性使用,禁止演绎,转载请保留原文链接及作者 https://blog.csdn.net/qq_39478237/article/details/82861686

问题

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

Example:

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

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

解决方案


 public List<List<Integer>> threeSum(int[] nums) {
		List<List<Integer>> result = new LinkedList<>();
		
		if (nums != null &&nums.length >2) {
			
			//首先对数组进行排序
			Arrays.sort(nums);
			//我们假设取第 i 个数为结果
			for(int i= 0;i<nums.length-2;)
			{
				//第二个数的可能起始位置
				int j = i+1;
				//第三个数的可能起始位置
				int k = nums.length - 1;
				
				while (j < k) {
					if (nums[j] + nums[k] == -nums[i]) {
						//当我们找到满足条件的解时,我们将结果添加到结果集里
						List<Integer> list = new ArrayList<>(3);
						
						list.add(nums[i]);
						list.add(nums[j]);
						list.add(nums[k]);
						result.add(list);
						//移动位置寻找下一组解
						k--;
						j++;
						
						//下面两个while是用来判断是否越界和判断移动后的数是否和之前的数相同,若相同则跳过该数
						while (j<k && nums[j] == nums[j-1]) {
							j++;
						}
						while (j<k&&nums[k] == nums[k+1]) {
							k--;
						}
					}
					//和大于 0,让k向前移动到小一点的数
					else if (nums[j]+nums[k]>-nums[i]) {
						k--;
						//下面是用来判断是否越界和判断移动后的数是否和之前的数相同,若相同则跳过该数
						while (j<k&&nums[k] == nums[k+1]) {
							k--;
						}
					}
					//和小于 0,让j向后移动到大一点的数
					else {
						j++;
						//下面是用来判断是否越界和判断移动后的数是否和之前的数相同,若相同则跳过该数
						while (j<k&&nums[j] == nums[j-1]) {
							j++;
						}
					}
				}
				//指向下一个要处理的数
				i++;
				//从左向右找第一个与之前处理的数的不同的数的下标
				while (i<nums.length-2&&nums[i] == nums[i-1]) {
					i++;
				}
			}
		}
		  
		  return result;
	        
	   }
展开阅读全文

没有更多推荐了,返回首页