一个简单的整数问题2-分块算法

一个简单的整数问题2–分块算法

---

题目描述

---

核心思路

这题可以采用分块思想，如果想要了解分块思想，请移步这篇文章分块基本思想

具体思想详见注释

---

代码

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1e5+10;
typedef long long LL;
//a数组是存储每个数的序列 add[i]+=d表示将第i块中的所有数都+d 这是一个懒标记
//sum[i]记录的是第i块中所有数的总和
LL a[N],add[N],sum[N];
//p=pos[i]表示下标i所在的块是第p块
//L[i]记录的是第i块的左端点   R[i]记录的是第i块的右端点
int pos[N],L[N],R[N];
int n,m;

//预处理
void init()
{
    int t= sqrt(n*1.0); //t表示每个块内的长度(即元素个数)
    int num=n/t;    //num记录的是有多少个块
    //这里处理的是最后剩余的那一小部分 让它自己成为独立的一块
    //比如n=10,t=3,那么分成n/t=10/3=3块后，我们发现第10个并不属于前面的3块
    //因此它必须新开一个块
    if(n%t)
        num++;
    //分别处理这num个块的左端点和右端点
    for(int i=1;i<=num;i++) //i遍历的是块号
    {
        L[i]=(i-1)*t+1; //第i块的左端点
        R[i]=i*t;   //第i块的右端点
    }
    //这里要注意 当上面处理完第num块时 R[num]是按照长度为t进行分配的右端点
    //但是这个右端点有可能比n还大 然而我们只需要处理到n就可以了
    //因此这里还要特殊处理最后一段的右端点 取到n即可 不一定取最后一段分配到的右端点
    R[num]=n;
    //遍历这num个块 预处理出下标j是属于哪个块  同时预处理出这个块内的总和
    for(int i=1;i<=num;i++) //i遍历的是块
    {
        for(int j=L[i];j<=R[i];j++)//j遍历的是这个块的元素
        {
            pos[j]=i;   //下标j所对应的元素应该属于块号i
            sum[i]+=a[j];   //记录第i块的元素总和
        }
    }
}

//区间修改  将区间[l,r]中的元素都+d
void change(int l,int r,int d)
{
    int p=pos[l];   //获取下标l所在的块号p
    int q=pos[r];   //获取下标r所在的块号q
    //如果区间[l,r]在同一个块内
    if(p==q)
    {
        //直接将区间[l,r]内的所有元素都+d
        for(int i=l;i<=r;i++)
            a[i]+=d;
        sum[p]+=(r-l+1)*d;  //记录第p块内的所有元素的总和
    }
    //否则说明区间[l,r]跨越了不同的块
    else
    {
        //先处理中间被完全覆盖的块 让这些块的懒标记都+d
        for(int i=p+1;i<=q-1;i++)//i枚举的中间被完全覆盖的块号
            add[i]+=d;
        //处理左边剩余部分 将区间[l,R[p]]内的所有元素都+d
        for(int i=l;i<=R[p];i++)
            a[i]+=d;
        sum[p]+=(R[p]-l+1)*d;   //记录左边剩余部分的所有元素的总和
        //处理右边剩余部分 将区间[L[q],r]内的所有元素都+d
        for(int i=L[q];i<=r;i++)
            a[i]+=d;
        sum[q]+=(r-L[q]+1)*d;   //记录右边剩余部分的所有元素的总和
    }
}

//区间查询
LL query(int l,int r)
{
    int p=pos[l];   //获取下标l所在的块号p
    int q=pos[r];   //获取下标r所在的块号q
    LL ans=0;   //记录区间[l,r]中的元素总和
    //如果区间[l,r]在同一个块内
    if(p==q)
    {
        //累加这个区间[l,r]中所有元素的总和
        for(int i=l;i<=r;i++)
            ans+=a[i];
        ans+=(r-l+1)*add[p];    //同时要记得加上第p块中的懒标记
    }
    //否则说明区间[l,r]跨越了不同的块
    else
    {
        //先累加中间被完全覆盖的这些块的懒标记
        for(int i=p+1;i<=q-1;i++)   //i遍历的是中间被完全覆盖的这些块的块号
            ans+=sum[i]+(R[i]-L[i]+1)*add[i];
        //累加左边剩余部分的所有元素的总和
        for(int i=l;i<=R[p];i++)
            ans+=a[i];
        //累加左侧剩余部分的第p块中的懒标记
        ans+=(R[p]-l+1)*add[p];
        //累加左=右边剩余部分的所有元素的总和
        for(int i=L[q];i<=r;i++)
            ans+=a[i];
        //累加右侧剩余部分的第q块中的懒标记
        ans+=(r-L[q]+1)*add[q];
    }
    return ans; //返回答案
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    init();     //预处理
    while(m--)
    {
        char op[2];
        int l,r,d;
        scanf("%s%d%d",op,&l,&r);
        if(*op=='C')    //区间修改
        {
            scanf("%d",&d);
            change(l,r,d);
        }
        else        //区间查询
            printf("%lld\n",query(l,r));
    }
    return 0;
}