洛谷比赛数学题*1

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));
}