51nod 1258 序列求和 V4 拉格朗日插值

题意

T(n) = n^k，S(n) = T(1) + T(2) + …… T(n)。给出n和k，求S(n)。
例如k = 2，n = 5，S(n) = 1^2 + 2^2 + 3^2 + 4^2 + 5^2 = 55。
由于结果很大，输出S(n) Mod 1000000007的结果即可。
第1行：一个数T，表示后面用作输入测试的数的数量。（1 <= T <= 500)
第2 - T + 1行：每行2个数，N, K中间用空格分割。(1 <= N <= 10^18, 1 <= K <= 50000)

分析

像这样的式子我们已经见过：

$$\sum\limits_{i=1}^{n} i^k$$
这是一个k+1的多项式，所以我们直接取k+2个点进行拉格朗日插值就好
原来的时间复杂度是 $$O(K^2)$$
我们预处理一下，也就是对于下式
$$\sum\limits_{k=1}^{n} y_k \prod\limits_{j=1,j\not=k}^{n} \frac{x-x_j}{x_k-x_j}$$
首先看分子，我们可以预处理出前缀和后缀就可以了
对于分母，我们发现如果我们的x取1-n的话，就有：
$$\begin{aligned} (1-2) (1-3) (1-4) \cdots (1-n) \\ (2-1) (2-3) (2-4)\cdots (2-n) \\ \end{aligned}$$
发现错位相等，只是头和尾不一样，就可以O(1) 转移了
加上快速幂时间复杂度是 $$O(KlogK)$$

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

const ll N = 50010;
const ll mod = 1000000007;

inline ll read()
{
  ll p=0; ll f=1; char ch=getchar();
  while(ch<'0' || ch>'9'){if(ch=='-') f=-1; ch=getchar();}
  while(ch>='0' && ch<='9'){p=p*10+ch-'0'; ch=getchar();}
  return p*f;
}

ll qpow(ll x,ll k,ll mo){ll s=1; while(k){if(k&1) s=s*x%mo; x=x*x%mo; k>>=1;} return s;}

ll n,k,p[N],q[N],y[N],inv[N];

ll calc()
{
  // if(n<=k+2) return y[n];
  p[0] = 1; for(ll i=1;i<=k+2;i++) p[i] = p[i-1] * (n-i) % mod;
  q[k+3] = 1; for(ll i=k+2;i>=1;i--) q[i] = q[i+1] * (n-i) % mod;
  ll w=1; for(ll i=2;i<=k+2;i++) w=w*(1-i)%mod; w = qpow(w,mod-2,mod);
  ll ans = 0; for(ll i=1;i<=k+2;i++)
  {
    ans = (ans + y[i] * p[i-1] % mod * q[i+1] % mod * w % mod) % mod;
    w = w * (i-k-2) % mod * inv[i] % mod; 
  }return ans;
}

int main()
{
  ll t=read();
  inv[1] = inv[0] = 1; for(ll i=2;i<=(ll)5e4+5;i++) inv[i] = (mod - mod/i) * inv[mod%i] % mod;
  while(t--)
  {
    n = read(); k = read(); n%=mod;
    for(ll i=1;i<=k+2;i++) y[i] = (y[i-1] + qpow(i,k,mod)) % mod;
    printf("%lld\n",(calc()+mod)%mod);
  }

  return 0;
}