原问题的描述为:
给定整数序列a1,a2,…,an,判断是否可以从中选出若干数,使它们的和恰好为k.
1≤n≤20
-108≤ai≤108
-108≤k≤108
样例:
输入
n=4
a={1,2,4,7}
k=13
输出:
Yes (13 = 2 + 4 + 7)
我们可以先求出所给数组所有的子集,然后每个子集的元素相加判断相加的和是否等于所给的k,如果相等则返回YES,否则输出NO,因此,我们可以用二进制来巧妙地求出所给数组所有的子集。代码如下:
public class Dfs {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[] a = new int[n];
for (int i = 0; i < a.length; i++) {
a[i]= sc.nextInt();
}
int k = sc.nextInt();int flag=0;
int res= new Dfs().d(a,k,flag);
if(res==0) {
System.out.println("NO");
}
//System.out.println(result);
}
public int d(int[] a,int k,int flag) {
ArrayList<ArrayList<Integer>> list = new ArrayList<>();//创建一个大集合
Arrays.sort(a);//将数组排序
for(int i=(int) Math.pow(2, a.length)-1;i>0;i--) {
ArrayList<Integer> newlist = new ArrayList<>();//创建一个子集合
for(int j=a.length-1;j>=0;j--) {
if(((i>>j)&1)==1) {
newlist.add(a[j]);
}
}
int sum=0;
for(int k1=0;k1<newlist.size();k1++) {
sum+=newlist.get(k1);
}
if(sum==k) {
flag=1;
System.out.print("YES ("+k+" = ");
for(int k1=0;k1<newlist.size()-1;k1++) {
System.out.print(newlist.get(k1)+"+");
}
System.out.println(newlist.get(newlist.size()-1)+")");
break;//找到了就可以结束循环了,避免相同元素时重复输出
}
list.add(newlist);
}
return flag;
}
}
*我们给定一个flag标记,如果flag=1则为YES,否则输出NO,对于二进制求子集可以这么理解:
例如一个数组:arr[ ] ={1,2,3},我们知道它的非空子集为{1}、{2}、{3}、{1,2}、{1,3}、{2,3}、{1,2,3},这个数组的自己个数为7,同理,我们可以发现任意一个给定长度为n的数组,它的非空子集个数为2的n次方-1(原谅我不会打数学符号),那么我们可以用二进制来表示每一个数,如下:
- 0001(1){3}
- 0010(2){2}
- 0011(3){2,3}
- 0100(4){1}
- 0101(5){1,3}
- 0110(6){1,2}
- 0111(7){1,2,3}
我们可以看到每一个数的二进制位如果为1的话就代表一个数,这样就可以表示出所有的数组的子集然后进行判断。*
附上测试截图:
别忘了顺手点个赞哦(嘻嘻嘻!)