深度优先搜索之部分和(平行状态下的求解)

1. 问题描述:

给定整数序列a1,a2,...,an,判断是否可以从中选出若干数,使它们的和恰好为k.
1≤n≤20
-10^8≤ai≤10^8
-10^8≤k≤10^8
样例:
输入:

n=4
a={1,2,4,7}
k=13
输出:
Yes (13 = 2 + 4 + 7)

 

2. 我们这里使用dfs来解决,可以这样想,我们对于当前的元素可以选择要,也可以选择不要,那么就形成了第一层,然后对于上面形成的两种结果继续执行相同的操作,选择要或者不要,那么就形成了第二层,然后在继续下去直到找到可以找到能够组成目标数字的组合,很明显通过上面的分析我们直到存在着两个平行的状态,分别对应着两棵子树分别是左子树和右子树,对于这两个平行的状态我们需要清楚的直到当他们退回来的时候是什么状态和我们需要如何处理退回来的状态

下面以一个简单的例子说明一下:(结合下面具体的代码来理解)

使用1  4  5  7  3  2  4 中的若干个数字组成目标数字6

     

上面经过搜索之后那么发现4 2符合要求那么它就直接返回了,而不会搜索它的兄弟了

3. 具体的代码如下:

import java.util.ArrayList;
import java.util.Scanner;
public class Mainy{
	static int kk;
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int arr[] = new int[n];
		for(int i = 0; i < n; i++){
			arr[i] = sc.nextInt();
		}
		int k = sc.nextInt();
		kk = k;
		dfs(arr, k, 0, new ArrayList<Integer>());
		sc.close();
	}
	
	private static void dfs(int[] arr, int k, int cur, ArrayList<Integer> ints) {
		if(k == 0){
			System.out.print("Yes (" + kk + " = ");
		    int size = ints.size();
		    for(int i = 0; i < size; i++) {
		        System.out.print(ints.get(i) + (i == size - 1 ? "" : " + "));
		    }
		    System.out.println(")");
		    System.exit(0);
		}
		if(k < 0 || cur == arr.length) return;	
		dfs(arr, k, cur + 1, ints);
		ints.add(arr[cur]);
		dfs(arr, k - arr[cur], cur + 1, ints);
		ints.remove(ints.size() - 1);
	}
}

当退回到dfs(a, k, cur + 1, ints);的时候说明不要当前的元素是不可以的所以需要加入当前的元素然后继续兄弟元素的搜索,当退回到 dfs(a, k - a[cur], cur + 1, ints);说明前面加入的元素不能够组成目标元素所以需要进行移除掉

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值