题目传送门
题目描述:
有个长度为n的序列a,你需要统计a中有多少个棒棒的子序列。
一个序列b被定义为棒棒的,当且仅当:
对于序列中每?个位置i,b_i都能够被i整除。
答案对1e9+7取模
Solution
我们很容易设出初始的dp状态:
d
p
[
i
]
[
j
]
表
示
前
i
个
数
组
成
长
度
为
j
的
序
列
的
方
案
数
dp[i][j] 表示前i个数组成长度为j的序列的方案数
dp[i][j]表示前i个数组成长度为j的序列的方案数
这个方程构造很直观,很容易推出以下式子:
d
p
[
i
]
[
j
]
+
=
d
p
[
i
−
1
]
[
j
−
1
]
∗
(
a
[
i
]
%
j
=
=
0
)
dp[i][j]\ +=\ dp[i-1][j-1]*(a[i]\ \% j==0)
dp[i][j] += dp[i−1][j−1]∗(a[i] %j==0)
考虑优化
对于第一维,发现第
i
i
i位只与第
i
−
1
i-1
i−1位有关,所以我们可以考虑滚动数组
这样的话只是优化了空间。。。
考虑优化时间。
我们继续推式子时能够发现,只有
a
[
i
]
%
j
=
=
0
a[i]\ \%\ j\ ==0
a[i] % j ==0的时候,它才能对答案产生贡献,也就是说其他的就根本不会对答案产生贡献。
对于每个数a[i]我们只要枚举他的因数即可。
枚举因数时间
O
(
n
)
O(\sqrt n)
O(n),枚举
a
[
i
]
a[i]
a[i]时间
O
(
n
)
O(n)
O(n)。
所以总的时间复杂度就是
O
(
n
n
)
O(n\sqrt n)
O(nn)
这个时间复杂度还是对的
这样的话连滚动数组就不用了。
Code
#include<bits/stdc++.h>
using namespace std;
int n,len=0;
const int P = 1e9+7;
int a[1010100];
int b[1010100];
int dp[1010100];
inline bool mycmp(int x,int y){
return x>y;
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
dp[0] = 1;
for (int i=1;i<=n;i++){
len = 0;
for (int j=1;j*j<=a[i];j++)
if (a[i] % j == 0)
j*j!=a[i]?b[++len] = j , b[++len] = a[i]/j : b[++len] = j;
sort(b+1,b+len+1,mycmp);
for (int i=1;i<=len;i++)
dp[b[i]] = (dp[b[i]] + dp[b[i]-1]) %P;
}
int ans=0;
for (int i=1;i<=1000000;i++) ans = (ans+dp[i]) %P;
cout<<ans;
return 0;
}