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
#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 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>l;
            cout<>T;
    while(T--)
    {
        solve();
    }
}

你可能感兴趣的:(线段树,算法,数据结构)