cf779D(记忆化dp)

题目链接: http://codeforces.com/problemset/problem/799/D

 

题意: 给出两个矩阵边长 a, b, 和 w, h, 以及一个 c 数组, 可选择 c 数组中任意数字乘上w 或 h. 数组中每个数字最多只能用一次. 求最少选择多少个数字可使得边长为 a, b 的矩阵能放到变化后的矩阵中.

 

思路: log2(1e5) = 17, 即最多需要对一条边乘17个数字, 要是完全暴力的话需要 2^34 的时间复杂度, 显然不行.

本题 dp 可解, 先给 c 降序排列, 用 dp[i][j] = k 表示到第 i 个元素, 其中一条边为 j 时另外一条边最大为 k.

那么动态转移方程式为: dp[i][j] = max(dp[i - 1][j / c[i]], dp[i - 1][j] * c[i])

但是用普通 dp 的话会有两个问题, 其一是 j / c[i] 整除问题, 这个可以先判断一下能否整除解决. 其二是 w * c[i] 非常大, 如:

100000 100000 99999 99999 2
30000 30000

设定  0 < j <= 1e5 显然不够, 要是 j 太大则会 tle. 虽然不一定没有解决的方法, 但是显然可以不用这么麻烦, 将其改成记忆化dp即可.

 

代码:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define ll long long
using namespace std;

const int MAXN = 4e5 + 10;
ll c[MAXN], dp[40][MAXN];
int n, a, b;

bool cmp(ll x, ll y){
    return x > y;
}

int dfs(int indx, ll w, ll h){
    if(w >= a && h >= b || w >= b && h >= a) return indx;
    if(indx > 34 || indx >= n) return MAXN;
    if(w >= 1e5) w = 1e5; //注意越界
    if(h >= 1e5) h = 1e5;
    if(dp[indx][w] != -1) return dp[indx][w];
    return dp[indx][w] = min(dfs(indx + 1, w * c[indx], h), dfs(indx + 1, w, h * c[indx]));
}

int main(void){
    int h, w;
    scanf("%d%d%d%d%d", &a, &b, &h, &w, &n);
    for(int i = 0; i < n; i++){
        scanf("%lld", &c[i]);
    }
    sort(c, c + n, cmp);
    memset(dp, -1, sizeof(dp));
    int sol = dfs(0, w, h);
    if(sol != MAXN) cout << sol << endl;
    else cout << -1 << endl;
    return 0;
}