-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathmain.cpp
More file actions
89 lines (81 loc) · 2.21 KB
/
Copy pathmain.cpp
File metadata and controls
89 lines (81 loc) · 2.21 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
constexpr int mod = 998'244'353;
constexpr int add(int a, int b) {
return a + b < mod ? a + b : a + b - mod;
}
constexpr int sub(int a, int b) {
return a < b ? a - b + mod : a - b;
}
constexpr int mul(int a, int b) {
return i64(a) * b % mod;
}
constexpr int pow(int x, int n) {
int ret = 1;
for (; n; n >>= 1) {
if (n & 1) ret = mul(ret, x);
x = mul(x, x);
}
return ret;
}
auto sol = [](int q, auto qs) {
int sz = 200'000;
vector fac(sz + 1, 1);
for (int i = 1; i <= sz; i++) fac[i] = mul(fac[i - 1], i);
vector inv(sz + 1, pow(fac[sz], mod - 2));
for (int i = sz; i >= 1; i--) inv[i - 1] = mul(inv[i], i);
auto bino = [&](int n, int k) {
if (k == 0 || k == n) return 1;
return mul(fac[n], mul(inv[k], inv[n - k]));
};
vector ret(q, 0), cand(0, 0);
for (int i = 0; i < q; i++) {
auto& [n, x] = qs[i];
if (x < 0) x = -x;
if (x >= n) ret[i] = x;
else cand.push_back(i);
}
sort(cand.begin(), cand.end(), [&](int i, int j) {
if (qs[i][0] / 400 != qs[j][0] / 400) return qs[i][0] < qs[j][0];
if (qs[i][0] / 400 % 2) return qs[i][0] + qs[i][1] < qs[j][0] + qs[j][1];
return qs[i][0] + qs[i][1] > qs[j][0] + qs[j][1];
});
int a = 1, b = 0, v1 = 1, v2 = 0;
for (int i : cand) {
auto [n, x] = qs[i];
int m = (n + x) / 2;
while (a < n) {
v2 = sub(add(mul(v2, 2), v1), mul(b + 1, bino(a, b)));
v1 = sub(mul(v1, 2), bino(a, b));
a++;
}
while (b > m) {
v1 = sub(v1, bino(a, b));
v2 = sub(v2, mul(b, bino(a, b)));
b--;
}
while (a > n) {
a--;
v1 = mul(add(v1, bino(a, b)), (mod + 1) / 2);
v2 = mul(add(sub(v2, v1), mul(b + 1, bino(a, b))), (mod + 1) / 2);
}
while (b < m) {
b++;
v1 = add(v1, bino(a, b));
v2 = add(v2, mul(b, bino(a, b)));
}
ret[i] = sub(mul(n + x, v1), mul(2, v2));
ret[i] = mul(ret[i], pow(pow(2, n - 1), mod - 2));
ret[i] = sub(ret[i], x);
}
return ret;
};
int main() {
cin.tie(0)->sync_with_stdio(0);
int q; cin >> q;
vector qs(q, array{ 0, 0 });
for (int i = 0; i < q; i++) cin >> qs[i][0] >> qs[i][1];
auto res = sol(q, qs);
for (int i = 0; i < q; i++) cout << res[i] << '\n';
}