去哪儿笔试：部分和问题 java

去哪儿笔试：部分和问题

题目描述

输入一个数组arr和一个目标数字k，问我们能否从数组中找到几个数字，使得它们的和为k
如果能找到，则输出YES和这些数字，否则输出NO

输出的第一行包括两个数，分别表示数组的长度和目标数字k
如果能找到，输出YES，以及这些数字；反之输出NO

输入描述
4 13
1 2 4 7

输出描述
YES
2 4 7

思路：

考查深搜，递归不太好写

数组中的每一个数都存在两种可能，选它或不选它。以第一个数字作为根节点，能够绘制出一棵二叉树，对这可二叉树进行搜索，总能找到一个结点，它的sum = k。

其中，index表示搜索的深度，sum表示累加和

import java.util.Scanner;public class Main {    public static boolean solution(int[] array, int index,             int sum, int k, int[] visit) {        if(sum > k) { // 如果sum 大于k，就不用搜索了            return false;        }        if(index == array.length) { // 如果搜索的深度刚好是数组的长度，那么需要判断sum是否等于k            return sum == k;        }        // 第一种情况，选它        if(solution(array, index + 1, sum, k, visit)) {            visit[index] = 0;  // 0 表示不选            return true;        }        // 第二种情况，不选它                if(solution(array, index + 1, sum + array[index], k, visit)) {            visit[index] = 1; // 1 表示选            return true;        }        return false;    }    public static void main(String[] args) {        Scanner in = new Scanner(System.in);        int n = in.nextInt();        int k = in.nextInt();        int[] array = new int[n];        int[] visit = new int[n];        for(int i = 0; i < n; i++) {            array[i] = in.nextInt();        }        if(solution(array, 0, 0, k, visit)) {            System.out.println("YES");            for(int i = 0; i < n; i++) {                if(visit[i] == 1) {                    System.out.print(array[i] + " ");                }            }            System.out.println();        }else {            System.out.println("NO");        }    }}