zlh秒的题,我根本不会,出题人钦点此题提高-,想来我要爆零了。。。
题面:https://www.luogu.org/problemnew/show/T50940
这题n,a都巨大无比,所以算法要尽量往k身上靠,先推一波式子:
把这里sigma内的东西拆开得到:
第k此项系数C(k,k)肯定是1,消去,再换一下i,j枚举顺序:
这里就出现了和原式很相似的东西,设一个二元函数A:
刚刚那个式子变为:
这样发现每次求A(k,n)都可由A(0~k-1,n-1)推过来!
但这样暴力递归复杂度为k^3,而其实发现我们需要的东西是A(k,n)->A(0~k-1,n-1)->A(0~k-2,n-2)->......
k相同的项只要求n最小的那个往后O(logk)递推下去即可,复杂度k^2logk,常数不要太差即可过此题。
PS:此题据zlh大佬所说当a为1时前面推的就gg了(除数为0),因此要特判这种情况,(为1~n的k次幂之和),这东西想必出题人自己都没想到。。
代码:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=1e9+7;
const int N=2050;
ll a,C[N][N],ans[N][N],mo;
void Dw(ll &x,ll y)
{x=(x-y+mod)%mod;}
void Ad(ll &x,ll y)
{x=(x+y)%mod;}
ll qpow(ll x,ll y)
{
ll res=1;
x%=mod;//注意这里x范围1e18,不加会死的很惨(wa40*3)
while(y)
{
if(y&1)res=res*x%mod;
x=x*x%mod,y>>=1;
}
return res;
}
ll ask(ll n,ll k)
{
ll res;
res=qpow(a,n+1)*qpow(n,k)%mod;
Dw(res,a);
for(int j=0;j<k;j++)
Dw(res,C[k][j]*ans[j][n-1-mo]%mod*a%mod);
res=res*qpow(a-1,mod-2)%mod;
return res;
}
void get(ll l,ll r,ll x)
{
ll tp;
for(ll i=l;i<=r;i++)
{
if(i==l)
{
if(i==0)ans[x][l-mo]=0;
else ans[x][l-mo]=ask(l,x);
tp=qpow(a,i);
}
else
{
tp=tp*a%mod;
ans[x][i-mo]=ans[x][i-1-mo];
Ad(ans[x][i-mo],qpow(i,x)*tp%mod);
}
}
}
int main()
{
ll n,k;
cin>>n>>a>>k;
C[0][0]=1;
for(int i=1;i<=2000;i++)
{
C[i][0]=1;
for(int j=1;j<=2000;j++)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
}
mo=max(0LL,n-k);
for(int i=0;i<k;i++)
get(max(0LL,n-k+i),n-1,i);
printf("%lld\n",ask(n,k));
}