题目链接
https://lydsy.com/JudgeOnline/problem.php?id=3434
题解
枚举选取的第一个点和最后一个点的坐标差
∑ΔP=(Δx1,Δx2,⋯ ,Δxn)(gcd(ΔP)−1c−2)∏k=1n(mi−Δxi+1) \sum_{\Delta P=(\Delta x_1,\Delta x_2,\cdots,\Delta x_n)}\binom{\gcd(\Delta P)-1}{c-2}\prod_{k=1}^n (m_i-\Delta x_i+1) ΔP=(Δx1,Δx2,⋯,Δxn)∑(c−2gcd(ΔP)−1)k=1∏n(mi−Δxi+1)
反演得到
∑T=1mini=1nmig(c,T)∏i=1nf(mi,T) \sum_{T=1}^{\min_{i=1}^n m_i}g(c,T)\prod_{i=1}^n f(m_i,T) T=1∑mini=1nmig(c,T)i=1∏nf(mi,T)
其中
f(m,T)=−⌊mT⌋(⌊mT⌋+1)2T+m⌊mT⌋g(c,T)=∑d∣T(d−1c−2)μ(Td) f(m,T)=-\frac{\lfloor\frac{m}{T}\rfloor(\lfloor\frac{m}{T}\rfloor+1)}{2}T+m\lfloor\frac{m}{T}\rfloor\\ g(c,T)=\sum_{d|T}\binom{d-1}{c-2}\mu(\frac{T}{d}) f(m,T)=−2⌊Tm⌋(⌊Tm⌋+1)T+m⌊Tm⌋g(c,T)=d∣T∑(c−2d−1)μ(dT)
观察发现
∏i=1nf(mi,T) \prod_{i=1}^n f(m_i,T) i=1∏nf(mi,T)
可以看作一个关于TTT的nnn次多项式,每一项的系数最多只有∑i=1nmi\sum_{i=1}^n \sqrt{m_i}∑i=1nmi种,因此可以整除分块分别求系数。假设求出的系数分别为a0,a1,⋯ ,ana_0,a_1,\cdots ,a_na0,a1,⋯,an,那么答案就是
∑i=0naiH(c,i,mini=1nmi) \sum_{i=0}^n a_i H(c,i,\min_{i=1}^n m_i) i=0∑naiH(c,i,i=1minnmi)
其中
H(c,i,k)=∑T=1kg(c,T)Ti H(c,i,k)=\sum_{T=1}^k g(c,T)T^i H(c,i,k)=T=1∑kg(c,T)Ti
可以预处理出HHH。
时间复杂度O(ncm+∑n2∑i=1nmi)O(ncm+\sum n^2\sum_{i=1}^n \sqrt{m_i})O(ncm+∑n2∑i=1nmi)。
代码
#include <cstdio>
#include <algorithm>
int read()
{
int x=0,f=1;
char ch=getchar();
while((ch<'0')||(ch>'9'))
{
if(ch=='-')
{
f=-f;
}
ch=getchar();
}
while((ch>='0')&&(ch<='9'))
{
x=x*10+ch-'0';
ch=getchar();
}
return x*f;
}
const int maxn=11;
const int maxc=20;
const int maxm=100000;
const int mod=10007;
const int inf=0x3f3f3f3f;
int p[maxm+10],prime[maxm+10],cnt,mu[maxm+10],C[maxm+10][maxc+2],g[maxc+2][maxm+10],h[maxc+2][maxn+2][maxm+10],power[maxm+10][maxn+2];
int getprime()
{
p[1]=mu[1]=1;
for(int i=2; i<=maxm; ++i)
{
if(!p[i])
{
prime[++cnt]=i;
mu[i]=mod-1;
}
for(int j=1; (j<=cnt)&&(i*prime[j]<=maxm); ++j)
{
int x=i*prime[j];
p[x]=1;
if(i%prime[j]==0)
{
mu[x]=0;
break;
}
mu[x]=mod-mu[i];
}
}
for(int i=0; i<=maxm; ++i)
{
C[i][0]=1;
for(int j=1; (j<=maxc)&&(j<=i); ++j)
{
C[i][j]=C[i-1][j]+C[i-1][j-1];
if(C[i][j]>=mod)
{
C[i][j]-=mod;
}
}
}
for(int i=1; i<=maxm; ++i)
{
power[i][0]=1;
for(int j=1; j<=maxn; ++j)
{
power[i][j]=power[i][j-1]*i%mod;
}
}
for(int c=2; c<=maxc; ++c)
{
for(int d=1; d<=maxm; ++d)
{
for(int T=d; T<=maxm; T+=d)
{
g[c][T]=(g[c][T]+C[d-1][c-2]*mu[T/d])%mod;
}
}
}
for(int c=2; c<=maxc; ++c)
{
for(int i=0; i<=maxn; ++i)
{
for(int k=1; k<=maxm; ++k)
{
h[c][i][k]=(h[c][i][k-1]+g[c][k]*power[k][i])%mod;
}
}
}
return 0;
}
int T,n,c,m[maxn+2],minm,a[maxn+2];
int main()
{
getprime();
T=read();
while(T--)
{
n=read();
c=read();
minm=inf;
for(int i=1; i<=n; ++i)
{
m[i]=read();
minm=std::min(minm,m[i]);
}
int ans=0;
for(int l=1,r; l<=minm; l=r+1)
{
r=inf;
for(int i=1; i<=n; ++i)
{
r=std::min(r,m[i]/(m[i]/l));
}
for(int i=0; i<=n; ++i)
{
a[i]=(i==0);
}
for(int i=1; i<=n; ++i)
{
int k=m[i]/l,x=1ll*m[i]*k%mod,y=(1ll*k*(k+1)/2)%mod*(mod-1)%mod;
for(int j=n; j; --j)
{
a[j]=(x*a[j]+y*a[j-1])%mod;
}
a[0]=a[0]*m[i]%mod*k%mod;
}
for(int i=0; i<=n; ++i)
{
ans=(ans+a[i]*(h[c][i][r]-h[c][i][l-1]+mod))%mod;
}
}
printf("%d\n",ans);
}
return 0;
}