巧用二进制解决挑战程序竞赛中的部分和问题

原问题的描述为:
给定整数序列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的话就代表一个数,这样就可以表示出所有的数组的子集然后进行判断。*
    附上测试截图:
    在这里插入图片描述
    在这里插入图片描述
    别忘了顺手点个赞哦(嘻嘻嘻!)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值