分块维护和板子

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
using ll = long long;
ll s[N];
ll b[N];
ll w[N];
ll add[N];
int l[N],r[N];
int tot,n,len,q;
int belong[N];


inline void init()
{
	len = sqrt(n),tot = (n+len-1)/len;
	for(int i=1;i<=tot;i++)
	 l[i] = r[i-1]+1,r[i] = i*len;
	 
	for(int i=1;i<=tot;i++){
		for(int j=l[i];j<=r[i];j++)
		 b[i]+=w[j],belong[j] = i;
		s[i] = s[i-1] + b[i];
	}
	
}

inline void modify(int ql,int qr,int c){
	int p = belong[ql],q = belong[qr];
	if(p==q){
		for(int i=ql;i<=qr;i++)w[i]+=c,b[p]+=c;
		for(int i=p;i<=tot;i++)s[i] = s[i-1]+b[i];
		return;
	}
	
	for(int i=ql;i<=r[p];i++)w[i]+=c,b[p]+=c;
	for(int i=qr;i>=l[q];--i)w[i]+=c,b[q]+=c;
	for(int i=p+1;i<=q-1;i++)add[i]+=c,b[i]+=(ll)(r[i]-l[i]+1)*c;
	
	for(int i=p;i<=tot;i++)s[i] = s[i-1]+b[i];
}



inline ll query(int ql,int qr){
	int p = belong[ql],q = belong[qr];
	ll res = 0;
	if(p==q){
		for(int i=ql;i<=qr;i++)res+=w[i]+add[p];
		return res;
	}
	
	res = s[q-1]-s[p];
	for(int i=ql;i<=r[p];++i)res+=w[i]+add[p];
	for(int i=qr;i>=l[q];--i)res+=w[i]+add[q];
	return res;
}



int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>q;
	for(int i=1;i<=n;i++)cin>>w[i];
	init();
	while(q--){
		int op,ql,qr;cin>>op>>ql>>qr;
		if(op==1){
			int c;
			cin>>c;modify(ql,qr,c);
		}
		else cout<<query(ql,qr)<<"\n";
	}
}
相关推荐
啊阿狸不会拉杆11 分钟前
《机器学习》第六章-强化学习
人工智能·算法·机器学习·ai·机器人·强化学习·ml
Stardep17 分钟前
算法入门20——二分查找算法——搜索插入位置
数据结构·算法·leetcode
qwerasda12385220 分钟前
青豆质量分类识别_YOLOv5_SPDConv_改进算法_目标检测_深度学习_计算机视觉
算法·计算机视觉·分类
老鼠只爱大米26 分钟前
LeetCode经典算法面试题 #141:环形链表(快慢指针、标记节点等多种方法详细解析)
算法·leetcode·链表·快慢指针·floyd算法·环形链表
alanesnape32 分钟前
什么是字面量?代码中的常量表示方式解析
算法
偷星星的贼1136 分钟前
C++中的访问者模式实战
开发语言·c++·算法
gjxDaniel1 小时前
A+B问题天堂版
c++·算法·字符串·字符数组
M__331 小时前
动态规划进阶:简单多状态模型
c++·算法·动态规划
未来之窗软件服务1 小时前
计算机等级考试—Dijkstra(戴克斯特拉)& Kruskal(克鲁斯卡尔)—东方仙盟
算法·计算机软考·仙盟创梦ide·东方仙盟
Hcoco_me1 小时前
大模型面试题89:GPU的内存结构是什么样的?
人工智能·算法·机器学习·chatgpt·机器人