洛谷[P1438] 无聊的数列

题目类型:差分,线段树

传送门:>Here<

题意:给出一个数列,每次给一个区间对应的加上一个等差数列,并询问某一个元素目前的值。

解题思路

所谓差分,我个人的理解就是用\(O(1)\)的方法来维护前缀和,当然查询变为了\(O(n)\)。差分就好像将前缀和变成了一个数一样——当一段区间需要全部加上\(k\)时:差分数组某一位上\(+k\),意味着这之后的所有元素都将\(+k\)。就好像一条带子拖到最后了。因此我们如果仅仅操作一个区间的话,那么要把后面多出来的带子减掉,于是我们再另外加一条负的带子在后面。

刚才谈论整个区间都加一个相同的数。如果整个区间加的是一个等差数列呢?相当于这个区间内所加的数,每个都比前面的多加\(d\)。效果就等价于在差分数组中,令这个区间的每个元素加上\(d\)。然后末尾要减去末项。依然使用刚才的比喻,将那么多条相同的带子依次叠放,假设区间长度是\(l\),那么最后一个元素那里肯定放着\(l\)条带子了。而我们在最后需要把这\(l\)条带子全部减掉。

因此,如果用差分来维护这道题,我们来总结一下步骤:(按照题意,等差数列的更新方法是\(l \ r \ k \ d\),代表左端点,右端点,首项,公差;设差分数组为\(s\)

  • \(s[l]+=k\)

  • \(s[l+1..r]+=d\)

  • \(s[r+1]-=k+d*(r-l)\)(末项)

由此我们发现,对于大多数的情况都是\(+d\),因此转化为一个区间更新的问题。差分数组的统计方法我们已经很熟悉了,需要从头遍历。因此元素\(p\)现在的值应该是:初始值 + \(s[1..p]\),因此转化为一个区间查询的问题

因此我们可以用线段树方便地\(O(logn)\)维护好

反思

一直以为差分和线段树维护的几乎是同一个东西,却从来没想过线段树可以用来维护差分!线段树维护差分,就好像求和的和。然而等差数列就好像是三维的一样,先由差分转化为二维,然后由线段树转化为线性。

正好像我们在找规律时所作的一样,差,差之差,差之差之差。那么这道题就好像倒过来,和,和的和,和的和的和……

Code

不需要建树,线段树写起来好像异常短小精悍……\(qwq\)


/*By DennyQi 2018*/
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 100010;
const int INF = 1061109567;
inline int Max(const int a, const int b){ return (a > b) ? a : b; }
inline int Min(const int a, const int b){ return (a < b) ? a : b; }
inline int read(){
    int x = 0; int w = 1; register char c = getchar();
    for(; c ^ '-' && (c < '0' || c > '9'); c = getchar());
    if(c == '-') w = -1, c = getchar();
    for(; c >= '0' && c <= '9'; c = getchar()) x = (x<<3) + (x<<1) + c - '0'; return x * w;
}
int N,M,opt,l,r,x,y;
int a[MAXN];
int val[MAXN<<2],lazy[MAXN<<2];
struct SegmentTree{
    inline void pushdown(int rt, int l, int r){
        if(lazy[rt]){
            int mid = (l+r)/2;
            val[rt<<1] += lazy[rt] * (mid-l+1);
            val[rt<<1|1] += lazy[rt] * (r-(mid+1)+1);
            lazy[rt<<1] += lazy[rt];
            lazy[rt<<1|1] += lazy[rt];
            lazy[rt] = 0;
        }
    }
    int query(int rt, int l, int r, int x, int y){
        if(l > y || r < x) return 0;
        if(x <= l && r <= y) return val[rt];
        pushdown(rt, l, r);
        int mid = (l+r)/2;
        return query(rt<<1,l,mid,x,y) + query(rt<<1|1,mid+1,r,x,y);
    }
    void update(int rt, int l, int r, int x, int y, int k){
        if(l > y || r < x) return;
        if(x <= l && r <= y){
            lazy[rt] += k;
            val[rt] += (r-l+1) * k;
            return;
        }
        pushdown(rt,l,r);
        int mid = (l+r)/2;
        update(rt<<1,l,mid,x,y,k), update(rt<<1|1,mid+1,r,x,y,k);
        val[rt] = val[rt<<1] + val[rt<<1|1];
    }
}qxz;
int main(){
    N = read(), M = read();
    for(int i = 1; i <= N; ++i){
        a[i] = read();
    }
    while(M--){
        opt = read();
        if(opt == 1){
            l = read(), r = read(), x = read(), y = read();
            qxz.update(1,1,N,l,l,x);
            qxz.update(1,1,N,l+1,r,y);
            qxz.update(1,1,N,r+1,r+1,-(x+y*(r-l)));
        }
        else{
            x = read();
            printf("%d\n", qxz.query(1,1,N,1,x)+a[x]);
        }
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/qixingzhi/p/9747787.html
今日推荐