标签：El return int dfrac 1ll Toll line Caves MOD
神仙题。
首先我们每次肯定会按 \(1\sim k\)，\(k+1\sim 2k\)，\(2k+1\sim 3k\) 的顺序选择勘测的洞穴，显然如果出现两个洞穴在某一时刻被勘测次数差 \(\ge 2\) 那肯定是不优的，这个感性理解一下即可。
那么我们假设 \(E_i\) 为在 \(i\) 处的宝藏被勘测到的期望天数，那么由于我们按 \(1\sim k,k+1\sim 2k,2k+1\sim 3k\) 的顺序勘测，有 \(E_{i+k}=E_{i}+1(i+k\le n)\)，以及 \(E_{i+k-n}=E_{i}·\dfrac{1}{2}+1(i+k>n)\)，最终答案就是 \(\dfrac{1}{n}·\sum\limits_{i=1}^nE_i\)。使用手动高斯消元可以做到 \(\mathcal O(n)\) 的复杂度，但还是不足以解决这道题。
我们考虑先手玩一些 \(n,k\) 比较小的情况找找规律，譬如 \(n=11,k=4\)，那么我们将 \(\bmod 4\) 相同的归为一组，有 \(E_4+1=E_8\)，\(E_1+2=E_5+1=E_9\)，\(E_2+2=E_6+1=E_{10}\)，\(E_3+2=E_7+1=E_{11}\)，我们将 \(ans\) 表示成 \(E_1\sim E_4\) 的线性组合，可以得到 \(ans=\dfrac{1}{11}·(3E_1+3+3E_2+3+3E_3+3+2E_4+1)\)，这样我们只要能求出 \(3\sum\limits_{i=1}^3E_i+2E_4\) 即可算出答案，而类似地我们也可以得到恒等式 \(E_1=\dfrac{1}{2}E_8+1=\dfrac{1}{2}(E_4+1)+1\)，\(E_2=\dfrac{1}{2}E_9+1=\dfrac{1}{2}(E_1+2)+1\)，这样我们建立了恒等式 \(E_1=\dfrac{1}{2}E_4+\dfrac{3}{2},E_2=\dfrac{1}{2}E_1+2,E_3=\dfrac{1}{2}E_2+2\)。问题就转化为 \(n=4,k=3\) 的情况。
受到这个问题的启发，我们考虑递归地解决问题，solve(N, K, A, B, f1, f2) 表示目前问题规模 \(n=N,k=K\)，\(A,B,f1,f2\) 为两个一次函数，表示 \(E_{i+k}=A(E_i)(i+k\le n)\)，\(E_{i+k-n}=B(E_i)\)，\(ans=\sum\limits_{i=1}^kf1(E_i)+\sum\limits_{i=k+1}^nf2(E_i)\)。递归求解即可，具体实现见代码。
时间复杂度 \(T\log^2n\)，1 个 log 是欧几里得的复杂度，1 个 log 是复合逆的快速幂。
const int MOD = 1e9 + 7;
const int INV2 = MOD + 1 >> 1;
int qpow(int x, int e) {
int ret = 1;
for (; e; e >>= 1, x = 1ll * x * x % MOD)
if (e & 1) ret = 1ll * ret * x % MOD;
return ret;
}
struct line {
int k, b;
line() {k = b = 0;}
line(int _k, int _b): k(_k), b(_b) {}
int at(int x) {return (1ll * x * k + b) % MOD;}
line operator + (const line &rhs) {return line((k + rhs.k) % MOD, (b + rhs.b) % MOD);}
line operator - (const line &rhs) {return line((k - rhs.k + MOD) % MOD, (b - rhs.b + MOD) % MOD);}
void print() {printf("y = %dx + %d\n", k, b);}
};
line merge(line F, line G) {//G(F(x))
return line(1ll * F.k * G.k % MOD, (1ll * G.k * F.b + G.b) % MOD);
}
line getinv(line F) {
int iv = qpow(F.k, MOD - 2);
return line(iv, (MOD - 1ll * iv * F.b % MOD) % MOD);
}
line qpow_l(line x, int e) {
line res = line(1, 0);
for (; e; e >>= 1, x = merge(x, x)) if (e & 1) res = merge(res, x);
return res;
}
line getsum(line x, int e) {
if (!e) return line(0, 0);
if (x.k == 1) return line(e, 1ll * e * (e + 1) / 2 % MOD * x.b % MOD);
line pw = qpow_l(x, e + 1) - x;
line res; res.k = 1ll * pw.k * qpow(x.k - 1, MOD - 2) % MOD;
res.b = 1ll * x.b * (res.k - e + MOD) % MOD * qpow(x.k - 1, MOD - 2) % MOD;
return res;
}
int solve(int n, int k, line A, line B, line F1, line F2) {
if (!k) {
int E1 = 1ll * A.b * qpow((1 - A.k + MOD) % MOD, MOD - 2) % MOD;
return F2.at(E1);
}
line f1 = F1 + merge(getsum(A, n / k), F2);
line f2 = F1 + merge(getsum(A, n / k - 1), F2);
line a = merge(getinv(B), qpow_l(getinv(A), n / k - 1));
line b = merge(getinv(B), qpow_l(getinv(A), n / k));
return (solve(k, n % k, a, b, line(f1.k, 0), line(f2.k, 0)) + 1ll * (n % k) * f1.b + 1ll * (k - n % k) * f2.b) % MOD;
}
int main() {
freopen("silicon20.in", "r", stdin);
freopen("silicon20.out", "w", stdout);
int qu; scanf("%d", &qu);
while (qu--) {
int n, k; scanf("%d%d", &n, &k); int d = __gcd(n, k); n /= d; k /= d;
printf("%d\n", 1ll * solve(n, k, line(1, 1), line(INV2, 1), line(1, 0), line(1, 0)) * qpow(n, MOD - 2) % MOD);
}
return 0;
}
标签：El,return,int,dfrac,1ll,Toll,line,Caves,MOD 来源： https://www.cnblogs.com/ET2006/p/Codeforces-868G.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。