【洛谷 2021.10.6团队模拟赛】巡回的梦魇之神喜欢数列【结论】

---

解题思路

设前缀和为 s ，那么有 $a_i=\frac{k}{i}{s}_{i-1}，s_i=s_{i-1}+a_i=s_{i-1}+\frac{k}{i}{s}_{i-1}=\frac{i+k}{i}{s}_{i-1}$
​
然后不难发现答案

$s_n={\prod }_{i=1}^n\frac{k+i}{i}=C_{n+k-1}^n$

线性预处理逆元和阶乘就好了。

时间复杂度：O(n+T)O(n+T)

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
using namespace std;

const ll mod=1e9+7;
int T;
ll n,k,in[10000010],sum;

ll ksm(ll a,ll b)
{
	ll ans=1,base=a;
	while(b)
	{
		if(b&1)
			ans=(ans*base)%mod;
		base=(base*base)%mod;
		b>>=1;
	}
	return ans;
}

int main(){
	scanf("%d",&T);
	in[1]=1,in[0]=1;
	for(ll i=2;i<=10000000;i++)
	{
		in[i]=in[i-1]*i%mod;
	}
	while(T--)
	{
		scanf("%lld%lld",&n,&k);
		ll x=in[n+k-1]*ksm(in[k-1],mod-2)%mod;
		x=x*ksm(in[n],mod-2)%mod;
		printf("%lld\n",x);
	}
}