摘苹果（线段树）---2023年天梯赛（GPLT）上海理工大学校内选拔赛-H

解析

线段树，维护每个点的数量，同时维护小于 $100$ 和小于 $10$ 的个数

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define inf 0x3f3f3f3f
#define L u<<1
#define R u<<1|1
const int N=1e5+5;
int n,m;
int a[N];
struct node{
	int l,r;
	int sum,cnt,flag;
}tr[N*4];
void pushup(int u){
	tr[u].sum=tr[L].sum+tr[R].sum;
	tr[u].cnt=tr[L].cnt+tr[R].cnt;
	tr[u].flag=tr[L].flag&tr[R].flag;
}
void build(int u,int l,int r){
	tr[u]={l,r,0,0};
	if(l==r){
		tr[u].sum=a[l];
		tr[u].cnt=(a[l]<100);
		tr[u].flag=(a[l]<10);
		return;
	}
	int mid=l+r>>1;
	build(L,l,mid);
	build(R,mid+1,r);
	pushup(u);
}
//查询总数 op=3
int querySum(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r){
		return tr[u].sum;
	}
	int mid=tr[u].l+tr[u].r>>1;
	int s=0;
	if(l<=mid) s+=querySum(L,l,r);
	if(r>mid) s+=querySum(R,l,r);
	return s;
}
//op=2
int query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r){
		return tr[u].cnt;
	}
	int mid=tr[u].l+tr[u].r>>1;
	int s=0;
	if(l<=mid) s+=query(L,l,r);
	if(r>mid) s+=query(R,l,r);
	return s;
}
void modify(int u,int l,int r){
	if(tr[u].flag) return;
	if(tr[u].l==tr[u].r){
		if(tr[u].sum>=10){
//			tr[u].sum=tr[u].sum/3*2;
			tr[u].sum-=(tr[u].sum-1)/3+1;
		}
		tr[u].cnt=(tr[u].sum<100);
		tr[u].flag=(tr[u].sum<10);
		return;
	}
	int mid=tr[u].l+tr[u].r>>1;
	if(l<=mid) modify(L,l,r);
	if(r>mid) modify(R,l,r);
	pushup(u);
}
void solve(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,1,n);
	for(int i=1;i<=m;i++){
		int op,x,y;
		cin>>op>>x>>y;
		if(op==1){
			modify(1,x,y);		
		}
		else if(op==2){
			cout<<query(1,x,y)<<endl;
		}
		else{
			cout<<querySum(1,x,y)<<endl; 
		}
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int tt=1;
//	cin>>tt;
	while(tt--) solve();
	return 0;
}