[CF837D] Round Subset（滚动数组，01背包）

题目链接：http://codeforces.com/contest/837/problem/D

题意：n个数里选k个数，使得它们的乘积末尾0个数最多。

只需要统计每个数的2和5的数量，一个作为容量，一个作为价值。f(i,k,j)表示前i个数选k个，一共有j个2的时候，5最多有几个。

外层枚举前i个数，内层做01背包就可以。但是会MLE，所以滚动数组。

特别注意的是，滚动数组在滚动的时候要拷贝整层，原因是我在更新01背包的时候没有及时复制。。。

还有f要注意当前层一定要从之前存在的状态更新过来，否则状态就不对了。顺便，用map写就太麻烦了。。

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 202;
const int maxm = 63*maxn;
int n, K;
int f[2][maxn][maxm];
int w2[maxn], w5[maxn], tot[maxn];

signed main() {
    // freopen("in", "r", stdin);
    LL a;
    while(~scanf("%d%d",&n,&K)) {
        memset(f, -1, sizeof(f));
        memset(tot, 0, sizeof(tot));
        memset(w2, 0, sizeof(w2));
        memset(w5, 0, sizeof(w5));
        for(int i = 1; i <= n; i++) {
            scanf("%lld", &a);
            while(a % 5 == 0) w5[i]++, a /= 5;
            while(a % 2 == 0) w2[i]++, a /= 2;
            tot[i] = tot[i-1] + w2[i];
        }
        f[0][0][0] = 0;
        int ret = 0;
        for(int i = 1; i <= n; i++) {
            for(int k = 1; k <= K; k++) {
                for(int j = tot[i]; j >= w2[i]; j--) {
                    f[1][k][j] = max(f[1][k][j], f[0][k][j]);
                    if(f[0][k-1][j-w2[i]] != -1) {
                        f[1][k][j] = max(f[0][k-1][j-w2[i]]+w5[i], f[1][k][j]);
                    }
                    ret = max(ret, min(j, f[1][k][j]));
                }
            }
            for(int k = 1; k <= K; k++) {
                for(int j = maxm-1; j >= 0; j--) {
                    f[0][k][j] = f[1][k][j];
                }
            }
        }
        printf("%d
", ret);
    }
    return 0;
}