题意:对一段序列进行查询和修改。其中有限制条件k。
修改i ([a, b])序列时, 要满足 (i - a) % k == 0。换句话说,就是要每k个数后修改。那么我们只需要保证 % k的值相同。
#include<cstdio>
#include<cstring>
const int maxn = 50000 +5;
int tree[11][11][maxn], n;
int A[maxn];
void Add(int k, int mod, int a, int val)
{
while(a <= n)
{
tree[k][mod][a] += val;
a += a & -a;
}
}
int Query(int k, int mod, int a)
{
int ret = 0;
while(a > 0)
{
ret += tree[k][mod][a];
a -= a & -a;
}
return ret;
}
int main()
{
while(scanf("%d", &n) == 1)
{
for(int i = 1; i <= n; i++)
scanf("%d", &A[i]);
memset(tree, 0, sizeof(tree));
int Q;
scanf("%d", &Q);
while(Q--)
{
int cmd;
scanf("%d", &cmd);
if(cmd == 1)
{
int a, b, k, c;
scanf("%d%d%d%d", &a, &b, &k, &c);
// a -> b 从a开始,每隔k个数,修改。就是 %k的值相同、
Add(k, a % k, a, c);
Add(k, a % k, b + 1, -c);
}
else if(cmd == 2)
{
int a;
scanf("%d", &a);
int ans = A[a];
for(int k = 1; k <= 10; k++)
{
ans += Query(k, a % k, a);
}
printf("%d\n", ans);
}
}
}
return 0;
}