题目链接:LeetCode 18—4Sum
题目大意是求无序数组中4个数的和等于目标数,并列出所有可能。
可采用递归思路在列表中固定一个数,相当于缩小了问题规模,将问题变成3个数的和等于目标数……. 采用这种递归的思路即可解除任意数目的数的和等于目标数的问题,实现代码如下:
public class Problem18 {
public static void main(String[] args) {
// TODO Auto-generated method stub
}
private List<List<Integer>> nSum(int[] nums, int start, int n, int target){
List<List<Integer>> result = new LinkedList<>();
// 去掉过大或过小的目标值的情况
if (target < nums[start]*n || target > nums[nums.length - 1]*n){
return result;
}
for (int i = start; i <= nums.length - n; ++i){
// 在这里剪枝,去掉重复
if (i > start && nums[i - 1] == nums[i]){
continue;
}
if (n == 1){
//二分查找
if (target > nums[i]){
continue;
}
if (target < nums[i]){
break;
}
result.add(new LinkedList<Integer>(Arrays.asList(target)));
break;
}
for (List<Integer> list : nSum(nums, i + 1, n - 1, target - nums[i])){
list.add(nums[i]);
result.add(list);
}
}
return result;
}
public List<List<Integer>> fourSum(int[] nums, int target) {
if (nums.length == 0){
return Arrays.asList();
}
Arrays.sort(nums);
return nSum(nums, 0, 4, target);
}
}