部分和问题(挑战程序设计竞赛)

问题描述:
给定一个整数K和一个整数数列a1,a2,…,an,判断是否可以从中选出若干个数,使得它们的和恰好为K。
样例:
输入:
n=4
a={1,2,4,7}
K=13

输出:
Yes(13 = 2 + 4 + 7)




解法一:
  这题可以套用生成子集的模板,求出这个整数数列的所有子集,去判断每一个子集是否满足条件。我们可以利用二进制求解,将数组中的n个元素对应n位的二进制。其中每一个元素对应一个二进制位。那么从0到2的n次方减一的每一个数就可以对应一个子集。也就是说所有子集一共有2的n次方个。(这里要注意测试数据的范围,要注意所有子集数有没有超出int)

代码如下:

import java.util.*;

public class 部分和 {
	public static void main(String[] args){
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] arr = new int[n];
		for(int i=0; i<n; i++){
			arr[i] = scanner.nextInt();
		}
		int k = scanner.nextInt();
		if(partialSum(arr,arr.length,k)){
			System.out.println("Yes");
		}else{
			System.out.println("No");
		}
	}

	private static boolean partialSum(int[] arr, int n, int k) {
		for(int i=f(n)-1; i>0; i--){//从2的n次方开始遍历
			ArrayList<Integer> res = new ArrayList();
			for(int j=n-1; j>=0; j--){
				if( ( (i>>j)&1 )==1){//判断第j位上是否为1
					res.add(arr[j]);
				}
			}
			int sum = 0;
			for(int j=0; j<res.size(); j++){//进行求和
				sum += res.get(j);
			}
			if(sum ==k){//和等于K,输出所有元素,返回true
				for(int j=0; j<res.size(); j++){
					System.out.print(res.get(j)+"  ");
				}
				System.out.println();
				return true;
			}
		}
		return false;
	}
	
	//计算2的n次方
	private static int f(int n) {
		int sum = 1;
		for(int i=0; i<n; i++){
			sum = sum*2;
		}
		return sum;
	}
}




解法二:
  可以用DFS的方法解决,可以和数独游戏作对比;数独游戏是每一格有九种情况的选择(0~9),而这题对每一元素有两种情况的选择(选和不选)。我们可以用还需要凑的数K和还可以选择的范围(用数组下标表示)来表示当前的情况。

代码如下:


import java.util.*;

public class DFS部分和 {
	public static void main(String[] args){
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] arr = new int[n];
		for(int i=0; i<n; i++){
			arr[i] = scanner.nextInt();
		}
		int k = scanner.nextInt();
		ArrayList<Integer> res = new ArrayList();//用于放选出的数据
		dfs(arr, k, 0, res);
	}

	private static void dfs(int[] arr, int k, int cur, ArrayList<Integer> res) {
		if(k==0){//找到满足条件的组合
			System.out.println("Yes");
			for(int i=0; i<res.size(); i++){
				System.out.print(res.get(i)+ " ");
			}
			System.exit(0);
		}
		if(k<0||cur == arr.length)//走不通,回溯
			return ;
		
		dfs(arr, k, cur+1, res);//不加当前的元素,K不变
		
		res.add(arr[cur]);
		dfs(arr, k-arr[cur], cur+1, res);//加上当前的元素
		res.remove(res.size()-1);//回溯,走不通时把刚加进来的元素删除
	}
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

逍遥自在”

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值