Algorithm之路十八：4Sum

题目：

给出一个整形数组，和一个target，在数组中找出所有的四元组，四元组中的数字之和为target。

举例：

S = [1, 0, -1, 0, -2, 2], and target = 0.

A solution set is:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

思路：

根据3Sum，得到4Sum的代码并不难。

代码：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class _4Sum {

	public static List<List<Integer>> fourSum(int[] nums, int target)
	{
		int length = nums.length;
		List<List<Integer>> list = new ArrayList<>();
		if(length < 4) return list;
		int i = 0,j = 1;
		Arrays.sort(nums);
		while(i < length -3)
		{
			j = i + 1;
			while(j < length - 2)
			{
				int m = j + 1;
				int n = length - 1;
				while(m < n)
				{
					int count = nums[i] + nums[j] + nums[m] + nums[n];
					if(count == target) list.add(Arrays.asList(nums[i],nums[j],nums[m],nums[n]));
					if(count <= target) while(nums[m++] == nums[m] && m < n);
					if(count >= target) while(nums[n--] == nums[n] && m < n);
				}
				while(nums[j++] == nums[j] && j < length -2);
			}
			while(nums[i++] == nums[i] && i < length -3);
		}
		return list;
    }
	public static void main(String[] args)
	{
		int[] S ={1, 0, -1, 0, -2, 2};
		System.out.println(fourSum(S,1));
	}
}

时间复杂度：

四元组中的第一个数为排好序的数组的第i个数时，第二个数的情况有n - 2 - i种，假设第二个数为Sorted数组的第j个数，则计算第三个数和第四个数的过程需要经过的数组元素的个数为n - j，则总的步骤数为

则时间复杂度O(n^3/6)。

空间复杂度：

临时变量占用的空间为O(1)，返回列表占用的空间由输入数组的解的个数决定。