黄老师的博客
题目传送门
有n个元素,如果我们把每m个元素分成一块,共有n/m块,每次区间加的操作会涉及O(n/m)个整块,以及区间两侧两个不完整的块中至多2m个元素。
我们给每个块设置一个加法标记(就是记录这个块中元素一起加了多少),每次操作对每个整块直接O(1)标记,而不完整的块由于元素比较少,暴力修改元素的值。
每次询问时返回元素的值加上其所在块的加法标记
这样总复杂度是O(n*(n/m)+n*m),根据均值不等式,
n*(n/m)+n*m>=2n根号n,当m取根号n时相等,复杂度最低。
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<malloc.h>
#include<math.h>
#include<string.h>
using namespace std;
typedef long long int ll;
#define int long long
const int N=50000+10;
ll read()
{
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,blo;
int v[N],bl[N],atag[N];
void add(int a,int b,int c)
{
for(int i=a;i<=min(bl[a]*blo,b);i++)//如果b在块a中则直接更新,若不在,则更新到a块的尾
v[i]+=c;
if(bl[a]!=bl[b])//若a和b不在同一块中,则从块b的开头更新到块b的尾
for(int i=(bl[b]-1)*blo+1;i<=b;i++)
v[i]+=c;
for(int i=bl[a]+1;i<=bl[b]-1;i++)
atag[i]+=c;//加法标记
}
signed main()
{
n=read();blo=sqrt(n);
for(int i=1;i<=n;i++)v[i]=read();
for(int i=1;i<=n;i++)bl[i]=(i-1)/blo+1;
for(int i=1;i<=n;i++)
{
int f=read(),a=read(),b=read(),c=read();
if(f==0)add(a,b,c);
if(f==1)printf("%d\n",v[b]+atag[bl[b]]);//最后v[b]的值再加上b块上的加法标记
}
return 0;
}