UESTC 1292 卿学姐种花 (分块)

题意:给出一个长度为n的序列,m次操作,操作分为两种:
1.在x的位置上加上数y,在x+1的位置加上数y-1,以此类推,一直到n或者y为0.
2.询问x位置上数字为多少

分析:乍一看像是线段树,但是貌似又很难用线段树解决,主要是区间不好处理,而分块对于这类题目有优势,在区间上更容易计数,对于每个块,对于1操作,如果某个块全覆盖,记录两个值,增值和差值,通过加加减减就能得到答案,如果不是全覆盖直接 sqrt(n) 处理值,复杂度 O(msqrt(n))

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
//#pragma comment(linker, "/STACK:1024000000,1024000000");

using namespace std;

#define INF 0x3f3f3f3f
#define maxn 20004

long long mod = 772002+233;
long long a[maxn];
int n,q;
int belong[maxn],l[maxn],r[maxn],block,num;
int val[maxn],del[maxn];

void init()
{
    block=sqrt(n);
    num=n/block;
    if(n%block) num++;
    for(int i=1;i<=n;i++)
    {
        belong[i]=(i-1)/block+1;
    }
    for(int i=1;i<=num;i++)
    {
        l[i]=block*(i-1)+1;
        r[i]=block*i;
        val[i]=del[i]=0;
    }
}

void update(int x,int y,int num)
{
    if(belong[x]==belong[y])
    {
        for(int i=x;i<=y;i++)
        {
            a[i]+=num--;
            a[i]%=mod;
        }
        return ;
    }
    for(int i=x;i<=r[belong[x]];i++)
    {
        a[i]+=num--;
        a[i]%=mod;
    }
    for(int i=belong[x]+1;ifor(int i=l[belong[y]];i<=y;i++)
    {
        a[i]+=num--;
        a[i]%=mod;
    }
}

long long query(int x)
{
    long long temp=a[x]+val[belong[x]]-(x-l[belong[x]])*del[belong[x]];
    if(temp<0)
    {
        temp+=(temp/mod)*mod;
        while(temp<0) temp+=mod;
    }
    return temp%mod;
}


int main()
{
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        for(int i=1; i<=n; i++)
        {
            scanf("%lld",&a[i]);
        }
        init();
        while(q--)
        {
            int op;
            scanf("%d",&op);
            if(op==1)
            {
                int l,r;
                scanf("%d%d",&l,&r);
                update(l,min(l+r-1,n),r);
            }
            else
            {
                int x;
                scanf("%d",&x);
                printf("%lld\n",query(x)%mod);
            }
        }
    }
    return 0;
}

你可能感兴趣的:(===数据结构===,分块)