动态规划问题笔记2-买糖果问题

/*
题目描述:
某糖果公司专门生产儿童糖果,它最受儿童欢迎的糖果有A1、A2两个序列,均采用盒式包装。包装好的A1类糖果体积为一个存储单位,而包装好的A2类糖果体积正好是A1类的
两倍。这两类糖果之所以广受儿童欢迎,是因为糖果中含有公司独家研发的魔幻因子。A1或A2序列中的糖果,看起来包装可能是一样的,但因为其中的魔幻因子含量不同被细分
为不同的产品。临近传统节日,公司的糖果供不应求。作为一个精明的糖果分销商,小东希望能够借此大赚一笔,于是带着现金开着货车来公司提货。货车的容量是确定的,小东
希望采购的糖果能够尽可能装满货车,且糖果的魔幻因子总含量最高。只要不超出货车容量,糖果总可以装入货车中。小东希望你能帮她解决这一问题。

输入:
输入中有多组测试数据。每组测试数据的第一行有两个整数n和v,1<=n<=10^5, 1<=v<=10^9,n为可供选购糖果数量,v为货车的容量。随后n行为糖果的具体信息,第一行
编号为1,第二行编号为2,以此类推,最后一行编号为n。每行包含两个整数ti和pi,1<=ti<=2, 1<=pi<=10^4,ti为糖果所属的序列,1为A1、2为A2,pi则是其中的魔
幻因子含量。

输出:
对每组测试数据,先在单独的一行中输出能采购的糖果中的魔幻因子最高含量,之后在单独的行中按编号从小到大的顺序输出以空格分隔的糖果编号,若有多组糖果组合均能满足
要求,输出编号最小的组。若没有糖果能够满足要求,则在第一行中输出0,第二行输出“No”。

如:
输入
    3 2
    1 2
    2 7
    1 3
输出
    7
    2
*/

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int max(const int& a, const int& b)
{
    if (a>b)return a;
    else return b;
}

void swap(int& a, int& b)
{
    int temp = a;
    a = b;
    b = temp;
}

void mysort(vector<int>& v)
{

    for (int i = 0; i < v.size() - 1; i++)
    {
        for (int j =0; j < v.size()-1-i; j++)
        {
            if (v[j]>v[j+1])
                swap(v[j], v[j+1]);
        }
    }
}

int main()
{
    int n, v;
    cin >> n >> v;
    vector<vector<int>> candy(n + 1, vector<int>(2));
    vector<vector<long>> dp(n + 1, vector<long>(v + 1));

    for (int i = 1; i <n+1; i++)
    {
        cin >> candy[i][0] >> candy[i][1];
    }
    for (int i = 0; i<n + 1; i++)
    {
        dp[i][0] = 0;
    }
    for (int i = 0; i<v + 1; i++)
    {
        dp[0][i] = 0;
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = 0; j <= v; j++)
        {
            if (candy[i][0]>j)
            {
                dp[i][j] = dp[i - 1][j];
            }
            else
            {
                dp[i][j] = max(dp[i - 1][j], candy[i][1] + dp[i - 1][j - candy[i][0]]);
            }
        }
    }
    vector<vector<int>> res;
    vector<int> candyNum;
    int temp = v;
    for (int i = n; i>0; i--)
    {
        if (dp[i][temp]>dp[i - 1][temp])
        {
            res.push_back(candy[i]);
            candyNum.push_back(candy[i][0]);
            temp -= candy[i][0];
        }
    }



    /*vector<int>::iterator it = candyNum.begin();
    sort(it, it + candyNum.size());*/
    mysort(candyNum);

    if (candyNum.size() == 0)
    {
        cout << 0 << endl;
        cout << "No" << endl;
    }

    cout << dp[n][v] << endl;
    for (int i = 0; i<candyNum.size(); i++)
    {
        cout <<candyNum[i]<<" ";
    }
    cout << endl;
    system("pause");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/shuoyueqishilove/article/details/82463248