CF868G El Toll Caves

题意：

有$n$个洞穴，其中一个有宝藏。
你每天有$k$次机会去洞穴中找宝藏，如果你去到的洞穴中有宝藏则有$\frac{1}{2}$的概率找到。每次寻找的概率是独立计算的。
问找到宝藏的期望天数。

数据范围：

$k\le n\le 5\ast 10^8$

Analysis：

发现这$k$次机会一定是有决策的，在随机意义下，我们选择尽量少的去找宝藏，那么就是$k$个$k$个选。在这样的情况下，无限找的话，每$gcd(n,k)$个地方的期望找到的天数一样，我们可以将$n,k$同时除$gcd$答案不变。
设$f_i$表示在第$i$个洞穴找到宝藏的期望天数，那么答案就是$\frac{\sum f_i}{n}$。
我们发现对于前$n-k$个洞穴，会有$f_{i+k}=f_i+1$。因为每一次找$i$前会先找$f_{i-k}$。
对于后$k$个洞穴，会有$f_{i+k-n}=(1-p)\ast (f_i+1)+p=(1-p)\ast f_i+1$。前一次没找到加上第一次就找到的概率。我们把这两种转移看做一种$A,B$变换
我们考虑把答案如下表达:
$$ans=\sum _{i=0}^{k-1}S1(f_i)+\sum _{i=k}^{n-1}S2(f_i)$$
以上的$A,B,S1,S2$都是形如$kx+b$的一次函数形式的变换。
考虑对前$k$个洞穴讨论，能够得到：
设$n$%$k=k^{\prime }$。
当$i$在$[k^{\prime },k)$ 有$f_{i+k^{\prime }-k}=B^{\prime }=A^{-\frac{n}{k}}(B^{-}(f_i))$。考虑第$i$个转移多少次得到$f_{i+k^{\prime }-k}$，然后对于转移做逆变换。
同理，对于$i$在$[0,k^{\prime })$，有$f_{i+k^{\prime }}=A^{\prime }=A^{-\frac{n}{k}-1}(B^{-}(f_i))$。
我们把求$f_i$的部分就转换成了求$n=k,k=k^{\prime }$的子问题。$A^{\prime },B^{\prime }$为新的$A,B$。
考虑求和部分也转换一下。
考虑把后面转移的$A$全部加起来，可以得到：
$$S1=S1+\sum _{i=0}^{\frac{n}{k}}S2(A^i)$$ $$S2=S1+\sum _{i=0}^{\frac{n}{k}-1}S2(A^i)$$
然后求复合函数的次幂部分，由于其满足结合律，我们可以快速幂。
那么这题就做完了，最后当$k=0$时，解个$x=kx+b$的方程即可。
复杂度$O(T\log n)$。

# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std;
typedef long long ll;
const ll mo = 1e9 + 7;
const ll inv2 = mo - mo / 2;
struct node
{
	ll k,b;
	node operator * (node r) const
	{ return (node){k * r.k % mo,(b + k * r.b) % mo}; }
	node operator + (node r) const
	{ return (node){(k + r.k) % mo,(b + r.b) % mo}; }
};
int T;
ll n,k,p;
inline ll gcd(ll a,ll b) { return !b ? a : gcd(b,a % b); }
inline ll pow(ll x,ll p)
{
	ll ret = 1;
	for (; p ; p >>= 1,x = x * x % mo)
	if (p & 1) ret = ret * x % mo;
	return ret;
}
inline node C(node x,ll p)
{
	node ret = (node){1,0};
	for (; p ; p >>= 1,x = x * x)
	if (p & 1) ret = x * ret;
	return ret;
}
inline node G(node x,ll p)
{
	if (!p) return (node){0,0};
	if (x.k == 1) return (node){p,p * (p + 1) % mo * inv2 % mo * x.b % mo};
	node a = C(x,p + 1),z; a.k = (a.k - x.k + mo) % mo,a.b = (a.b - x.b + mo) % mo;
	z.k = a.k * pow(x.k + mo - 1,mo - 2) % mo;
	z.b = (z.k - p + mo) * pow(x.k + mo - 1,mo - 2) % mo * x.b % mo;
	return z;
}
inline ll solve(ll n,ll k,node A,node B,node s1,node s0)
{
	if (!k) return A.b * pow(mo + 1 - A.k,mo - 2) % mo * s0.k % mo;
	node ns1 = s1 + (s0 * G(A,n / k)),ns0 = s1 + (s0 * G(A,n / k - 1));
	node nA = B; ll ni = pow(nA.k,mo - 2); nA.k = ni,nA.b = (mo - nA.b * ni % mo) % mo;
	node nB = nA; ni = pow(A.k,mo - 2); A.k = ni,A.b = (mo - A.b * ni % mo) % mo;
	nA = C(A,n / k - 1) * nA,nB = C(A,n / k) * nB;
	ll S = ((n % k) * ns1.b % mo + (k - n % k) * ns0.b % mo) % mo; ns1.b = ns0.b = 0;
	return (solve(k,n % k,nA,nB,ns1,ns0) + S) % mo;
}
int main()
{
//	freopen("a.in","r",stdin);
	scanf("%d",&T);
	while (T--)
	{
		scanf("%lld%lld",&n,&k); p = inv2;
		ll d = gcd(n,k); n /= d,k /= d;
		printf("%lld\n",solve(n,k,(node){1,1},(node){mo + 1 - p,1},(node){1,0},(node){1,0}) * pow(n,mo - 2) % mo);
	}
	return 0;
}