牛客小白月赛12 F-华华开始学信息学(树状数组+平方分割思想)

题目

思路来源

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=40397065

题解

平方分割思想，

如果D很大，我们暴力BIT单点修改就是

D很小的话，就把它塞进D的桶里，查询的时候查[x,y]有多少数是D的倍数即可

由于有D个桶，所以每次查询得遍历所有桶，O(D)

总复杂度，令D取，均摊复杂度有

即D小的话就塞进桶里，大的话就暴力修改，

查询的时候复杂度，也就是

数论做过这种拆两半的题，数据结构还是刷少了吖见识不够

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
ll tree[maxn],sum[maxn];
int n,m,b;
int op,x,y;
ll ans;
void add(int x,int y)
{
	for(int i=x;i<=n;i+=i&-i)
	tree[i]+=y;
}
ll query(int x)
{
	ll sum=0;
	for(int i=x;i>0;i-=i&-i)
	sum+=tree[i];
	return sum;
}
int main()
{
	scanf("%d%d",&n,&m);
	b=sqrt(n*log(n));
	while(m--)
	{
		scanf("%d%d%d",&op,&x,&y);
		if(op==1)
		{
			if(x<=b)sum[x]+=y;
			else
			{
				for(int i=x;i<=n;i+=x)
				add(i,y);
			}
		}
		else if(op==2)
		{
			ans=0;
			ans+=query(y)-query(x-1);
			for(int i=1;i<=b;++i)
			ans+=sum[i]*((y/i)-(x-1)/i);
			printf("%lld\n",ans);
		}
	}
	return 0;
}