leetcode:4Sum

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Somnus_k/article/details/82586085

思路:在3Sum基础上多加一个数,先将数组排序,然后先固定两个数i,j,设定两个指针,一个从i+1开始,一个从nums.length-1开始,根据sum和target的大小情况,分别移动两个指针,并将符合的情况保存下来。时间复杂度(O(n3))。

代码:

public class FourSum18 {

	public static void main(String[] args) {

		int[] nums = {1, 0, -1, 0, -2, 2};
		System.out.println(fourSum(nums,0));
		
	}

	public static List<List<Integer>> fourSum(int[] nums, int target) {
		List<List<Integer>> ls = new ArrayList<>();

		Arrays.sort(nums);

		for (int i = 0; i < nums.length - 3; i++) {
			for (int j = i + 1; j < nums.length - 2; j++) {
				int left = j + 1;
				int right = nums.length - 1;
				while (left < right) {
					int sum = nums[i] + nums[j] + nums[left] + nums[right];
					if (sum == target) {
						List<Integer> list = Arrays.asList(nums[i], nums[j], nums[left], nums[right]);
						if (!ls.contains(list))
							ls.add(list);
						left++;
						right--;
					} else if (sum > target) {
						right--;
					} else {
						left++;
					}
				}
			}
		}

		return ls;
	}
}

输出:[[-2, -1, 1, 2], [-2, 0, 0, 2], [-1, 0, 0, 1]]

猜你喜欢

转载自blog.csdn.net/Somnus_k/article/details/82586085