【题目泛做】宝藏（期望）（类欧几里得）

其实是CF868G的加强版

把概率变成任意了（好吧也没加强什么东西）做法还是一样。

注意，按理说应该要特判$p=1$的情况，然而没有卡。。。

题解：

首先确定策略，我们的策略显然是每次贪心找到当前已经被访问次数最少的洞穴。

那么每次就是访问$(i-1)\ast k+1\to i\ast k$在$\%n$意义下的所有洞穴。

设$e_i$表示宝藏在$i$号洞穴时的期望探测步数，则答案为$\frac{\sum e_i}{n}$

那么$p=1$的时候的答案为（设$t=\lfloor \frac{n-1}{k}\rfloor$）：
$$\sum _{i=1}^n\lfloor \frac{i-1}{k}+1\rfloor =n+k\cdot t\cdot (t-1)/2+(n-t\cdot k)\cdot t$$

然后乘上$n$的逆元就行了

其实$e_i$有表达式，我们考虑令前$j-1$次访问这个山洞都没有找到宝藏，且在第$j$次找到：

$$e_i=\sum _{j=1}^{\infty }(1-p{)}^{j-1}p\lfloor \frac{i-1+n\cdot (j-1)}{k}+1\rfloor$$

这个并不能用类欧几里得做，因为上标是$\infty$

直接对于$i=1,\cdots ,n$求和也很麻烦。

但是注意到这给了我们$e_i$之间的递推关系。

对于$k<i\le n$，$e_i=e_{i-k}+1$

对于$1\le i\le k$，$e_i=(1-p)e_{i-k+n}+1$

我们发现，$\%gcd(n,k)$后相同的位置形成了一个环，并且每个环上的方程是一样的，也就是说，我们把$n,k$同时除掉gcd答案不会改变。

于是我们可以利用一次函数复合得到关于$e_1$的方程，然后顺着解下去，复杂度$O(n)$。

考虑一个更加优秀的做法，我们先把$e_i$变成关于$i$的函数$f(i)$（为了写起来方便）。

显然$f$在$i=1,2\cdots ,n$的取值形成的数列就是$e_i$。

现在给定$n,k$，并且在$i=1,2,\cdots n$中，以$k$为分界线函数值有一次函数复合关系。

则考虑构造一次函数$A(x),B(x)$，使得$f(i)=B(f(i-k+n))=A(f(i-k))$

然后考虑迭代，显然我们需要对于前后分开考虑，设置一次函数$S1,S2$，令我们当前需要求的是：

$$\sum _{i=1}^{k}S1(f(i))+\sum _{i=k+1}^{n}S2(f(i))$$

考虑迭代，设$n^{\prime }=k,k^{\prime }=n\%k$，那么我们需要考虑构造$A^{\prime },B^{\prime },S{1}^{\prime },S{2}^{\prime }$使得迭代出来的形式和上面相同，及需要找到合法的这四个一次函数，使得：

$$\begin{gathered} \sum _{i=1}^{k}S1(f(i))+\sum _{i=k+1}^{n}S2(f(i))=\sum _{i=1}^{k^{\prime }}S{1}^{\prime }(f(i))+\sum _{i=k^{\prime }+1}^{n^{\prime }}S{2}^{\prime }(f(i)) \\ f(i)=B^{\prime }(f(i-k^{\prime }+n^{\prime }))=A^{\prime }(f(i-k^{\prime })) \end{gathered}$$

接下来声明两个个记号：

对于函数$F$，以下如无特殊说明$F^{-1}$表示$F$的复合逆，即$F(F^{-1}(x))=F^{-1}(F(x))=I(x)$，其中$I$表示复合操作的单位元，即$I(x)=x$

对于函数$A$，令$A^0=I,A^1=A,A^i=A(A^{i-1})$，即自我复合$i$次。

很显然地，我们可以通过辗转相除直接得到$A^{\prime }$和$B^{\prime }$的表达式：
$$\begin{gathered} A^{\prime }=A^{-\lfloor \frac{n}{k}-1\rfloor }(B^{-1}) \\ {B}^{\prime }=A^{-\lfloor \frac{n}{k}\rfloor }(B^{-1}) \end{gathered}$$

