LeetCode-三数之和

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

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

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

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

class Solution {
		public List<List<Integer>> threeSum(int[] nums) {
			List<List<Integer>> list = new ArrayList();

			Arrays.sort(nums);
			int a, b, c;

			for (int i = 0; i < nums.length - 2; ) {
				a = nums[i];
				int j = i + 1;
				int k = nums.length - 1;
				while (j < k) {
					b = nums[j];
					c = nums[k];
					if (a + b + c == 0) {
						List<Integer> item = new ArrayList<>();
						item.add(a);
						item.add(b);
						item.add(c);
						list.add(item);
						j++;
						k--;
						while (j < k && nums[j] == nums[j - 1]) {
							j++;
						}
						while (j < k && nums[k + 1] == nums[k]) {
							k--;
						}
					} else if (a + b + c > 0) {
						k--;
						while (j < k && nums[k + 1] == nums[k]) {
							k--;
						}
					} else {
						j++;
						while (j < k && nums[j] == nums[j - 1]) {
							j++;
						}
					}

				}
				++i;
				while (i < nums.length - 2 && nums[i] == nums[i - 1]) {
					i++;
				}

			}

			return list;
		}
	}

猜你喜欢

转载自blog.csdn.net/s695540301/article/details/80301241