[WC2014]时空穿梭(莫比乌斯反演)

https://www.cnblogs.com/CQzhangyu/p/7891363.html

不难推到$\sum\limits_{D=1}^{m_1}\sum\limits_{d|D}C_{d-1}^{c-2}\mu(\frac D d)\prod\limits_{i=1}^n\frac {(2m_i-({\lfloor \frac {m_i} {D} \rfloor}+1)\times D){\lfloor \frac {m_i} {D} \rfloor}}{2}$。

$O(Tnm)$,可以拿80甚至100。

我们发现,求和部分与累积部分都含有D,这使分块加速变得困难。

化一下式子发现,当将$\lfloor\frac{m}{D}\rfloor$看作常数时,右边可以化成一个n次多项式。

$O(cm\log m+nmc)$预处理出多项式系数,$O(n\sqrt{m})$整除分块,$O(n^2)$暴力求多项式系数即可。

常数过大BZ被卡。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 5 using namespace std;
 6 
 7 const int N=100010,M=100000,mod=10007,inv2=5004;
 8 bool b[N];
 9 int T,c,n,tot,mn,ans,m[12],p[N],mu[N],C[N][22],s[N][21][12],sj[N][12],f[12],g[N][22];
10 
11 int main(){
12     freopen("space.in","r",stdin);
13     freopen("space.out","w",stdout);
14     mu[1]=1;
15     rep(i,2,M){
16         if (!b[i]) p[++tot]=i,mu[i]=-1;
17         for (int j=1; j<=tot && i*p[j]<=M; j++){
18             b[i*p[j]]=1;
19             if (i%p[j]==0) break;
20             mu[i*p[j]]=-mu[i];
21         }
22     }
23     rep(i,0,M){
24         C[i][0]=1;
25         rep(j,1,min(i,20)) C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
26     }
27     rep(i,1,M){
28         int tmp=1;
29         rep(k,0,11) sj[i][k]=tmp,tmp=1ll*tmp*i%mod;
30     }
31     rep(j,0,20) rep(i,1,M) if (mu[i])
32         for (int k=i; k<=M; k+=i) g[k][j]=(g[k][j]+1ll*mu[i]*C[k/i-1][j]%mod+mod)%mod;
33     rep(i,1,M) rep(j,0,20) rep(k,0,11) s[i][j][k]=(s[i-1][j][k]+1ll*g[i][j]*sj[i][k])%mod;
34     for (scanf("%d",&T); T--; ){
35         scanf("%d%d",&n,&c); mn=N; ans=0;
36         rep(i,1,n) scanf("%d",&m[i]),mn=min(mn,m[i]);
37         for (int i=1,lst; i<=mn; i=lst+1){
38             lst=mn; rep(j,1,n) lst=min(lst,m[j]/(m[j]/i));
39             int tmp=1; rep(j,1,n) tmp=1ll*tmp*(m[j]/i)%mod*inv2%mod;
40             memset(f,0,sizeof(f)); f[0]=1;
41             rep(j,1,n) for (int k=j; ~k; k--) f[k]=(2ll*f[k]*m[j]%mod-1ll*f[k-1]*(m[j]/i+1)%mod+mod)%mod;
42             rep(j,0,n) ans=(ans+1ll*tmp*f[j]%mod*(s[lst][c-2][j]-s[i-1][c-2][j]+mod)%mod)%mod;
43         }
44         printf("%d\n",ans);
45     }
46     return 0;
47 }

 

posted @ 2019-01-14 09:11  HocRiser  阅读(302)  评论(0编辑  收藏  举报