网易校招题之牛牛的背包

题目描述

牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为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种情况。
思路:
如果n个零食的总体积小于背包体积,那么每种零食有放入和不放入两种情况,一共有2^n
如果n个零食的总体积大于背包体积,先排序,依次放入1号、2号。。。m号,直到体积大于背包体积,然后进行下一轮2号。。x号。。。
#include<iostream>
using namespace std;
int methodCount = 1;

void DFS(long long v[], int size, long long w,long long curSum, int pos)
{
    if (curSum <= w)
    {
        ++methodCount;
        for (int i = pos + 1; i < size; ++i)
            DFS(v, size, w, curSum+v[i], i);
    }
    else
        return;
}

int main()
{
    int n;
    long long w;
    long long v[30];
    long long sum = 0;
    cin >> n >> w;
    for (int i = 0; i < n; ++i)
    {
        cin >> v[i];
        sum += v[i];
    }
    //深度优先遍历,访问每个节点的下面的节点是否也符合条件
    //符合条件继续向下,不符合,回到上一层
    if (sum <= w)
        methodCount = 1 << n;
    else//对每个零食体积进行深搜
    {
        sort(v, v + n);
        for (int i = 0; i < n; ++i)
            DFS(v, n, w,v[i],i);
    }
        cout << methodCount << endl;
    return 0;
}




全部评论

相关推荐

点赞 评论 收藏
分享
只写bug的程序媛:才15,我招行20多万,建设银行50多万,说放弃就放弃
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务