给定一个包含 n 个数的整数数组 S,写一个函数在 S 中找到所有使得和为给定整数 target 的四元组 ( a,b,c,d )。
注意事项:
1、四元组 ( a,b,c,d ) 中,需要满足 a <= b <= c <= d
2、答案中不可以包含重复的四元组。
格式:
输入行输入一个有 n 个整数的数组 S 和一个整数 target,最后输出所有满足要求的四元组。
样例输入
S = [ 1,0,-1,0,-2,2 ]
target = 0
样例输出
( -1,0,0,1 )
( -2,-1,1,2 )
( -2,0,0,2 )
该问题的解题思路很简单,就是从一定长度的数组中找出四个元素的组合,并且使得这些组合中的元素的和为给定的值target即可。那么,该问题分解开来就是两个问题。
1、从n中取出m(n>m)的组合问题
2、所有的组合的筛选过滤和排序问题。
第一个问题可以参考我的博客组合问题,该文解决了求从n中取m的组合问题。第一个问题解决了,那么第二个问题就不是问题了。以下求解代码:
package suanfa;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Queue;
import java.util.Set;
public class FourTuple {
public static void main(String[] args) {
int[] s = new int[]{1,0,-1,0,-2,2 };
int target =0;
Queue<String> queue = Combination.compute(s.length, 4);//http://blog.csdn.net/zll441488958/article/details/79271895文章中有该函数的实现。
Iterator<String> ite = getResult(queue,s,target).iterator();
while(ite.hasNext()){
System.out.println(ite.next());
}
}
public static Set<String> getResult(Queue<String> queue, int[] s, int target) {
Set<String> set = new HashSet<String>();
while (!queue.isEmpty()) {
String queueString = queue.poll();
String[] strArr =queueString.substring(1, queueString.length()-1).split(",");
int sum = 0;
int[] tuple = new int[strArr.length];
for (int i = 0; i < strArr.length; i++) {
sum += s[Integer.parseInt(strArr[i]) - 1];
tuple[i] = s[Integer.parseInt(strArr[i]) - 1];
}
if (sum == target) {
Arrays.sort(tuple);
set.add(Arrays.toString(tuple));
}
}
return set;
}
}
结果如下:
[-1, 0, 0, 1]
[-2, -1, 1, 2]
[-2, 0, 0, 2]