题目链接不放,杭电一搜就有
线段树作为区间可加性信息的分治数据结构,通常用于解决具有可加性的区间信息统计,这玩意非常好用。
本题题目:要求维护的是
∑
i
=
L
R
⌊
A
i
B
i
⌋
\sum_{i=L}^{R}{\lfloor \frac{Ai}{Bi} \rfloor}
∑i=LR⌊BiAi⌋
更新操作只有[L,R]表示,区间内A的值+1;
,然后B保证是一个1到n的置换,即B大小为n,且1<=Bi<=n且Bi两两不同;
设定ai初始值全0;
解法
本题是非常灵活的线段树用法,可以说是铜牌题了;
容易想到:+1,相当于(Bi-Ai%Bi)-1,可以倒着维护一个Ci = (Bi - Ai%Bi)
然后当Ci变成0的时候让Ci再次变成Bi,然后更新这个点;
但是容易想到会超时;
不过这个题关键的地方在于:B是1-n的置换;
那么每一步均摊下来有多少个数要更新:
假设add(1,N);
Bi = 1的点要更新1次;
Bi = 2的点要更新1/2次;
Bi = 3的点要更新1/3次;
Bi = 4的点要更新1/4次;
如此得到调和级数(1+1/2+1/3+1/4……+1/N) = lnN;
好哇,每一步最多有lnN个点的Ai/Bi的值会+1而已,区间内如果没有要更新的,就打上lazy标志即可;
如何判断子树存在有数要增加,那就维护一个区间最小值;当_min[k] = 1说明子树在add操作后存在有点+1;
那我维护出一个数组求得是
∑
A
i
B
i
\sum{\frac{Ai}{Bi}}
∑BiAi即可;在更新_min的时候,把sum也维护出来;
时间复杂度介于log到log**2之间;
code:
#include <bits/stdc++.h>
const double eps = 1e-6;
const double PI = acos(-1);
const int INF = 0x3f3f3f3f;
#define ms(a,k) memset(a,k,sizeof(a))
#define X first
#define Y second
#define pii pair<int ,int >
#define lowbit(a) a&(-a)
typedef long long ll ;
typedef unsigned long long ull;
using namespace std;
inline void read(int &x){scanf("%d",&x);}
const int mod = 998244353;
const int maxn = 100000+10;
int _min[maxn*4],sum[maxn*4],b[maxn],lazy[maxn*4];//线段树的两个信息,其中lazy标志只用于_min因为sum不存在区间+;
int n,q;
void build(int k,int l,int r)
{
_min[k] = INF;sum[k] = 0;lazy[k] = 0;
if(l==r){_min[k] = b[l];return ;}
int mid = l+r>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
_min[k] = min(_min[k<<1],_min[k<<1|1]);
}
void down(int k)
{
lazy[k<<1] += lazy[k]; _min[k<<1] -= lazy[k];
lazy[k<<1|1] += lazy[k]; _min[k<<1|1] -= lazy[k];
lazy[k] = 0;
}
void add(int k,int l,int r,int L,int R)//[L,R]加1,要更新
{
if(_min[k]==1)
{//当前这个区间的_min要-1,于是就有添加;
if(l==r)
{
_min[k] = b[l]; sum[k]++;
return ;
}
down(k);
int mid = l+r>>1;
if(L<=mid){ add(k<<1,l,mid,L,R); }
if(mid+1<=R){ add(k<<1|1,mid+1,r,L,R); }
_min[k] = min(_min[k<<1],_min[k<<1|1]);
sum[k] = sum[k<<1]+sum[k<<1|1];
}
else
{
if(L<=l&&r<=R){lazy[k]++;_min[k]--;return ;}
down(k);
int mid = l+r>>1;
if(L<=mid){ add(k<<1,l,mid,L,R); }
if(mid+1<=R){ add(k<<1|1,mid+1,r,L,R); }
_min[k] = min(_min[k<<1],_min[k<<1|1]);
}
}
int query(int k,int l,int r,int L,int R)
{
if(L<=l&&r<=R){return sum[k];}
int mid = l+r>>1;
int res= 0;
if(L<=mid)res += query(k<<1,l,mid,L,R);
if(mid+1<=R)res += query(k<<1|1,mid+1,r,L,R);
return res;
}
void test (int k,int l,int r)
{//检错
if(l==r)printf("%d %d %d %d %d\n",k,l,r,_min[k],sum[k]);
else {int mid = l+r>>1;test(k<<1,l,mid);test(k<<1|1,mid+1,r);}
}
int main()
{
while(~scanf("%d%d",&n,&q))
{
for(int i =1;i<=n;++i)read(b[i]);
build(1,1,n);
while(q--)
{
//printf("debug\n");
char s[10];int L,R;
scanf("%s %d %d",s,&L,&R);
if(s[0] == 'a')add(1,1,n,L,R);
else printf("%d\n",query(1,1,n,L,R));
//test(1,1,n);
}
}
return 0;
}