题目描述
猴猴最爱吃香蕉了。每天猴猴出门都会摘很多很多的香蕉,每个香蕉都有一个甜度,猴猴不一定要把所有的香蕉都吃掉,猴猴每天都有一个心情值K,猴猴希望当天吃的香蕉满足这么一个条件,这些香蕉的甜度乘积恰好等于K,但是猴猴并不知道有多少种方法,于是猴猴把这个问题交给你。
输入
第一行一个数D,表示有D天。
接下来2*D行,每天的数据有两行
每天的第一行两个数n和K,表示当天摘的香蕉数和心情值
每天的第二行有n个用空格隔开的整数表示香蕉的甜度
(n<=1000,K<=100000000,D<=20)
输出
D行,每行一个数表示方案数,输出对1000000007取模的结果。
样例输入
3
3 3 1 1 3
3 6
2 3 6
3 5
1 5 1
样例输出
4
2
4
提示
对于30%的数据:n,K<=100
对于60%的数据:n<=1000,K<=10000
对于100%的数据:n<=1000,K<=100000000,D<=20
思路
对于每种组合成K的方法,其都是K的约数,因此只需要获得K的约数,枚举甜度和约数,根据转移方程dp[j]=dp[j]+dp[j/i]即可推出答案
代码实现
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6+10;
const ll mod = 1000000007;
unordered_map<int,ll>dp;
int jk[N],ba[N];
int n,k;
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int cnt=0;
dp.clear();
scanf("%d%d",&n,&k);
for(int i=1;i<=sqrt(k);i++)
{
if(k%i==0)
{
jk[cnt++]=i;
if(i*i!=k) jk[cnt++]=k/i;
}
}
sort(jk,jk+cnt);
for(int i=1;i<=n;i++)
{
scanf("%d",&ba[i]);
for(int j=cnt-1;j>=0;j--)
{
if(jk[j]%ba[i]==0)
{
dp[jk[j]]+=dp[jk[j]/ba[i]];
if(jk[j]==ba[i]) dp[jk[j]]++;
dp[jk[j]]%=mod;
}
}
}
printf("%lld\n",dp[k]);
}
return 0;
}