题意
有两个数列a和b,a的初始值为0,b数组是1~n的一个排列,有两个操作:
1. 将a数组区间[l,r]的每个数+1。
2. 输出
Σa[i]/b[i](l<=i<=r)
Σ
a
[
i
]
/
b
[
i
]
(
l
<=
i
<=
r
)
。
题解
由于操作数一共10^5,所以对于每一个位置i的贡献,我们顶多更新n/b[i]次,所以总的更新次数是n*log(n),我们通过线段树进行区间余数最小值的维护。询问的时候,假如询问区间内,线段树上当前区间的余数最小值<=0,说明需要更新答案,然后找到需要更新的叶子节点进行更新,所以每次更新的复杂度为O(log(n)),因此算法的总复杂度为O(n*logn*logn)。
AC代码
#include<stdio.h>
#include<iostream>
#define N 100005
using namespace std;
int a[N],tree[N*4],mi[N*4],add[N*4];
char op[30];
void build(int L,int R,int root)
{
mi[root]=N;
tree[root]=add[root]=0;
if(L==R)
{
mi[root]=a[L];
return ;
}
int mid=L+R>>1;
build(L,mid,root<<1);
build(mid+1,R,root<<1|1);
mi[root]=min(mi[root<<1],mi[root<<1|1]);
}
void pushdown(int root)
{
mi[root<<1]-=add[root];
mi[root<<1|1]-=add[root];
add[root<<1]+=add[root];
add[root<<1|1]+=add[root];
add[root]=0;
}
void update(int l,int r,int L,int R,int root)
{
if(l<=L&&R<=r)
{
add[root]++;
mi[root]--;
return ;
}
if(add[root])pushdown(root);
int mid=L+R>>1;
if(r<=mid)update(l,r,L,mid,root<<1);
else if(l>mid)update(l,r,mid+1,R,root<<1|1);
else
{
update(l,mid,L,mid,root<<1);
update(mid+1,r,mid+1,R,root<<1|1);
}
tree[root]=tree[root<<1]+tree[root<<1|1];
mi[root]=min(mi[root<<1],mi[root<<1|1]);
}
void update(int L,int R,int root)
{
if(L==R)
{
int val=(-mi[root]/a[L])+1;
tree[root]+=val;
mi[root]+=val*a[L];
return ;
}
if(add[root])pushdown(root);
int mid=L+R>>1;
if(mi[root<<1]<=0)update(L,mid,root<<1);
if(mi[root<<1|1]<=0)update(mid+1,R,root<<1|1);
tree[root]=tree[root<<1]+tree[root<<1|1];
mi[root]=min(mi[root<<1],mi[root<<1|1]);
}
int query(int l,int r,int L,int R,int root)
{
if(l<=L&&R<=r)
{
if(mi[root]<=0)update(L,R,root);
return tree[root];
}
if(add[root])pushdown(root);
int mid=L+R>>1;
if(r<=mid)return query(l,r,L,mid,root<<1);
else if(l>mid)return query(l,r,mid+1,R,root<<1|1);
else return query(l,mid,L,mid,root<<1)+query(mid+1,r,mid+1,R,root<<1|1);
tree[root]=tree[root<<1]+tree[root<<1|1];
mi[root]=min(mi[root<<1],mi[root<<1|1]);
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,1);
for(int i=0;i<m;i++)
{
int l,r;
scanf("%s%d%d",op,&l,&r);
if(op[0]=='a')update(l,r,1,n,1);
else printf("%d\n",query(l,r,1,n,1));
}
}
}