【hdu4267 A Simple Problem with Integers】 线段树之区间操作

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4267

 

题目大意：给你一列数区间范围为[1，n]，区间每个数有一个对应值a[i]。接下来有Q个操作

操作1： "1 a b k c"， 区间[a，b]内满足条件（i-a）%k==0的数值a[i]加c。

操作2：“2 a” ，输出a[i]的值。

 

解题思路：这题蛋都碎成渣渣了，RE几次，MLE无数次。

1、按线段树建树情况可推出，即使成为完全平衡树开也只有2*n-1个节点，开2*n的大小完全没问题啊，但它就是RE，不解。

2、这题我的初始化开始没有建build（）函数，而是直接对flag[][]初始化，这里MLE无数次，建树开build（）函数成段成段初始化就过了，蛋都碎成渣渣了。

 

说说解题思路吧，重点在（i-a）%k==0，这样的i是断开的，无法成段操作，傻×才去暴力。注意到k比较小，枚举k以及余数的组合，k=1有1种，k=2有2种，……k=10有10种，所以总共有（1+2+3....+10=55种组合），我们将每种组合压缩成一个状态。

建树操作： 成段初始化，不然MLE。

更新操作： 当查询到对应区间[tl，tr]，对应的组合状态flag[u][mo]加上c，返回。

查询操作：对x路过的区间进行操作，当x与k（1<=k<=10）对应的余数值这个状态有值时，加上这个值。

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

#define lz 2*u,l,mid
#define rz 2*u+1,mid+1,r
const int maxn=500005;
int flag[4*maxn][55], val[maxn];

void build(int u, int l, int r)  ///开始没这样建树初始化，而是直接memset(flag,0,sizeof(flag))，一直MLE，蛋都碎成渣渣了
{
    memset(flag[u],0,sizeof(flag[u]));
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(lz);
    build(rz);
}

void Update(int u, int l, int r, int tl, int tr, int mo, int k, int c)
{
    if(tl<=l&&r<=tr)
    {
        for(int i=1; i<k; i++) mo+=i;
        flag[u][mo]+=c;
        return ;
    }
    int mid=(l+r)>>1;
    if(tr<=mid) Update(lz,tl,tr,mo,k,c);
    else if(tl>mid) Update(rz,tl,tr,mo,k,c);
    else
    {
        Update(lz,tl,mid,mo,k,c);
        Update(rz,mid+1,tr,mo,k,c);
    }
}

void Query(int u, int l, int r, int x, int &res)
{
    for(int i=1; i<=10; i++)
    {
        int mo=x%i;
        for(int j=1; j<i; j++) mo+=j;
        res+=flag[u][mo];
    }
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(x<=mid) Query(lz,x,res);
    else Query(rz,x,res);
}

int main()
{
    int n, Q;
    while(~scanf("%d",&n))
    {
        for(int i=1; i<=n; i++) scanf("%d",val+i);
        int op, a, b, k, c;
        build(1,1,n);
        scanf("%d",&Q);
        while(Q--)
        {
            scanf("%d",&op);
            if(op==1)
            {
                scanf("%d%d%d%d",&a,&b,&k,&c);
                Update(1,1,n,a,b,a%k,k,c);
            }
            else
            {
                scanf("%d",&a);
                int res=0;
                Query(1,1,n,a,res);
                printf("%d\n",res+val[a]);
            }
        }
    }
    return 0;
}