2021 Byte Camp 2021 Qualification Round M Expected area 题解

题面

给定 $2n$ 个线段 $[l_i,r_i]$ $(1 \le i \le n)$ 和 $[s_i,t_i]$ $(1 \le i \le n)$,对于所有排列 $p$ 求 $\bigcup_{i=1}^{n} [l_i,r_i] \times [s_{p_i}, t_{p_i}]$ (面积并)之和。其中 $1 \le n \le 2 \times 10^5$。

分析

拆开来算贡献。

对于第一维($x$ 轴),将其通过左闭右开区间离散化为小块。假设每块被覆盖了 $s_i$ 次,也就意味着这一个块对应了 $s_i$ 个 $y$ 轴线段。比如说,$s_i=0$,那么上面没有任何线段;$s_i=1$,对应了 $1$ 个线段,也就是所有线段均有可能竖着放在它上面,对答案的总贡献为所有线段的总长度;类似地,$s_i=2$ 就是枚举一对线段计算贡献之和。

可以发现,对于每一个小块只有参数 $s_i$ 有效。于是,我们将问题转化为了一个一维的问题,即对于从线段集合中选择 $k=0,1,\dots,n$ 条线段产生的总贡献,形式化地 $f(k)=\sum_{S \subseteq U, |S|=k} |\bigcup_{i \in S} [s_i, t_i]|$。

再拆开来算一次贡献,同样离散化。假设每块被覆盖了 $s_i$ 次,这个块产生贡献当且仅当选出的 $k$ 个块包含 $s_i$ 中的某一个,即方案数就是 $\binom{n}{k} - \binom{n-s_i}{k}$,对于贡献还要乘上块的长度。观察上面的式子,第一部分是平凡的,关键是第二部分,因为我们最终计算的是 $f(k)$ 每个点的值,因此同样的 $n-s_i$ 产生的长度贡献可以记录在一起。于是,我们定义这部分的贡献函数 $g(k)=\sum_{i=k}^n a_i C(i,k)$,其中 $a_i$ 就是长度和。

将 $g$ 展开

注意到上式的下标分别为 $i$ 和 $i-k$,即差为定值 $k$,使用 FFT 计算即可。

最后,还有一点小问题,$f(k)$ 的贡献是无序的。它表示从第二个集合中选择了 $k$ 条线段去和横轴某一小块的 $k$ 次覆盖对应,剩余随便,也就是需要乘上两部分内部的排列,即 $k! \times (n-k)!$。

代码

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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
#include <iostream>
#include <cstdio>
#include <cassert>
#include <cstring>
#include <cmath>
#include <functional>
#include <algorithm>
#include <utility>
#include <vector>
#define ms(a,b) memset(a,b,sizeof(a))
using namespace std;
using ll = long long;
using PII = pair<int,int>;
const int mod = 998244353;
const int inf = 1 << 30;
const int maxn = 200000 + 5;

inline int add(int x, int y) {
x += y;
return x >= mod ? x -= mod : x;
}
inline int sub(int x, int y) {
x -= y;
return x < 0 ? x += mod : x;
}
inline int mul(int x, int y) {
return 1ll * x * y % mod;
}
inline int qpow(int x, ll n) {
int r = 1;
while (n > 0) {
if (n & 1) r = 1ll * r * x % mod;
n >>= 1; x = 1ll * x * x % mod;
}
return r;
}
inline int inv(int x) {
return qpow(x, mod - 2);
}

namespace Comb {
const int maxc = 2000000 + 5;
int f[maxc], inv[maxc], finv[maxc];
void init() {
inv[1] = 1;
for (int i = 2; i < maxc; i++)
inv[i] = (mod - mod / i) * 1ll * inv[mod % i] % mod;
f[0] = finv[0] = 1;
for (int i = 1; i < maxc; i++) {
f[i] = f[i - 1] * 1ll * i % mod;
finv[i] = finv[i - 1] * 1ll * inv[i] % mod;
}
}
int C(int n, int m) {
if (m < 0 || m > n) return 0;
return f[n] * 1ll * finv[n - m] % mod * finv[m] % mod;
}
}
using Comb::C;

using Poly = vector<int>;

namespace PolyOpe {
#ifdef XLor
#define print(a) cout << "\033[32;1m" << #a << " -> ", err(a)
void err(const Poly& a) {
std::cout << "[ ";
for (auto& x: a) std::cout << x << ' ';
std::cout << "]\033[39;0m" << std::endl;
}
#else
#define print(a)
#endif

int rev[maxn << 3], aa[maxn << 3], bb[maxn << 3];
inline int build(int m) {
int step = 0, n = 1;
for (; n < m; n <<= 1) ++step;
for (int i = 1; i < n; i++)
rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (step - 1));
return n;
}

