华华开始学信息学(分块思想+树状数组)

最新推荐文章于 2024-09-15 11:40:06 发布

要用bug来打败bug

最新推荐文章于 2024-09-15 11:40:06 发布

阅读量259

点赞数

分类专栏：笔记文章标签： c++ acm竞赛算法

本文链接：https://blog.csdn.net/g3125976202/article/details/123983367

版权

笔记专栏收录该内容

18 篇文章 0 订阅

订阅专栏

本文介绍了一种高效的算法，用于在一个区间内快速计算一个数及其倍数的数量，通过将修改操作区分为大于sqrt(n)的直接树状数组操作和小于sqrt(n)的预先记录后查询时计算两部分，实现了O(sqrt(n))的查询速度。适合解决涉及大量倍数查找的问题。

摘要由CSDN通过智能技术生成

题目链接

思路：

区间一个数的倍数的个数，我们可以 O(1) 算出的，问题在于要找的倍数有很多，考虑分块思想，把修改操作分为两类。

大于 sqrt(n) ：

直接在树状数组中暴力 add

小于 sqrt(n) ：

先记录下来，在每次查询操作时暴力计算

时间复杂度：

一次查询约是 O(sqrt(n))

Code:

#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <iomanip>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
//#include <unordered_map>
#define guo312 std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define Inf LONG_LONG_MAX
#define inf INT_MAX
#define endl "\n"
#define PI 3.1415926535898
using namespace std;
const int N=1e5+10;
int n,m,lk;
ll c[N];
int lowbit(int x){
	return x&(-x);
}
void add(int x,ll num){
	for( ;x<=n;c[x]+=num,x+=lowbit(x));
}
ll query(int x){
	ll re=0;
	for( ;x>0;re+=c[x],x-=lowbit(x));
	return re;
}
ll lazy[N];
int main(){
guo312;
	cin>>n>>m; lk=sqrt(n);
	ll op,l,r,d,k;
	for(int i=1;i<=m;i++){
		cin>>op;
		if(op==1){
			cin>>d>>k;
			if(d<lk){
				lazy[d]+=k;
			}
			else{
				for(int j=d;j<=n;j+=d){
					add(j,k);
				}
			}
		}
		else{
			cin>>l>>r;
			ll ans=query(r)-query(l-1);
			for(int i=1;i<lk;i++){
				ans+=(r/i-(l-1)/i)*lazy[i];
			}
			cout<<ans<<endl;
		}
	}
	return 0;
}