HDU - 4267 A Simple Problem with Integers(线段树)

点我看题

题意：给n个数，两个操作，1 a b k c，对于区间a<=i<=b,对于满足（i-a）%k==0的i，val[i]加上c，2 a，查询修改之后的val[a]的值。

分析：暴力肯定会T，比较容易联想到线段树，那么如何利用线段树呢?因为k的值很小（1<=k<=10），那么我们可以知道所有数模不同的k一共可以得到55种不同的情况。k=1 1种（0），k=2 2种（0,1），……，k=10 10种（0,1,2，……，9），共55种。开始的时候，我们用一个二维数组b[11][11]记录模不同的k的的到不同结果的‘位置’，在更新的时候，利用这个位置和线段树中add[55]的下标的对应关系进行更新，查询也类似。

参考代码：

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>

using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lson rt<<1
#define rson rt<<1|1
const int maxn = 5e4+10;
int n,q;
int val[maxn];
struct SegTree{
    int l,r;
    int add[55];
    int sum;
};
SegTree st[maxn<<2];
int b[11][11];

void Build( int l, int r, int rt)
{
    st[rt].l = l;
    st[rt].r = r;
    mem(st[rt].add,0);
    st[rt].sum  = 0;
    if( l == r)
        return;
    int mid = (l+r)>>1;
    Build(l,mid,lson);
    Build(mid+1,r,rson);
}

void PushDown( int rt)
{
    if( st[rt].sum)
    {
        st[lson].sum += st[rt].sum;
        st[rson].sum += st[rt].sum;
        st[rt].sum = 0;
        for( int i = 0; i < 55; i++)
        {
            st[lson].add[i] += st[rt].add[i];
            st[rson].add[i] += st[rt].add[i];
            st[rt].add[i] = 0;
        }
    }
}

//更新
void Update( int L, int R, int rt, int val, int mod, int tmp)
{
    if( L == st[rt].l && R == st[rt].r)
    {
        st[rt].sum += val;
        st[rt].add[b[mod][tmp]] += val;
        return;
    }
    PushDown(rt);
    int mid = (st[rt].l+st[rt].r)>>1;
    if( R <= mid)
        Update(L,R,lson,val,mod,tmp);
    else if( L > mid)
        Update(L,R,rson,val,mod,tmp);
    else
    {
        Update(L,mid,lson,val,mod,tmp);
        Update(mid+1,R,rson,val,mod,tmp);
    }
//    PushUp(rt);
}

//查询
int Query( int rt, int pos)
{
    if( st[rt].l == st[rt].r && st[rt].l == pos)
    {
        int tmp = 0;
//        for( int i = 0; i < 55; i++)
        for( int i = 1; i <= 10; i++)
            tmp += st[rt].add[b[i][pos%i]];
        return val[pos]+tmp;
    }

    PushDown(rt);
    int mid = (st[rt].l+st[rt].r)>>1;
    if( pos <= mid)
        Query(lson,pos);
    else
        Query(rson,pos);
}

int main()
{
    while( ~scanf("%d",&n))
    {
        int cnt = 0;
        for( int i = 1; i <= 10; i++)
            for( int j = 0; j < i; j++)
                b[i][j] = cnt++;
        for( int i = 1; i <= n; i++)
            scanf("%d",&val[i]);
        Build(1,n,1);
        scanf("%d",&q);
        while( q--)
        {
            int op,a,b,c,k;
            scanf("%d",&op);
            if( op == 1)
            {
                scanf("%d%d%d%d",&a,&b,&k,&c);
                Update(a,b,1,c,k,a%k);
            }
            else if( op == 2)
            {
                scanf("%d",&a);
                printf("%d\n",Query(1,a));
            }
        }
    }

    return 0;
}