nyoj1058 部分和问题 dfs

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_33913037/article/details/78383556

部分和问题

时间限制:1000 ms  |  内存限制:65535 KB
难度:2
描述
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO”
样例输入
4 13
1 2 4 7
样例输出
YES
2 4 7

简单dfs问题 两种策虐 选择当前和不选择当前数
java代码


import java.util.Scanner;

/**
 * 
 */
public class Main1058 {
    static int n, k;
    static int a[]=new int[25], b[]=new int[25];
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        n=scanner.nextInt();
        k=scanner.nextInt();
        for (int i=0;i<n;i++){
            a[i]=scanner.nextInt();
        }
        if (dfs(0,0)){
            System.out.println("YES");
            for(int i = 0; i < n; ++i)
                if(b[i]==1) System.out.printf("%d ", a[i]);
            System.out.printf("\n");
        }else {
            System.out.println("NO");
        }
    }

    private static boolean dfs(int i, int sum) {
        if (i==n){
            return sum==k;
        }
        b[i]=0;
        if (dfs(i+1,sum)){
            return true;
        }
        b[i]=1;
        if (dfs(i+1,sum+a[i])){
            return true;
        }
        return false;
    }
}


展开阅读全文

没有更多推荐了,返回首页