hdu 4267 A Simple Problem with Integers

最新推荐文章于 2022-12-01 00:10:51 发布

shiqi_614

最新推荐文章于 2022-12-01 00:10:51 发布

阅读量1.8k

点赞数

分类专栏：线段树及数据结构文章标签：数据结构 struct c 优化

本文链接：https://blog.csdn.net/shiqi_614/article/details/7985547

版权

线段树及数据结构专栏收录该内容

82 篇文章 1 订阅

订阅专栏

题意：给你两个操作，，“1 a b k c”表示在区间[a,b]内位置满足（i-a）%k=0的数加上c，k的范围在1到10之间。“2 a”表示查询第a个数。

首先明确，(i-a)%k=0等价于i%k=a%k，即i满足i+k满足，i+2*k也满足……。但是如果想用如线段树，数状数组等数据结构去优化，要求数必须是连续的一段的，所以思路就是所间隔的数整合在一起。设数状数组bit[i][j]表示当k等于i的时候，将位置pos满足pos%i==j的所有数放到这个树状数组里，那么就将所有的数按这个位置模k是多少，分别放到了[0.……k-1]的数状数组里。询问a位置的数，就等于bit[0...10]里这个数的和了。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long LL;
const int N=50005;
int org[N];
struct BIT
{
	int valu[N];
	int lowbit(int x){return x&(-x);}
	void clear(){memset(valu,0,sizeof(valu));}
	void updata(int st,int ed,int c)
	{
		for(int i=ed;i>=1;i-=lowbit(i)) 
			valu[i]+=c;
		for(int i=st-1;i>=1;i-=lowbit(i))
			valu[i]-=c;
	}
	int getvalu(int pos)
	{
		int sum=0;
		for(int i=pos;i<N;i+=lowbit(i))
			sum+=valu[i];
		return sum;
	}
}bit[11][11];
int main()
{
	int n,m;
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=0;i<=10;i++) 
			for(int j=0;j<=10;j++) bit[i][j].clear();
		for(int i=1;i<=n;i++) scanf("%d",&org[i]);

		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			int a,b,k,c,op;
			scanf("%d",&op);
			if(op==1)
			{
				scanf("%d%d%d%d",&a,&b,&k,&c);
				int st=a/k;
				int ed=a/k+(b-a)/k;
				bit[k][a%k].updata(st+1,ed+1,c);
			}
			else 
			{
				int sum=0;
				scanf("%d",&a);
				for(int i=1;i<=10;i++)
					sum+=bit[i][a%i].getvalu(a/i+1);
				printf("%d\n",sum+org[a]);
			}
		}
	}
	return 0;
}