[编程题] 小易喜欢的数列
时间限制:1秒
空间限制:32768K
小易非常喜欢拥有以下性质的数列:
1、数列的长度为n
2、数列中的每个数都在1到k之间(包括1和k)
3、对于位置相邻的两个数A和B(A在B前),都满足(A <= B)或(A mod B != 0)(满足其一即可)
例如,当n = 4, k = 7
那么{1,7,7,2},它的长度是4,所有数字也在1到7范围内,并且满足第三条性质,所以小易是喜欢这个数列的
但是小易不喜欢{4,4,4,2}这个数列。小易给出n和k,希望你能帮他求出有多少个是他会喜欢的数列。
1、数列的长度为n
2、数列中的每个数都在1到k之间(包括1和k)
3、对于位置相邻的两个数A和B(A在B前),都满足(A <= B)或(A mod B != 0)(满足其一即可)
例如,当n = 4, k = 7
那么{1,7,7,2},它的长度是4,所有数字也在1到7范围内,并且满足第三条性质,所以小易是喜欢这个数列的
但是小易不喜欢{4,4,4,2}这个数列。小易给出n和k,希望你能帮他求出有多少个是他会喜欢的数列。
输入描述:
输入包括两个整数n和k(1 ≤ n ≤ 10, 1 ≤ k ≤ 10^5)
输出描述:
输出一个整数,即满足要求的数列个数,因为答案可能很大,输出对1,000,000,007取模的结果。
输入例子1:
2 2
输出例子1:
3毫无疑问,dp。
dp[i][j] 表示当数列个数为 i 个时,以 j 结尾有多少个符合要求的数列。
dp[i][j] = dp[i-1][k] 的和( k<=j | | k%j!=0 )
这样时间复杂度是O(n*k^2) ,超时。
优化,既然 dp[i-1][k] 这个要累加的次数比较多,那么我们就把他的和在上一次求解中先求出来,然后进行累减就行了。
优化后,dp[i][j] = pre_i_total - dp[i-1][k] (k>j&&k%j==0)
时间复杂度O(n*k*logk)
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
ll dp[15][100005];
const ll mod=1e9+7;
int main()
{
ll n,k;
while(scanf("%lld%lld",&n,&k)==2)
{
memset(dp,0,sizeof(dp));
ll pre_i_total=0,nex_i_total=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=k;j++)
{
ll ad=pre_i_total;
for(int u=2;u*j<=k;u++)
ad=dp[i-1][u*j]>ad?ad+mod-dp[i-1][u*j]:ad-dp[i-1][u*j];
if(i==1) dp[i][j]=1;
else dp[i][j]=ad%mod;
nex_i_total=(nex_i_total+dp[i][j])%mod;
}
pre_i_total=nex_i_total;
nex_i_total=0;
}
ll ans=0;
for(int u=1;u<=k;u++) ans=(ans+dp[n][u])%mod;
printf("%lld\n",ans);
}
return 0;
}