P1438 无聊的数列

题目要求区间修改和单点查询,不难想到线段树,区间修改为区间从左到右依次加一个等差数列,则可以用线段树+差分解决,使用差分数组则可以实现区间加同一个数来实现区间加等差数列

eg:初始数组:0 0 0 0 0

等差数列:首项1,公差2,从下标1加到4

操作之后的序列:1 3 5 7 0

差分数组:1 2 2 2 -7

所以只需要将l位置加首项,l+1到r区间加等差,r+1减去末尾项即可实现区间加等差数列操作

对于单点查询只需要查询前缀和即可

注意避免数组越界问题

ac代码:

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
#define pb push_back
#define int long long
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef pair<int,int> pii;
const int N=1e5+10;
struct Tree
{
    int l,r;
    int sum,lazy;
}tr[N<<2];
int w[N];
int n,m,op,l,r,k,d;
void change(int u,int lazy)
{
    tr[u].sum+=(tr[u].r-tr[u].l+1)*lazy;
    tr[u].lazy+=lazy;
}
void pushdown(int u)
{
    if(tr[u].lazy)
    {
        change(u<<1,tr[u].lazy);
        change(u<<1|1,tr[u].lazy);
        tr[u].lazy=0;
    }
}
void pushup(int u)
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void build(int u,int l,int r)
{
    if(l==r)tr[u]={l,r,w[l]-w[l-1],0};
    else 
    {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        pushup(u);
    }
}
void modify(int u,int l,int r,int k)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        change(u,k);
        return ;
    }
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    if(l<=mid)modify(u<<1,l,r,k);
    if(r>mid)modify(u<<1|1,l,r,k);
    pushup(u);
}
int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    int res=0;
    if(l<=mid)res+=query(u<<1,l,r);
    if(r>mid)res+=query(u<<1|1,l,r);
    return res;
}
void solve()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>w[i];
    build(1,1,n);
    while(m--)
    {
        cin>>op;
        if(op==1)
        {
            cin>>l>>r>>k>>d;
            modify(1,l,l,k);
            if(l+1<=r)modify(1,l+1,r,d);
            if(r<n)modify(1,r+1,r+1,-1*(k+(r-l)*d));
        }
        else 
        {
            cin>>l;
            cout<<query(1,1,l)<<endl;
        }
    }
}
signed main()
{
    Mirai;
    int T=1;
    //cin>>T;
    while(T--)
    {
        solve();
    }
}

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值