網易實習2019程式設計題 牛牛揹包

Inequality-Sign發表於2018-03-29

牛牛準備參加學校組織的春遊, 出發前牛牛準備往揹包裡裝入一些零食, 牛牛的揹包容量為w。
牛牛家裡一共有n袋零食, 第i袋零食體積為v[i]。
牛牛想知道在總體積不超過揹包容量的情況下,他一共有多少種零食放法(總體積為0也算一種放法)。

這個題筆試的時候沒有寫出來

import java.util.*;
public class Main{
    public static int  count = 1;
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int nums = scanner.nextInt();
        int w = scanner.nextInt();

        if(nums == 0 || w == 0){
            System.out.println(0);
            return;
        }
        long[] shiwu = new long[nums];

        long sum = 0;
        for (int i = 0; i < nums; i++) {
            shiwu[i] = scanner.nextInt();
            sum += shiwu[i];
        }
        if(sum <= w){
            System.out.println((int)Math.pow(2,nums));
            return;
        }

        helpDfs(shiwu,w,0,0);
        System.out.println(count);

    }
    public static void helpDfs(long[] shiwu,int total,long sum,int cur){
        if(cur < shiwu.length){
            if(sum > total) return;
            if(sum + shiwu[cur] <= total){
                count++;
                helpDfs(shiwu,total,sum+shiwu[cur],cur+1);
            }
            helpDfs(shiwu,total,sum,cur+1);
        }

    }
}

相關文章