BZOJ 3636 教义问答手册（分治）

最新推荐文章于 2021-04-02 17:24:23 发布

_Ark

最新推荐文章于 2021-04-02 17:24:23 发布

阅读量229

点赞数

分类专栏： CDQ分治整体二分

本文链接：https://blog.csdn.net/Ike940067893/article/details/102149477

版权

CDQ分治同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

整体二分

2 篇文章 0 订阅

订阅专栏

题意

一个整数数列，多次询问某段区间 $l_i,r_i]$ 内，选出若干个长度为 $L$ 且不相交的连续段使选出来的数和最大。

分析

首先想朴素的区间 $D P$
设 $f [i] [j]$ 表示区间 $[i, j]$ 的答案。 $O(n^2)$ 转移比较显然。
$f[i][j]=max(f[i][j-1],f[i][j-L]+\sum_{k=j-L+1}^ja[j])$
为了方便我们把 $\sum_{k=j-L+1}^ja[j]$ 记作 $b [j]$ ，则
$f [i] [j] = m a x (f [i] [j - 1], f [i] [j - L] + b [j])$
空间和时间都会爆。然后想想多组询问怎么做。

发现 $L\le50$ ，于是就有巧妙的分治做法。

对于区间 $[q L, q R]$ ，中点是 $m i d$ 。我们把跨越 $m i d$ 的询问拿出来求答案，没有跨越的分治下去，就变成了子问题。

那么对于一个询问 $[q L, q R]$ ，它的答案有两种情况。

一种是没有选跨越 $m i d$ 的连续段，那么就是 $[q L, m i d]$ 和 $[m i d + 1, q R]$ 的答案加起来。
另一种是跨越了 $m i d$ 的。因为 $L$ 很小我们枚举跨越 $m i d$ 的一段在左边有多长，记作 $j$ ，则右边长为 $L - j$ 。那么有 $1 < j < L, 1 < L - j < L$ 。
那么答案就是 $[q L, m i d - j]$ 的答案加上 $[m i d + 1 + L - j, q R]$ 的答案再加上中间一段的答案 $b [m i d + L - j]$

发现上面需要算答案的区间只有 $O (n L)$ 级别的且连续，可以直接预处理。

然后就做完了。时间复杂度 $O(nL\log n)$ 。具体实现见代码。

CODE

#include <bits/stdc++.h>
using namespace std;
char cb[1<<18],*cs,*ct;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<18,stdin),cs==ct)?0:*cs++)
inline void rd(int &x) {
	x = 0; char ch; int flg=1; while(!isdigit(ch=getc()))if(ch=='-')flg=-flg;
	do x=x*10+ch-'0'; while(isdigit(ch=getc())); x *= flg;
}
const int MAXN = 100005;
const int MAXL = 55;
int n, m, L, a[MAXN], ans[MAXN];
struct query { int l, r, id; }q[MAXN], tmp1[MAXN], tmp2[MAXN];
int fL[MAXL][MAXN], fR[MAXL][MAXN];
inline void workL(int l, int r) {
	for(int i = 0; i <= r-l && i < L; ++i) {
		fL[i][r-i+1] = 0;
		for(int j = r-i; j >= l; --j)
			fL[i][j] = j+L-1 > r-i ? 0 : max(fL[i][j+1], fL[i][j+L] + a[j+L-1]);
	}
}
inline void workR(int l, int r) {
	for(int i = 0; i <= r-l && i < L; ++i) {
		fR[i][l+i-1] = 0;
		for(int j = l+i; j <= r; ++j)
			fR[i][j] = j-L+1 < l+i ? 0 : max(fR[i][j-1], fR[i][j-L] + a[j]);
	}
}
void cdq(int l, int r, int ql, int qr) {
	if(ql > qr || r-l+1 < L) return;
	int mid = (l + r) >> 1, it1 = 0, it2 = 0;
	workL(l, mid); workR(mid+1, r);
	for(int i = ql; i <= qr; ++i) {
		if(q[i].r <= mid) tmp1[++it1] = q[i];
		else if(q[i].l > mid) tmp2[++it2] = q[i];
		else {
			ans[q[i].id] = fL[0][q[i].l] + fR[0][q[i].r];
			for(int j = max(1, L-(q[i].r-mid)); j <= mid-q[i].l+1 && j < L; ++j)
				ans[q[i].id] = max(ans[q[i].id], (mid-j < l ? 0 : fL[j][q[i].l]) + (mid+L-j+1 > r ? 0 : fR[L-j][q[i].r]) + a[mid+L-j]);
		}
	}
	for(int i = 1; i <= it1; ++i) q[ql+i-1] = tmp1[i];
	for(int i = 1; i <= it2; ++i) q[ql+it1+i-1] = tmp2[i];
	cdq(l, mid, ql, ql+it1-1); cdq(mid+1, r, ql+it1, ql+it1+it2-1);
}
int main () {
	rd(n), rd(L);
	for(int i = 1; i <= n; ++i) rd(a[i]), a[i] += a[i-1];
	for(int i = n; i >= L; --i) a[i] -= a[i-L];
	rd(m);
	for(int i = 1; i <= m; ++i) rd(q[i].l), rd(q[i].r), q[i].id = i;
	cdq(1, n, 1, m);
	for(int i = 1; i <= m; ++i) printf("%d\n", ans[i]);
}