P2257 YY的GCD
(默认 \(n \le m\))
设:
则:
令 \(T = kd\),则有:
至此,单次查询的时间复杂度为 \(O(\frac n {\ln n} \sqrt n)\), 总的时间复杂度就为 \(O(n + T(\frac n {\ln n} \sqrt n))\),对于 \(T \le 10^4, n \le 10^7\),显然 \(TLE\)。
那么如何优化呢?
\(ans\) 的得来要枚举 \(k \in prime\) 和 \(d\),乍一看是没得优化的,所以不妨换一种思路:枚举 \(T\)。
首先考虑 \(T\) 的取值范围,易知 \(T \in [1, n]\)。
再考虑 \(F_T\) 会在什么时候,以什么样的方式对答案作出贡献:
\(\forall k \in prime\) 且 \(k | T\),\(F_T\) 会对答案作出 \(\mu_{\frac T k} \times F_T\) 的贡献。
式子也就推出来了:
后面的 \(\sum\limits_{k \in prime, k | T} \mu_{\frac T k}\) 可以在求完 \(\mu\) 数组后 \(O(n \log \log n)\) 维护,过程跟埃氏筛相似,对于每一个素数 \(p\),将 \(f_{kp}(1 \le k \le \lfloor \frac{10^7}p \rfloor)\) 都加上 \(\mu_k\)。再做一个前缀和,套 整除分块,就能把单次查询的时间复杂度降到 \(O(\sqrt n)\),总的时间复杂度就是 \(O(n + n \log \log n + T \sqrt n)\),随便过(虽然常数不小)。
\(Code\)
#include <bits/stdc++.h>
#define MAXN 10000100
using namespace std;
typedef long long ll;
int T, n, m;
int cnt, prime[700100], mu[MAXN], pre[MAXN];
bool vis[MAXN];
template<typename _T>
void read(_T &_x) {
_x = 0;
_T _f = 1;
char _ch = getchar();
while (_ch < '0' || '9' < _ch) {
if (_ch == '-') _f = -1;
_ch = getchar();
}
while ('0' <= _ch && _ch <= '9') {
_x = (_x << 3) + (_x << 1) + (_ch & 15);
_ch = getchar();
}
_x *= _f;
}
template<typename _T>
void write(_T _x) {
if (_x < 0) {
putchar('-');
_x = -_x;
}
if (_x > 9) write(_x / 10);
putchar('0' + _x % 10);
}
void getmu(const int &n) {
mu[1] = 1;
for (int i = 2; i <= n; i++) {
if (!vis[i]) {
prime[++cnt] = i;
mu[i] = -1;
}
for (int j = 1; j <= cnt && i * prime[j] <= n; j++) {
vis[i * prime[j]] = 1;
if (i % prime[j] == 0) break;
mu[i * prime[j]] = -mu[i];
}
}
for (int j = 1; j <= cnt; j++) {
for (int i = 1; i * prime[j] <= n; i++) {
pre[i * prime[j]] += mu[i];
}
}
for (int i = 2; i <= n; i++) pre[i] += pre[i - 1];
}
ll calc(const int &n, const int &m) {
ll res = 0;
for (int l = 1, r; l <= n; l = r + 1) {
r = min(n / (n / l), m / (m / l));
res += 1ll * (pre[r] - pre[l - 1]) * (n / l) * (m / l);
}
return res;
}
int main() {
getmu(1e7);
read(T);
while (T--) {
read(n), read(m);
if (n > m) write(calc(m, n));
else write(calc(n, m));
putchar('\n');
}
return 0;
}
作者:chy12321
出处:https://www.cnblogs.com/chy12321/p/16725182.html
版权:本作品采用「署名-非商业性使用-相同方式共享 4.0 国际」许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现