码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 4267 A Simple Problem with Integers --树状数组

时间:2014-09-12 18:58:03      阅读:142      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   io   os   ar   for   art   

题意:给一个序列,操作1:给区间[a,b]中(i-a)%k==0的位置 i 的值都加上val  操作2:查询 i 位置的值

解法:树状数组记录更新值。 由 (i-a)%k == 0 得知 i%k == a%k,又因为k <= 10,想到建55棵树状数组,即对每个(k,x%k)都建一棵树状数组,每次更新时,在第(k,a%k)棵树状数组上更新a这个点,更新值为val,然后再b+1处更新值为-val,即在[a,b]内更新了val。

查询pos的时候,求出每一个树状数组(k,pos%k)的sum值即可。

代码:

bubuko.com,布布扣
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 50007

int c[N][11][11];
int n,ka[N];

int lowbit(int x){ return x & (-x); };

void modify(int pos,int k,int mod,int val)
{
    if(pos == 0) return;
    while(pos <= n)
    {
        c[pos][k][mod] += val;
        pos += lowbit(pos);
    }
}

int getsum(int pos)
{
    int res = 0;
    int tmp = pos;
    while(pos > 0)
    {
        for(int i=1;i<=10;i++)
            res += c[pos][i][tmp%i];
        pos -= lowbit(pos);
    }
    return res;
}

int main()
{
    int m,a,b,k,val,i,j;
    int op;
    while(scanf("%d",&n)!=EOF)
    {
        for(i=1;i<=n;i++)
            scanf("%d",&ka[i]);
        memset(c,0,sizeof(c));
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d",&op);
            if(op == 1)
            {
                scanf("%d%d%d%d",&a,&b,&k,&val);
                modify(a,k,a%k,val);
                modify(b+1,k,a%k,-val);
            }
            else
            {
                scanf("%d",&a);
                printf("%d\n",ka[a] + getsum(a));
            }
        }
    }
    return 0;
}
View Code

 

HDU 4267 A Simple Problem with Integers --树状数组

标签:style   blog   http   color   io   os   ar   for   art   

原文地址:http://www.cnblogs.com/whatbeg/p/3968827.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!