题目大意:给一数列,有两种操作,一种是在[a, b]内,对(i - a) % k == 0 的所有值加上c,另一种是查询某个位置的值。
思路:由于本题k<=10,所以一共有1+2+3+...+10 = 55种余数,所以可以开55个树状数组,用sum[k][mod][i]表示i % k == mod,便于在询问的时候减少循环次数。
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
using namespace std;
const int maxn = 55555;
int aa[maxn];
int sum[11][11][maxn];
int n, q;
int lowbit(int x)
{
return x & (-x);
}
void modify(int pos, int k, int mod, int val)
{
for(int i=pos; i>0; i-=lowbit(i))
{
sum[k][mod][i] += val;
}
}
int query(int pos, int k)
{
int res = 0;
for(int i=pos; i<=n; i+=lowbit(i))
{
for(int j=1; j<=10; j++)
res += sum[j][k%j][i];
}
return res;
}
int main()
{
while(scanf("%d", &n) != EOF)
{
for(int i=1; i<=n; i++)
{
scanf("%d", &aa[i]);
}
memset(sum, 0, sizeof sum);
scanf("%d", &q);
while(q--)
{
int op;
int a, b, k, c;
scanf("%d", &op);
if(op == 1)
{
scanf("%d%d%d%d", &a, &b, &k, &c);
modify(b, k, a%k, c);
modify(a-1, k, a%k, c*(-1));
}
else
{
scanf("%d", &a);
int ans = query(a, a);
printf("%d\n", ans + aa[a]);
}
}
}
return 0;
}