51nod 1354：选数字

51nod 1354：选数字

题目链接：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1354

题目大意：有$T(T \leqslant 20)$组数据，每组给出$n(n \leqslant 1000)$个数和$K(K \leqslant 100,000,000)$，问在这$n$个数中选取若干个，积为$K$的方案数有多少.

DP+离散化

与01背包类似，定义状态$dp[i][j]$为前$i$个数中选取若干个数，积为$j$的方案数.

但积过大（$K \leqslant 100,000,000$），考虑到实际有用状态并不多，有用状态为$dp[i][d]$，其中$d$为$K$的因数.

故可以预处理出$K$的因数，将其离散化，从而把第二维降到$D(K)$，$D(K)$表示$K$的因子数.

算法时间复杂度为$O(n \times D(K) \times lg(D(K)))$，$lg$为离散化所带来的时间.

代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#define N 1005
using namespace std;
typedef long long ll;
const ll M=1000000007;
int T,n,K,a[N],k,fact[N],dp[N][N],t,ans;
int main(void){
    scanf("%d",&T);
    while(T--){
        memset(dp,0,sizeof(dp));
        scanf("%d%d",&n,&K);
        for(t=1,k=0;t*t<K;++t)if(K%t==0){
            fact[k++]=t;
            fact[k++]=K/t;
        }if(K==t*t)fact[k++]=t;
        sort(fact,fact+k);
        for(int i=1;i<=n;++i){
            scanf("%d",&a[i]);
            for(int j=0;j<k;++j)if(dp[i-1][j]){
                dp[i][j]=(dp[i][j]+dp[i-1][j])%M;
                if((ll)K%((ll)fact[j]*a[i]))continue;
                t=lower_bound(fact,fact+k,fact[j]*a[i])-fact;
                dp[i][t]=(dp[i][t]+dp[i-1][j])%M;
            }
            if(K%a[i])continue;
            t=lower_bound(fact,fact+k,a[i])-fact;
            dp[i][t]=(dp[i][t]+1)%M;
        }
        printf("%d\n",dp[n][k-1]);
    }
}