HDU 6315 Naive Operations [线段树]

题意

有两个数列a和b,a的初始值为0，b数组是1~n的一个排列，有两个操作：
1. 将a数组区间[l,r]的每个数+1。
2. 输出$\Sigma 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));
        }
    }
}