思路:
考虑合法的序列的形态:公差为奇数的等差序列,且公差为奇数
那么我们能得到一个 nlogn 的算法:对于每个长度 n,枚举奇数公差 d,对答案的贡献为 n/d
接下来考虑如何优化,我们令 f(i,j) 表示 st>=i 且 ed<=j 的等差数列数量(公差为奇数)
转移如下:f(1,n) = f(1,n-1) + f(2,n) - f(2,n-1) + (st=1 且 ed=n 的答案)
其中 f(2,n-1) 在 f(1,n-1) 和 f(2,n) 中重复计算,需要减去一次
最后一项的值为 (n-1) 的奇因子个数(显然公差整除 ed-st)
显然,f(i,j) 和 f(i+k,j+k) 是等价的
那么我们定义 f(n) 表示 len<=n 的等差数列数列(公差为奇数)
f(n) = f(n-1) + f(n-1) - f(n-2) + (n-1 的奇因子个数)
其中预处理每个数的因子个数可以 O(n) 解决
细节:如何处理奇因子?
由于任何一个数中的偶数因子不影响答案,那么 n 的答案即 n/2 的答案!
注意:统计答案时记得考虑空集情况
如 f(1) = 2
扩展:如何线性预处理因子个数
令 n = \prod pk^ck
显然 d(n) = \prod (ck+1)
设当前数为 i,质数为 p
由于每个数 ip 是由其最小质因子筛出来的
若 i 和 p 互质,那么 d(ip) = d(i) * d§
否则记录每个数的最小质因子出现次数 cnt
d(ip) = d(i) / (cnt(i)+1) * (cnt(ip)+1)
代码
#include<bits/stdc++.h>
const int maxn=1e7+5;
const int maxt=1e5+5;
int tot,q[maxt],d[maxn],p[maxn/10],cnt[maxn];
bool flg[maxn];
long long f[maxn];
void init(int n)
{
d[1]=1;
for(int i=2;i<=n;++i)
{
if(!flg[i])
p[++tot]=i,cnt[i]=1,d[i]=2;
for(int j=1;j<=tot&&i*p[j]<=n;++j)
{
int now=i*p[j];
flg[now]=1;
if(i%p[j]==0)
{
cnt[now]=cnt[i]+1;
d[now]=d[i]/(cnt[i]+1)*(cnt[now]+1);
break;
}
cnt[now]=1;
d[now]=d[i]*d[p[j]];
}
}
f[0]=1,f[1]=2;
for(int i=2;i<=n;++i)
{
if((i&1)^1)
d[i]=d[i/2];
f[i]=f[i-1]+f[i-1]-f[i-2]+d[i-1];
}
}
int main()
{
int Q,mx;
scanf("%d",&Q);
for(int i=1;i<=Q;++i)
scanf("%d",&q[i]),mx=std::max(mx,q[i]);
init(mx);
for(int i=1;i<=Q;++i)
printf("%lld\n",f[q[i]]);
return 0;
}
来源:zr