牛牛的背包问题

【问题】

牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为w。

牛牛家里一共有n袋零食, 第i袋零食体积为v[i]。

牛牛想知道在总体积不超过背包容量的情况下,他一共有多少种零食放法(总体积为0也算一种放法)。

输入描述:

输入包括两行
第一行为两个正整数n和w(1 <= n <= 30, 1 <= w <= 2 * 10^9),表示零食的数量和背包的容量。
第二行n个正整数v[i](0 <= v[i] <= 10^9),表示每袋零食的体积。

输出描述:

输出一个正整数, 表示牛牛一共有多少种零食放法。

示例1

输入

3 10
1 2 4

输出

8

说明

三种零食总体积小于10,于是每种零食有放入和不放入两种情况,一共有2*2*2 = 8种情况。

【解决】

① 针对每种零食,有两种情况,不添加进背包或者添加进背包。dfs实现。

import java.util.*;

public class Main{
    static int count = 0;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int w = sc.nextInt();
        int[] v = new int[n];
        long sum = 0;
        count = 0;
        for (int i = 0;i < n;i ++){
            v[i] = sc.nextInt();
            sum += v[i];
        }
        if (sum <= w){
            System.out.println((int)Math.pow(2,n));
        }else {
            dfs(0,0,n,v,w);
            System.out.println(count + 1);
        }
    }
    public static void dfs(long cursum,int index,int n,int[] v,int w){
        if (index < n){
            if (cursum > w){
                return;
            }
            dfs(cursum,index + 1,n,v,w);
            if (cursum + v[index] <= w){
                count ++;
                dfs(cursum + v[index],index + 1,n,v,w);
            }
        }
    }
}

猜你喜欢

转载自my.oschina.net/liyurong/blog/1788669