『数位DP及其优化』CF1327F AND Segments

$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}$

题目描述

你有三个整数 $n,k,m$ 以及 $m$ 个限制 $(l_1,r_1,x_1),(l_2,r_2,x_2),\dots ,(l_m,r_m,x_m)$。

计算满足下列条件的，长度为 $n$ 的序列 $a$ 的个数：

• 对于每个 $1\le i\le n$，$0\le a_i<2^k$。
• 对于每个 $1\le i\le m$，数字的按位与 $a[l_i]\text{ and }{a}_{[}{l}_i+1]\text{ and }\dots \text{ and }a[r_i]=x_i$。

两个序列 $a,b$ 被认为是不同的，当且仅当存在一个位置 $i$ 满足 $a_i\ne b_i$。

由于答案可能过大，请输出其对 $998244353$ 取模的结果。

输入格式

第一行输入三个整数 $n,k,m(1\le n\le 5\cdot 10^5;1\le k\le 30;0\le m\le 5\cdot 10^5)$，分别表示数组 $a$ 的长度，$a$ 中元素的值域，以及限制的个数。

接下来 $m$ 行，每行描述一个限制 $l_i,r_i,x_i(1\le l_i\le r_i\le n;0\le x_i<2^k)$，分别表示限制的线段区间以及按位与值。

输出格式

输出一行一个整数，表示满足条件的序列 $a$ 的个数，对 $998244353$ 取模的结果。

说明/提示

你可以在 这里 获得有关按位与的信息。

在一个样例中，合法的序列 $a$ 有：$[3,3,7,6]$，$[3,7,7,6]$ 以及 $[7,3,7,6]$。

---

$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}$

这个题我们首先按位考虑，把问题转化为了：

• 一个长度为 $n$ 的 $01$ 序列，给定两种限定：一种是区间内全部为 $1$,另一种是区间内至少有一个 $0$，问题有多少个合法的符合符合上述方案。

考虑上述两种方案，显然前者只需要直接赋值即可。

对于第二种限制，发现对于当前节点 $i$ 来说，如果选取的时$0$，设上一次选取 $0$ 的位置为$j$，那么一定需要保证不存在第二种区间$[l,r]$满足$j<l\le r<i$。因此我们得到结论：

• 当第 $i$ 个位置被钦定为 $0$ 后，上一个 $0$ 的位置 $j$ 必须大于右端点小于$i$ 的任意左端点。即，对于任意取件$r_i<i$，$j$ 至少为 min ⁡ { l i + 1 } \min\{l_i+1\} min{li​+1}。

由于这是计数题，难以考虑组合数的方法，我们考虑DP。

设$f_i$表示到第 $i$ 个位置，且当前位置一定为 $0$ 的方案数。

• 当$a_i=1$时，强制$f_i=0$。
• 找到满足条件的最小$j$ $$f_i=\sum f_j$$

考虑优化，我们发现累加的 $j$ 一个连续的区间，我们直接用前缀和优化即可。

现在我们的问题转化为了如何求解 $j$ 的限制范围。

很妙的方法就是：

L[r[i]] = L[i];
for i = 1 - n : L[i]= max(L[i-1], L[i]).

---

$\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}$

#include <bits/stdc++.h>
#define int long long

using namespace std;
const int N = 6e5;
const int P = 998244353;

int n, m, k;
int s[N], Lmax[N], f[N], l[N], r[N], x[N];

int read(void)
{
	int s = 0, w = 0; char c = getchar();
	while (c < '0' || c > '9') w |= c == '-', c = getchar();
	while (c >= '0' && c <= '9') s = s*10+c-48, c = getchar();
	return w ? -s : s;
}

int calc(int t)
{
	for (int i=1;i<=n;++i) s[i] = Lmax[i] = f[i] = 0;
	for (int i=1;i<=m;++i) {
		if ((x[i] >> t) & 1) s[l[i]] ++, s[r[i] + 1] --;
		else Lmax[r[i]] = max(Lmax[r[i]], l[i]);
	}
	for (int i=1;i<=n+1;++i) {
		s[i] += s[i-1];
		Lmax[i] = max(Lmax[i-1], Lmax[i]);
	}
	int sum = 1, j = 0; f[0] = 1;
	for (int i=1;i<=n+1;++i)
	{
		if (s[i] > 0) { f[i] = 0; continue; }
		while (j < Lmax[i-1]) sum = (sum - f[j]) % P, j ++;
		f[i] = sum, sum = (sum + f[i]) % P;
	}
	f[n+1] = (f[n+1] % P + P) % P;
	return f[n+1];
}

signed main(void)
{
	n = read(), k = read(), m = read();
	for (int i=1;i<=m;++i) {
		l[i] = read();
		r[i] = read();
		x[i] = read();
	}
	int res = 1;
	for (int i=0;i<k;++i) res = res * calc(i) % P;
	cout << res << endl;
	return 0;
}