同样地，考虑$i=n^{\prime }+1,n^{\prime }+2\cdots ,n$最终落在$[1,n^{\prime }]$中的哪个位置上，我们同样可以得到$$\begin{gathered} S{1}^{\prime }=S1+\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }S2(A^i) \\ S{2}^{\prime }=S1+\sum _{i=1}^{\lfloor \frac{n}{k}-1\rfloor }S2(A^i) \end{gathered}$$

发现逆元，复合的$i$次幂都很好计算。

但是这个东西相当扯淡：$$\sum _{i=1}^{t}S2(A^i)$$

直接计算是没有办法搞的，但是我们可以算这个东西：
$$S2(\sum _{i=1}^t{A}^i)$$

但是实际上这俩不是同一个东西，我们虽然明确了这里用到的所有函数，除了$f$全部都是一次函数，但是我们仔细考虑一下，上面两个东西在算$S2$的常数项被算了几次的时候就出事了，下面那个只算了一次，上面那个算了$t$次，但是其他地方都work得很好。

所以对于所有的$S$，我们需要把常数项单独提取出来算，才能保证答案正确。

递归边界$k=0$，这时候$f(i)=A(f(i))，$大力计算即可。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define gc get_char
#define cs const

using std::cerr;
using std::cout;

cs int mod=1e9+7;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
inline void ex_gcd(int a,int b,int &x,int &y){
	if(!b){x=1,y=0;return ;}ex_gcd(b,a%b,y,x);y-=a/b*x;
}
inline int inv(int a){
	assert(a);int x,y;
	ex_gcd(a,mod,x,y);
	return (x%mod+mod)%mod;
}

struct func{
	int k,b;
	func(){}
	func(int _k,int _b):k(_k),b(_b){}
	friend func operator+(cs func &a,cs func &b){return func(add(a.k,b.k),add(a.b,b.b));}
	friend func operator-(cs func &a,cs func &b){return func(dec(a.k,b.k),dec(a.b,b.b));}
	friend func operator*(cs func &a,int b){return func(mul(a.k,b),mul(a.b,b));}
	inline int operator()(int x)cs{return add(mul(x,k),b);}
	inline func operator()(cs func &r)cs{return func(mul(k,r.k),add(mul(k,r.b),b));}
};

inline func inv(cs func &a){
	assert(a.k);int iv=inv(a.k);
	return func(iv,mul(dec(0,a.b),iv));
}

inline func pow_comp(func a,int b){
	func res(1,0);
	for(;b;b>>=1,a=a(a))if(b&1)res=res(a);
	return res;
}

inline func pow_sum(cs func &a,int n){
	int k=a.k,b=a.b;if(k==1)return func(n,mul((ll)n*(n+1)/2%mod,b));
	int iv=inv(dec(1,k)),k1=mul(mul(k,iv),dec(1,power(k,n)));
	return func(k1,mul(mul(b,iv),dec(n,k1)));
}

inline int calc(int n,int k,cs func &a,cs func &b,int s1,int s2){
	if(!k)return mul(mul(n,mul(a.b,s2)),inv(dec(1,a.k)));
	int p=n/k,t=n%k;
	func S2=pow_sum(a,p-1)*s2;Inc(S2.k,s1);
	func S1=S2+pow_comp(a,p)*s2;
	func ia=inv(a),ib=inv(b);
	func A=pow_comp(ia,p-1)(ib),B=ia(A);
	int tp=add(mul(S1.b,t),mul(S2.b,k-t));
	return add(tp,calc(k,t,A,B,S1.k,S2.k));
}

signed main(){
#ifdef zxyoi
	freopen("treasure.in","r",stdin);
#endif
	int T;scanf("%d",&T);
	while(T--){
		int n,k,p;scanf("%d%d%d",&n,&k,&p);
		if(p==1){
			int ans=n--;int t=n/k;
			Inc(ans,mul(k,(ll)t*(t-1)/2%mod));
			Inc(ans,mul(n-t*k+1,t));
			cout<<mul(ans,inv(n+1))<<"\n";
		}
		else cout<<mul(inv(n),calc(n,k,func(1,1),func(dec(1,p),1),1,1))<<"\n";
	} 
	return 0;
}