小米大礼包

小米大礼包

http://www.nowcoder.com/questionTerminal/7d78d8f671c2461aaeb304efb74b2310

题解

题目难度:中等难度

知识点:查找、数组、动态规划

解析问题:本题可以分析为典型的01背包问题,使用动态规划就可以解决问题。在分析问题时,主要解析为以下三步。
第一步:确定【状态】和【选择】。
在本题中,【状态】就是“剩余的总和”和“可选择的数”,【选择】就是“放入这个数”和“不放入这个数”。
第二步:要明确 dp 数组的定义。
第三步:根据【选择】,对状态转换的逻辑进行计算。

二维动态规划求解

第二步定义数组意义:在这里,使用二维的思维创建dp的形式。设置 dp[i][j] = true 或者 false ,表示前 i 个数的总和是否等于 j 。
第三步分析逻辑:如果不把 nums[i] 算入子集,那么是否能够恰好等于总和,取决于上一个状态 dp[i-1][j]。如果把 nums[i] 算入子集,那么是否能够恰好等于总和,取决于状态 dp[i - 1][j-nums[i-1]]。

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int N,M;
    cin>>N;
    vector<int> Prices;
    for(int i = 0; i< N; ++i)
    {
        int price;
        cin>>price;
        Prices.push_back(price);
    }
    cin>>M;
    //创建dp的状态
    vector<vector<bool>> dp(N+1,vector<bool>(M+1,false));
    //对边缘数据进行设定
    for (int i = 0; i <= N; ++i)
        dp[i][0] = true;

    for (int i = 1; i <= N; ++i) {
        for (int j = 1; j <= M; ++j) {
            if (j - Prices[i - 1] < 0) {
               // 数值过高,不能购买
                dp[i][j] = dp[i - 1][j]; 
            } else {
               // 购买或不购买
                dp[i][j] = dp[i - 1][j] | dp[i - 1][j-Prices[i-1]];
            }
        }
    }
    cout<< dp[N][M]<<endl;
    return 0;
}

一维动态规划求解

为了更好的节约空间上的复杂度,在观察上述的dp[i][j]的等式时,可以总结出dp[i][j]始终是跟dp[i-1][...]成相关性的,因此可以进行一个维度的降低处理。

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int N,M;
    cin>>N;
    vector<int> Prices;
    for(int i = 0; i< N; ++i)
    {
        int price;
        cin>>price;
        Prices.push_back(price);
    }
    cin>>M;
    //创建dp的状态
    vector<bool> dp(M + 1, false);
    // base case
    dp[0] = true;
    for (int i = 0; i < N; i++) 
        for (int j = M; j >= 0; j--) 
            if (j - Prices[i] >= 0) 
                dp[j] = dp[j] || dp[j - Prices[i]];
    cout<< dp[M]<<endl;
    return 0;
}
全部评论

相关推荐

11-18 09:44
Java
小白也想要offer:简历别放洋屁,搞不还还放错了,当然你投外企除外,以上纯属个人观点
点赞 评论 收藏
分享
像好涩一样好学:这公司我也拿过 基本明确周六加班 工资还凑活 另外下次镜头往上点儿
点赞 评论 收藏
分享
评论
点赞
2
分享
牛客网
牛客企业服务