BZOJ1558 等差数列

题目链接:戳我

实话实话,看了几篇题解真的没看懂,我觉得讲的都有问题。这里对于线段树维护的s写了一点我自己的理解。

看到等差数列,我们考虑对数列做差,这样如果是等差数列,那么值应该相等。(比较容易维护,修改操作就变成了两个单点修改+一个区间修改,如果还是不理解的话可以参考一下代码)

查询比较麻烦,有这样的情况需要考虑:

举个例子——
1 3 5 6 9 12
它的差分数列为 2 2 1 3 3

最佳选择肯定是(1,3,5)(6,9,12),ans=2
但是我们纯看差分数列找相同的,ans=3

为什么会多算呢?原因就在于差分数列为1的这个位置其实是两个等差数列拼合点,不需要考虑在内。

所以我们设\(s[0],s[1],s[2],s[3]\)分别表示当前区间中,不考虑两端、不考虑左端点、不考虑右端点、两端都考虑,形成等差数列的个数。

两个区间合并的时候,左区间的右端点和右区间的左端点是至少要考虑一个的。因为如果都不考虑的娿,终究就遗漏了一个点了。两个都考虑时,如果中间差值相同,那么两个等差数列可以拼做一个,ans要-1.

注意查询的时候区间写(l,r-1)qwqwqwq

备注一下l,r记录的是左右端点的数值。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
#define MAXN 120000
int v[MAXN],n;
struct Data{int s[4],l,r;};
struct Node
{
    int l,r,v;
    Data x;
}t[MAXN<<2];

Data operator + (Data x,Data y)
{
    Data cur;
    cur.l=x.l,cur.r=y.r;

    cur.s[0]=min(x.s[2]+y.s[0],x.s[0]+y.s[1]);
    cur.s[0]=min(cur.s[0],x.s[2]+y.s[1]-(y.l==x.r));

    cur.s[1]=min(x.s[1]+y.s[1],x.s[3]+y.s[0]);
    cur.s[1]=min(cur.s[1],x.s[3]+y.s[1]-(y.l==x.r));

    cur.s[2]=min(x.s[2]+y.s[2],x.s[0]+y.s[3]);
    cur.s[2]=min(cur.s[2],x.s[2]+y.s[3]-(y.l==x.r));

    cur.s[3]=min(x.s[3]+y.s[2],x.s[1]+y.s[3]);
    cur.s[3]=min(cur.s[3],x.s[3]+y.s[3]-(y.l==x.r));

    return cur;
}

inline int ls(int x){return x<<1;}

inline int rs(int x){return x<<1|1;}

inline void solve(int now,int k)
{
    t[now].v+=k;
    t[now].x.l+=k;
    t[now].x.r+=k;
}
inline void pushdown(int now)
{
    if(t[now].v)
    {
        solve(ls(now),t[now].v);
        solve(rs(now),t[now].v);
        t[now].v=0;
    }
}

inline void Build(int now,int l,int r)
{
    t[now].l=l,t[now].r=r;
    if(l==r)
    {
        t[now].x.s[0]=0;
        t[now].x.s[1]=t[now].x.s[2]=t[now].x.s[3]=1;
        t[now].x.l=t[now].x.r=v[l];
        return;
    }
    int mid=(l+r)>>1;
    Build(ls(now),l,mid);
    Build(rs(now),mid+1,r);
    t[now].x=t[ls(now)].x+t[rs(now)].x;
}


inline Data Query(int now,int ll,int rr)
{
    int l=t[now].l,r=t[now].r;
    if(l==ll&&r==rr) return t[now].x;
    pushdown(now);
    int mid=(l+r)>>1;
    if(rr<=mid) return Query(ls(now),ll,rr);
    if(mid<ll) return Query(rs(now),ll,rr);
    return Query(ls(now),ll,mid)+Query(rs(now),mid+1,rr);
}

inline void Modify(int now,int ll,int rr,int k)
{
    int L=t[now].l,R=t[now].r;
    if(L==ll&&R==rr)
    {
        t[now].v+=k;
        t[now].x.l+=k,t[now].x.r+=k;
        return;
    }
    pushdown(now);
    int mid=(L+R)>>1;
    if(rr<=mid) Modify(ls(now),ll,rr,k);
    else if(mid<ll) Modify(rs(now),ll,rr,k);
    else
    {
        Modify(ls(now),ll,mid,k);
        Modify(rs(now),mid+1,rr,k);
    }
    t[now].x=t[ls(now)].x+t[rs(now)].x;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    freopen("std.out","w",stdout);
    #endif 
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&v[i]);
    for(int i=1;i<n;i++) v[i]=v[i+1]-v[i];
    Build(1,1,n-1);
    int Q;
    scanf("%d",&Q);
    for(int i=1;i<=Q;i++)
    {
        char op[10];
        int s,t,a,b;
        scanf("%s",op);
        if(op[0]=='B')
        {
            scanf("%d%d",&s,&t);
            if(s==t) printf("1\n");
            else printf("%d\n",Query(1,s,t-1).s[3]);
        }
        else if(op[0]=='A')  
        {
            scanf("%d%d%d%d",&s,&t,&a,&b);
            if(s!=1) Modify(1,s-1,s-1,a); 
            if(s!=t) Modify(1,s,t-1,b);
            if(t!=n) Modify(1,t,t,-(t-s)*b-a); 
        }
    }
    return 0;
}
posted @ 2019-03-09 15:47  风浔凌  阅读(120)  评论(0编辑  收藏  举报