题目
题目概要
给出长度为
n
n
n 的序列
a
a
a,有
q
q
q 次询问:在区间
[
l
,
r
]
[l,r]
[l,r] 中选出若干个长度为
L
L
L 的子区间,两两无交,使被覆盖的元素的和最大。
数据范围与提示
n
=
q
∈
{
1
0
3
,
1
0
5
}
n=q\in\{10^3,10^5\}
n=q∈{103,105} 但
L
≤
50
L\le 50
L≤50 。
思路
分治,处理经过中线 m m m 的询问(即满足 l ⩽ m ⩽ r l\leqslant m\leqslant r l⩽m⩽r 的询问)。枚举中间选了区间 [ i , i + L ) ( m − L < i ⩽ m ) [i,i+L)\;(m-L<i\leqslant m) [i,i+L)(m−L<i⩽m) 后,预处理所有区间 [ x , i ) [x,i) [x,i) 和 [ i + L , x ] [i+L,x] [i+L,x] 的答案——即在该范围内随意选若干个长度为 L L L 的子区间的最大求和。显然可以 O ( n ) \mathcal O(n) O(n) 递推。然后可以更新一次询问。
不经过中线的询问则划分到两边去。一共
O
(
log
n
)
\mathcal O(\log n)
O(logn) 层(即询问的最多被处理次数),而每一层还有额外复杂度
O
(
n
L
)
\mathcal O(nL)
O(nL),所以总复杂度
O
(
n
L
log
n
+
q
log
n
)
\mathcal O(nL\log n+q\log n)
O(nLlogn+qlogn)
本质是猫树。虽然它一点也不像树啊。
代码
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long int_;
inline int readint(){
int x; scanf("%d",&x); return x;
}
const int MaxN = 200005;
int_ s[MaxN], dp[MaxN]; int L;
void leftDP(int x,int l){
for(int i=x; i>x-L+1&&i>=l; --i)
dp[i] = 0;
if(x-L+1 < l) return ;
dp[x-L+1] = max(0ll,s[x]-s[x-L]);
for(int i=x-L; i>=l; --i)
dp[i] = max(dp[i+1],
dp[i+L]+s[i+L-1]-s[i-1]);
}
void rightDP(int x,int r){
for(int i=x; i<x+L-1; ++i)
dp[i] = 0;
dp[x+L-1] = max(0ll,s[x+L-1]-s[x-1]);
for(int i=x+L; i<=r; ++i)
dp[i] = max(dp[i-1],
dp[i-L]+s[i]-s[i-L]);
}
struct Query{
int l, r, id;
};
Query q[MaxN], xyx[MaxN];
int_ ans[MaxN];
void solve(int l,int r,int ql,int qr){
if(r-l+1 < L || ql > qr) return ;
int mid = (l+r)>>1;
for(int j=max(mid-L+2,l); j<=mid; ++j){
leftDP(j-1,l), rightDP(j+L,r);
dp[j] = dp[j+L-1] = 0;
for(int i=ql; i<=qr; ++i)
if(q[i].l <= j && j+L-1 <= q[i].r)
ans[q[i].id] = max(ans[q[i].id],
dp[q[i].l]+dp[q[i].r]
+s[j+L-1]-s[j-1]);
}
leftDP(mid,l), rightDP(mid+1,r);
for(int i=ql; i<=qr; ++i)
if(q[i].l <= mid && mid < q[i].r)
ans[q[i].id] = max(ans[q[i].id],
dp[q[i].l]+dp[q[i].r]);
int pl = ql, pr = qr;
for(int i=ql; i<=qr; ++i)
if(q[i].r <= mid)
xyx[pl ++] = q[i];
else if(q[i].l > mid)
xyx[pr --] = q[i];
for(int i=ql; i<=qr; ++i)
q[i] = xyx[i];
solve(l,mid,ql,pl-1);
solve(mid+1,r,pr+1,qr);
}
int main(){
int n = readint(); L = readint();
for(int i=1; i<=n; ++i)
s[i] = s[i-1]+readint();
int Q = readint();
for(int i=1; i<=Q; ++i){
q[i].l = readint();
q[i].r = readint();
q[i].id = i;
}
solve(1,n,1,Q);
for(int i=1; i<=Q; ++i)
printf("%lld\n",ans[i]);
return 0;
}