AcWing 4801 选数(二维费用背包的建立)

最新推荐文章于 2024-11-02 20:20:37 发布

Rw-chen

最新推荐文章于 2024-11-02 20:20:37 发布

阅读量167

点赞数 1

文章标签： c++ acm竞赛 cpp

本文链接：https://blog.csdn.net/weixin_46211066/article/details/121599309

版权

AcWing 4180.选数字
问题描述：

给定 n 个整数 $a_1,a_2,…,a_n$ 。

请你从中选取恰好 k 个数，要求选出的数的乘积的末尾 0 的数量尽可能多。

请输出末尾 0 的最大可能数量。

输入格式第一行包含两个整数 $n, k 。$

第二行包含 n 个整数 $a_1,a_2,…,a_n$ 。

输出格式一个整数，表示末尾 0 的最大可能数量。

数据范围前 6 个测试点满足 $1 \leq n, k \leq 10$ 所有测试点满足 $1≤n≤200，1≤k≤n，1≤a_i≤10^{18}$

样例
在这里插入图片描述

#include <iostream>
#include <cstring>
using namespace std;

//末位零个数都 是2^a * 5^a 所构造出来的
//log5(1e18) = 25; //更小所以选他做重量限制
//log2(1e18) = 59; //更大当作价值
// f[i][j][k] 前i个物品种，体积不超过j(直接让每个数字的体积为1), 重量不超过k的最大价值

//fly 第一次听懂难题，我再酝酿酝酿
const int N = 210, M = N * 25;
//压低一维 不然炸空间
int f[N][M]; //体积 重量 价值
int v[N], w[N]; //重量 价值
int n, m;

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; ++i)
    {
        long long num;
        cin >> num;
        while(num % 5 == 0) num /= 5, ++v[i];
        while(num % 2 == 0) num /= 2, ++w[i];
    }
    
    //恰好背包的初始化
    memset(f, -0x3f, sizeof f);
    f[0][0] = 0;
    
    for(int i = 1; i <= n; ++i) //前n个数字
        for(int j = m; j > 0; --j)  //体积不超过m(选m个数字)
            for(int k = i * 25; k >= v[i]; --k) //重量不超过m * 25, 改成递增是为了增加速度，因为一层最多25个(5个数)
                f[j][k] = max(f[j][k], f[j-1][k - v[i]] + w[i]);  //选还是不选
                
    int ans = 0;
    for(int i = 1; i <= m * 25; ++i)
        ans = max(ans, min(i, f[m][i])); // min(5, 2);
        
    cout << ans << endl;
    
    return 0;
}