【PA2015】【BZOJ4293】Siano

Description

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

第一行包含两个正整数n,m(1<=n,m<=500000),分别表示亩数和收割次数。
第二行包含n个正整数,其中第i个数为ai,依次表示每亩种植的草的生长能力。
接下来m行,每行包含两个正整数d[i],bi,依次描述每次收割。
数据保证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。

Source

By Claris

注意到所有位置草长度的大小关系总可以看成是不变的,而收割是无视区间下标问题的
那么我们可以先排序然后再构线段树.
每次二分出最小的那个长度大于等于b的位置就行了.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 500010
#define GET (ch>='0'&&ch<='9')
#define lchild rt<<1,l,mid
#define rchild rt<<1|1,mid+1,r
#define ln rt<<1
#define rn rt<<1|1
#define LL long long
using namespace std;
int n,m,pos;
LL sum[MAXN],a[MAXN],d,b,last;
struct seg
{
    int l,r;
    LL maxn,sum,flag,t;
    seg()   {   flag=-1;    }
}tree[MAXN<<2];
inline void in(LL &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
inline void update(int rt,LL t)
{
    tree[rt].maxn+=t*a[tree[rt].r];
    tree[rt].sum+=(sum[tree[rt].r]-sum[tree[rt].l-1])*t;
    tree[rt].t+=t;
}
inline void push_down(int rt)
{
    if (tree[rt].flag!=-1)
    {
        tree[ln].flag=tree[rn].flag=tree[rt].flag;tree[ln].t=tree[rn].t=0;
        tree[ln].maxn=tree[rn].maxn=tree[rt].flag;
        tree[ln].sum=(tree[ln].r-tree[ln].l+1)*tree[rt].flag;
        tree[rn].sum=(tree[rn].r-tree[rn].l+1)*tree[rt].flag;
        tree[rt].flag=-1;
    }
    if (tree[rt].t) update(ln,tree[rt].t),update(rn,tree[rt].t),tree[rt].t=0;
}
inline void push_up(int rt)
{
    tree[rt].sum=tree[ln].sum+tree[rn].sum;
    tree[rt].maxn=tree[rn].maxn;
}
int find(int rt)
{
    int L=tree[rt].l,R=tree[rt].r;
    if (L==R)   return L;
    push_down(rt);
    if (tree[ln].maxn>=b)   return find(ln);    else    return find(rn);
}
LL query(int rt)
{
    int L=tree[rt].l,R=tree[rt].r;LL ret=0;
    if (R<pos)  return 0;
    if (L>=pos) 
    {
        ret=tree[rt].sum;tree[rt].sum=(R-L+1)*b;
        tree[rt].t=0;tree[rt].maxn=b;tree[rt].flag=b;
        return ret;
    }
    push_down(rt);ret=query(ln)+query(rn);push_up(rt);
    return ret;
}
void build(int rt=1,int l=1,int r=n)
{
    tree[rt].l=l;tree[rt].r=r;
    if (l==r)   return;
    int mid=(l+r)>>1;build(lchild);build(rchild);
}
int main()
{
    scanf("%d%d",&n,&m);build();
    for (int i=1;i<=n;i++)  in(a[i]);sort(a+1,a+n+1);
    for (int i=1;i<=n;i++)  sum[i]=sum[i-1]+a[i];
    for (;m;m--)
    {
        in(d);in(b);
        update(1,d-last);last=d;
        if (tree[1].maxn<b) {   puts("0");continue; }
        pos=find(1);printf("%lld\n",query(1)-b*(n-pos+1));
    }
}

你可能感兴趣的:(线段树)