inline void ntt(int a[], int n, int op = 0) {
for (int i = 0; i < n; i++) if (i < rev[i]) swap(a[i], a[rev[i]]);
for (int h = 2; h <= n; h <<= 1) {
int wn = qpow(3, (mod - 1) / h);
if (op == -1) wn = inv(wn);
for (int i = 0; i < n; i += h) {
int w = 1;
for (int j = i; j < i + h / 2; j++) {
int u = a[j], t = 1ll * a[j + h / 2] * w % mod;
a[j] = (u + t) % mod;
a[j + h / 2] = (u - t + mod) % mod;
w = 1ll * w * wn % mod;
}
}
}
if (op == -1) {
ll rn = inv(n);
for (int i = 0; i < n; i++) a[i] = 1ll * a[i] * rn % mod;
}
}
inline Poly operator * (const Poly& a, const Poly& b) {
int lim = build((int)a.size() + (int)b.size() - 1);
for (int i = 0; i < (int)a.size(); i++) aa[i] = a[i];
for (int i = (int)a.size(); i < lim; i++) aa[i] = 0;
for (int i = 0; i < (int)b.size(); i++) bb[i] = b[i];
for (int i = (int)b.size(); i < lim; i++) bb[i] = 0;
ntt(aa, lim); ntt(bb, lim);
for (int i = 0; i < lim; i++) aa[i] = 1ll * aa[i] * bb[i] % mod;
ntt(aa, lim, -1);
Poly ans;
for (int i = 0; i < (int)a.size() + (int)b.size() - 1; i++)
ans.push_back(aa[i]);
return ans;
}
inline Poly operator + (Poly f, const Poly& g) {
f.resize(max(f.size(), g.size()));
for (int i = 0; i < (int)g.size(); i++) {
f[i] = add(f[i], g[i]);
}
return f;
}
inline Poly operator - (Poly f, const Poly& g) {
f.resize(max(f.size(), g.size()));
for (int i = 0; i < (int)g.size(); i++) {
f[i] = sub(f[i], g[i]);
}
return f;
}
}
using namespace PolyOpe;

int n, nn, l[maxn], r[maxn];
int a[maxn], pre[maxn];
int g[maxn], f[maxn];

int main() {
Comb::init();
scanf("%d", &n);
vector<int> lsh;
auto gid = [&](int x) {
return lower_bound(lsh.begin(), lsh.end(), x) - lsh.begin() + 1;
};

for (int i = 1; i <= n; i++) {
scanf("%d%d", l + i, r + i);
lsh.push_back(l[i]);
lsh.push_back(r[i]);
}
sort(lsh.begin(), lsh.end());
lsh.resize(unique(lsh.begin(), lsh.end()) - lsh.begin());
nn = lsh.size();
for (int i = 1; i <= n; i++) {
int L = gid(l[i]);
int R = gid(r[i]);
pre[L]++;
pre[R]--;
}
int all = 0;
for (int i = 1; i < nn; i++) {
pre[i] += pre[i - 1];
assert(0 <= pre[i] && pre[i] <= n);
a[n - pre[i]] = add(a[n - pre[i]], lsh[i] - lsh[i - 1]);
all = add(all, lsh[i] - lsh[i - 1]);
}

Poly x, y;
for (int i = 0; i <= n; i++) {
x.push_back(mul(a[i], Comb::f[i]));
y.push_back(Comb::finv[n - i]);
}
Poly res = x * y;

for (int k = 0; k <= n; k++) {
f[k] = add(f[k], mul(all, C(n, k)));
f[k] = sub(f[k], mul(Comb::finv[k], res[n + k]));
f[k] = mul(f[k], Comb::f[k]);
f[k] = mul(f[k], Comb::f[n - k]);
}

ms(pre, 0);
lsh.clear();
for (int i = 1; i <= n; i++) {
scanf("%d%d", l + i, r + i);
lsh.push_back(l[i]);
lsh.push_back(r[i]);
}
sort(lsh.begin(), lsh.end());
lsh.resize(unique(lsh.begin(), lsh.end()) - lsh.begin());
nn = lsh.size();
for (int i = 1; i <= n; i++) {
int L = gid(l[i]);
int R = gid(r[i]);
pre[L]++;
pre[R]--;
}
int ans = 0;
for (int i = 1; i < nn; i++) {
pre[i] += pre[i - 1];
assert(0 <= pre[i] && pre[i] <= n);
ans = add(ans, mul(lsh[i] - lsh[i - 1], f[pre[i]]));
}
ans = mul(ans, Comb::finv[n]);
printf("%d\n", ans);
return 0;
}