BZOJ1558 [JSOI2009]等差数列 【线段树】

题目链接

BZOJ1558

题解

等差数列，当然是差分一下

差分值相同的连续位置形成等差数列，我们所选的两个等差数列之间可以有一个位置舍弃
例如：
\(1 \; 2 \; 3 \; 6 \; 8 \; 10\)
差分后是
\(1\; 1\; 3 \; 2\; 2\)
左边两个\(1\)形成等差，右边两个\(2\)形成等差，中间的\(3\)位于两个等差数列的边界，可以舍弃
所以现在问题就转化为了：
在一个区间中选定若干个相同数字的区间，区间之间可以有一个空隙，求最少的区间数
可以用线段树维护

每个节点储存一下左右端点的值，以及\(c[2][2]\)表示左右端点选与不选时的答案

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<map>
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define cls(s) memset(s,0,sizeof(s))
#define LL long long int
#define ls (u << 1)
#define rs (u << 1 | 1)
using namespace std;
const int maxn = 100005,maxm = 100005,INF = 1000000000;
inline int read(){
	int out = 0,flag = 1; char c = getchar();
	while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
	return out * flag;
}
int n,Q,A[maxn],D[maxn];
LL cnt[maxn << 2][2][2],add[maxn << 2],rn[maxn << 2],ln[maxn << 2];
void pd(int u){
	if (add[u]){
		ln[ls] += add[u]; rn[ls] += add[u]; add[ls] += add[u];
		ln[rs] += add[u]; rn[rs] += add[u]; add[rs] += add[u];
		add[u] = 0;
	}
}
void pup(int u){
	ln[u] = ln[ls]; rn[u] = rn[rs];
	if (rn[ls] == ln[rs]){
		cnt[u][0][0] = cnt[ls][0][1] + cnt[rs][1][0] - 1;
		cnt[u][0][1] = cnt[ls][0][1] + cnt[rs][1][1] - 1;
		cnt[u][1][0] = cnt[ls][1][1] + cnt[rs][1][0] - 1;
		cnt[u][1][1] = cnt[ls][1][1] + cnt[rs][1][1] - 1;
	}
	else {
		cnt[u][0][0] = min(cnt[ls][0][0] + cnt[rs][1][0],cnt[ls][0][1] + cnt[rs][0][0]);
		cnt[u][0][1] = min(cnt[ls][0][1] + cnt[rs][0][1],cnt[ls][0][0] + cnt[rs][1][1]);
		cnt[u][1][0] = min(cnt[ls][1][1] + cnt[rs][0][0],cnt[ls][1][0] + cnt[rs][1][0]);
		cnt[u][1][1] = min(cnt[ls][1][1] + cnt[rs][0][1],cnt[ls][1][0] + cnt[rs][1][1]);
	}
}
void build(int u,int l,int r){
	if (l == r){
		cnt[u][1][1] = 1; cnt[u][0][1] = cnt[u][1][0] =  1; cnt[u][0][0] = 0;
		ln[u] = rn[u] = D[l];
		return;
	}
	int mid = l + r >> 1;
	build(ls,l,mid);
	build(rs,mid + 1,r);
	pup(u);
}
void modify(int u,int l,int r,int L,int R,int v){
	if (l >= L && r <= R){
		ln[u] += v; rn[u] += v; add[u] += v;
		return;
	}
	pd(u);
	int mid = l + r >> 1;
	if (mid >= L) modify(ls,l,mid,L,R,v);
	if (mid < R) modify(rs,mid + 1,r,L,R,v);
	pup(u);
}
struct node{
	LL ln,rn,cnt[2][2];
	node(){}
	node(LL a,LL b,LL c[][2]):ln(a),rn(b) {
		cnt[0][0] = c[0][0]; cnt[0][1] = c[0][1];
		cnt[1][0] = c[1][0]; cnt[1][1] = c[1][1];
	}
};
node query(int u,int l,int r,int L,int R){
	if (l >= L && r <= R) return node(ln[u],rn[u],cnt[u]);
	pd(u);
	int mid = l + r >> 1;
	if (mid >= R) return query(ls,l,mid,L,R);
	if (mid < L) return query(rs,mid + 1,r,L,R);
	node a = query(ls,l,mid,L,R),b = query(rs,mid + 1,r,L,R);
	LL c[2][2];
	if (a.rn == b.ln){
		c[0][0] = a.cnt[0][1] + b.cnt[1][0] - 1;
		c[0][1] = a.cnt[0][1] + b.cnt[1][1] - 1;
		c[1][0] = a.cnt[1][1] + b.cnt[1][0] - 1;
		c[1][1] = a.cnt[1][1] + b.cnt[1][1] - 1;
	}
	else {
		c[0][0] = min(a.cnt[0][0] + b.cnt[1][0],a.cnt[0][1] + b.cnt[0][0]);
		c[0][1] = min(a.cnt[0][1] + b.cnt[0][1],a.cnt[0][0] + b.cnt[1][1]);
		c[1][0] = min(a.cnt[1][1] + b.cnt[0][0],a.cnt[1][0] + b.cnt[1][0]);
		c[1][1] = min(a.cnt[1][1] + b.cnt[0][1],a.cnt[1][0] + b.cnt[1][1]);
	}
	return node(a.ln,b.rn,c);
}
int main(){
	n = read();
	REP(i,n) A[i] = read(),D[i] = A[i] - A[i - 1];
	build(1,1,n);
	Q = read();
	char opt; LL l,r,a,b;
	while (Q--){
		opt = getchar(); while (opt != 'A' && opt != 'B') opt = getchar();
		l = read(); r = read();
		if (opt == 'A'){
			a = read(); b = read();
			modify(1,1,n,l,l,a);
			if (l < r) modify(1,1,n,l + 1,r,b);
			if (r < n) modify(1,1,n,r + 1,r + 1,-(a + (r - l) * b));
		}
		else {
			if (l == r) puts("1");
			else{
				node u = query(1,1,n,l + 1,r);
				printf("%lld\n",u.cnt[1][1]);
			}
		}
	}
	return 0;
}