[bzoj4293] Siano

Description

农夫Byteasar买了一片n亩的土地，他要在这上面种草。
他在每一亩土地上都种植了一种独一无二的草，其中，第i亩土地的草每天会长高a[i]厘米。
Byteasar一共会进行m次收割，其中第i次收割在第d[i]天，并把所有高度大于等于b[i]的部分全部割去。Byteasar想知道，每次收割得到的草的高度总和是多少，你能帮帮他吗？

Input

第一行包含两个正整数$n,m(1 \leq n,m \leq 500000)$，分别表示亩数和收割次数。
第二行包含$n$个正整数，其中第$i$个数为$a[i](1 \leq a[i] \leq 1000000)$，依次表示每亩种植的草的生长能力。
接下来m行，每行包含两个正整数$d[i],b[i](1 \leq d[i] \leq 10^{12}，0 \leq b[i] \leq 10^{12})$，依次描述每次收割。
数据保证$d[1]<d[2]<...<d[m]$，并且任何时刻没有任何一亩草的高度超过$10^{12}$。

Output

输出$m$行，每行一个整数，依次回答每次收割能得到的草的高度总和。

Sample Input

4 4
1 2 4 3
1 1
2 2
3 0
4 4

Sample Output

6
6
18
0

HINT

第1天，草的高度分别为1,2,4,3，收割后变为1,1,1,1。

第2天，草的高度分别为2,3,5,4，收割后变为2,2,2,2。

第3天，草的高度分别为3,4,6,5，收割后变为0,0,0,0。

第4天，草的高度分别为1,2,4,3，收割后变为1,2,4,3。

题解

好久没写这样比较复杂的数据结构题了（貌似还没写过），当时在考场上想出方法但是写挂了，下来以后写完又调了很久才A，看来代码能力下降了啊。
这道题在思维上不难，就是在单调序列上的线段树。注意到把草按生长速度从小到大排序后高度总是不下降的，所以可以二分找到高度大于$b$的最左端点，然后区间查询再赋值。至于草的生长，就是一个区间加，在实现上就是维护两个标记：区间赋值和区间加和区间和、区间最小高度。另外，注意一下pushdown时的顺序。

Code

#include<bits/stdc++.h>
#define ls a[u].ch[0]
#define rs a[u].ch[1]
using namespace std;

int n,m,root,cnt=0;
long long v[500010];
struct node{
    int ch[2];
    long long sum,ll,add,fz;
}a[1000010];

void update(int u){
    a[u].sum=a[ls].sum+a[rs].sum;
    a[u].ll=a[ls].ll;
}

void pushdown(int u,int l,int r)
{
    int mid=(l+r)/2;
    if(a[u].fz!=-1){
        a[ls].fz=a[rs].fz=a[u].fz;
        a[ls].add=a[rs].add=0;
        a[ls].sum=a[u].fz*(mid-l+1),a[rs].sum=a[u].fz*(r-mid);
        a[ls].ll=a[rs].ll=a[u].fz;
        a[u].fz=-1;
    }
    if(a[u].add){
        a[ls].add+=a[u].add,a[rs].add+=a[u].add;
        a[ls].sum+=a[u].add*(v[mid]-v[l-1]),a[rs].sum+=a[u].add*(v[r]-v[mid]);
        a[ls].ll+=(v[l]-v[l-1])*a[u].add,a[rs].ll+=(v[mid+1]-v[mid])*a[u].add;
        a[u].add=0;
    }
}

void build(int &u,int l,int r){
    u=++cnt;a[u].fz=-1;
    if(l==r)return;
    int mid=(l+r)/2;
    build(ls,l,mid);build(rs,mid+1,r);
    update(u);
}

int search(int u,int l,int r,long long b){
    int mid;
    while(true){
        mid=(l+r)/2;
        pushdown(u,l,r);
        if(l==r)break;
        if(a[u].ll>b)break;
        if(a[rs].ll<=b)u=rs,l=mid+1;
        else u=ls,r=mid;
    }
    if(a[u].ll<b)l++;
    if(l>n)return 0;
    return l;
}

long long query(int u,int l,int r,int ql,int qr,long long b){
    pushdown(u,l,r);
    if(l==ql&&r==qr)return a[u].sum-b*(r-l+1);
    int mid=(l+r)/2;
    if(qr<=mid)return query(ls,l,mid,ql,qr,b);
    else if(ql>mid)return query(rs,mid+1,r,ql,qr,b);
    else return query(ls,l,mid,ql,mid,b)+query(rs,mid+1,r,mid+1,r,b);
    update(u);
}

void modify(int u,int l,int r,int ql,int qr,long long b){
    pushdown(u,l,r);
    if(l==ql&&r==qr){
        a[u].sum=b*(r-l+1);
        a[u].ll=a[u].fz=b;
        return;
    }
    int mid=(l+r)/2;
    if(qr<=mid)modify(ls,l,mid,ql,qr,b);
    else if(ql>mid)modify(rs,mid+1,r,ql,qr,b);
    else modify(ls,l,mid,ql,mid,b),modify(rs,mid+1,r,mid+1,r,b);
    update(u);
}

int main()
{
    long long pre=0,d,b;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%lld",&v[i]);
    sort(v+1,v+n+1);
    for(int i=1;i<=n;i++)v[i]+=v[i-1];
    build(root,1,n);
    for(int i=1;i<=m;i++){
        scanf("%lld%lld",&d,&b);
        a[root].add+=d-pre;
        a[root].sum+=(d-pre)*v[n];
        a[root].ll+=(d-pre)*v[1];
        int st=search(root,1,n,b);
        if(!st)printf("0\n");
        else {
            printf("%lld\n",query(root,1,n,st,n,b));
            modify(root,1,n,st,n,b);
        }
        pre=d;
    }
    return 0;
}