一维数状数组区间修改,查询

模板题CODEVS-1082

给你N个数,有两种操作:

1:给区间[a,b]的所有数增加X

2:询问区间[a,b]的数的和。             

第一行一个正整数n,接下来n行n个整数,

再接下来一个正整数Q,每行表示操作的个数,

如果第一个数是1,后接3个正整数,

表示在区间[a,b]内每个数增加X,如果是2,

表示操作2询问区间[a,b]的和是多少。

一维树状数组可以考虑用差分来做,但是扩展不到二维。

所以我们令di=(ai~an)的增量

思路和差分一样

复制代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
#define lowbit(x) x&(-x)
int n,m,a[1000005],c1[1000005],c2[1000005];
void update(int v,int val)
{
    for(int pp=v;pp<=n;c1[pp]+=val,c2[pp]+=val*v,pp+=lowbit(pp));
}
int query(int l,int r)
{
    int ans=0;
    for(int pp=r;pp>0;ans+=(r+1)*c1[pp],ans-=c2[pp],pp-=lowbit(pp));
    l--;
    for(int pp=l;pp>0;ans-=(l+1)*c1[pp],ans+=c2[pp],pp-=lowbit(pp));
    return ans;
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
    update(i,a[i]-a[i-1]);
    
    for(int i=1;i<=m;i++)
    {
        int p,l,r,val;
        scanf("%d %d %d",&p,&l,&r);
        if(p==1)
        {
            scanf("%d",&val);
            update(l,val);
            update(r+1,-val);
        }else
        {
            printf("%d\n",query(l,r));
        }
    }
            
    return 0;
}
复制代码

 

 

 

 

 

posted @   dancer16  阅读(255)  评论(0编辑  收藏  举报
编辑推荐:
· .NET制作智能桌面机器人:结合BotSharp智能体框架开发语音交互
· 软件产品开发中常见的10个问题及处理方法
· .NET 原生驾驭 AI 新基建实战系列:向量数据库的应用与畅想
· 从问题排查到源码分析:ActiveMQ消费端频繁日志刷屏的秘密
· 一次Java后端服务间歇性响应慢的问题排查记录
阅读排行:
· 互联网不景气了那就玩玩嵌入式吧,用纯.NET开发并制作一个智能桌面机器人(四):结合BotSharp
· Vite CVE-2025-30208 安全漏洞
· 《HelloGitHub》第 108 期
· MQ 如何保证数据一致性?
· 一个基于 .NET 开源免费的异地组网和内网穿透工具
描述
点击右上角即可分享
微信分享提示