【计算思维作业】C.思维之花-简单背包

最新推荐文章于 2024-10-08 20:32:22 发布

Gusare

最新推荐文章于 2024-10-08 20:32:22 发布

阅读量479

点赞数 6

文章标签：算法 c++

本文链接：https://blog.csdn.net/2301_79620132/article/details/138613506

版权

题目

李老师正准备暑假旅行，他有一个容量为L的行李箱和n个物品（n不超过20），每个物品都有自己的体积，物品可以放入行李箱，但行李箱中物品的总体积不能超过行李箱容量，李老师现在想知道他有多少种携带物品的方案（一个物品都不带也算一种方案）

输入数据

第一行为两个正整数n和L，分别代表物品总数和行李箱容量，n<=20，L<=1e9 接下来一行为n个正整数vi，代表第i个物品的体积，vi<=1e8

输出数据

方案数

样例输入

3 10
2 4 5

样例输出

思路

虽然题目已经提示我们这是简单背包问题，但是以我贫瘠的算法知识水平，我不会01背包的所有方案问题……

所以只能用其他方法了。

仔细思考一会儿，如果用枚举的思路，需要对物品求所有的组合情况，然后对每个组合求和，若和小于等于背包容量，则计数器加一。

所以难点现在变成了求所有组合。以我贫瘠的算法知识，我不会。

求助于gpt，gpt给我了一份代码，我对这份原始代码进行了加工与优化，最后把这道题AC啦。

先看看gpt辅助下我的代码吧。

回顾一下数学知识：

集合中有n个元素

子集数： $2^n$ 非空子集数： $2^n$ $-1$ 非空真子集数： $2^n$ $-2$

代码和解析

#include <iostream>
#include <vector>
#include <numeric> // 包含 accumulate 函数
using namespace std;
int main() {
    int n,l,count;
    cin>>n>>l;
    count=0;
    vector<int> nums(n,0);
    for(int i=0;i<n;i++)
        cin>>nums[i];
    for (int i = 0; i < (1 << n); ++i) { // 从0到2^n-1
        std::vector<int> combination; // 当前组合的容器
        // 构造当前组合
        for (int j = 0; j < n; ++j) {
            if (i & (1 << j)) {//检查第i位是否为1
                combination.push_back(nums[j]);
            }
        }
        int sum = accumulate(combination.begin(), combination.end(), 0);
        if(sum<=l) count++;
    }
    cout<<count<<endl;
    return 0;
}

代码的逻辑和我在“思路”里阐释的一样，求所有组合情况的每个和，与容量比较，得到count值即为所有方案数。

核心和难点就在第二个for：详细剖析一下

1<<n：位运算，值为 $2^n$ ，也就是子集数

i：0，1，2…… $2^n-1$