ARC133E
首先套路的,对于每个 K,求出有多少种情况的贡献 ≥K。
然后也是套路的,看到中位数,考虑把 ≥K 的数变为 1,其余变为 0。
于是只有 0,1,发现如果出现了 (a,0,0) 或 (a,1,1),那么就跟 a 没关系了。
先考虑最终答案跟 a 有关,手玩一下发现那个周期将序列 x,y 分成了 G=gcd(N,M) 组,每组包含 N/G 个 x 中的数和 M/G 个 y 中的数。那么考虑一定是形如 (a,1,0) 的形式,那么要么是 N/G 个 x 中的数全是 0,M/G 个 y 中的数全是 1。要么正好反一反。然后每组是独立的,所以方案数是 ((N/G)K−1(M/G)V−K+1+(M/G)K−1(N/G)V−K+1)G。但是注意只能在 A≥K 的时候才能统计进去。
那么与 a 无关的话,其实就是总的减去有关的,然后发现具有对称性,即 K=p 时出来为 1 的方案数与 K=V−p+2 时出来为 0 的方案数是一样的,所以直接除以 2 就行了。
Code:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 200005, mod = 998244353, inv2 = (mod + 1) / 2;
int n, m, V, A;
int ans, sum;
int qpow(int x, int y) {
int res = 1;
while (y) {
if (y & 1) res = 1ll * res * x % mod;
x = 1ll * x * x % mod;
y >>= 1;
}
return res;
}
void add(int &a, int b) {
a += b;
if (a >= mod) a -= mod;
if (a < 0) a += mod;
}
int main() {
scanf("%d%d%d%d", &n, &m, &V, &A);
int g = __gcd(n, m);
sum = 1ll * (V + 1) * qpow(V, n + m) % mod;
for (int k = 1; k <= V + 1; ++k) {
int tmp = qpow((1ll * qpow(k - 1, m / g) * qpow(V - k + 1, n / g) % mod + 1ll * qpow(k - 1, n / g) * qpow(V - k + 1, m / g) % mod) % mod, g);
if (A >= k) add(ans, tmp);
add(sum, -tmp);
}
add(ans, 1ll * sum * inv2 % mod);
printf("%d", ans);
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话