Description
农夫Byteasar买了一片n亩的土地,他要在这上面种草。
他在每一亩土地上都种植了一种独一无二的草,其中,第i亩土地的草每天会长高a[i]厘米。
Byteasar一共会进行m次收割,其中第i次收割在第d[i]天,并把所有高度大于等于b[i]的部分全部割去。Byteasar想知道,每次收割得到的草的高度总和是多少,你能帮帮他吗?
Input
第一行包含两个正整数
n,m(1≤n,m≤500000)
n
,
m
(
1
≤
n
,
m
≤
500000
)
,分别表示亩数和收割次数。
第二行包含
n
n
个正整数,其中第个数为
a[i](1≤a[i]≤1000000)
a
[
i
]
(
1
≤
a
[
i
]
≤
1000000
)
,依次表示每亩种植的草的生长能力。
接下来m行,每行包含两个正整数
d[i],b[i](1≤d[i]≤1012,0≤b[i]≤1012)
d
[
i
]
,
b
[
i
]
(
1
≤
d
[
i
]
≤
10
12
,
0
≤
b
[
i
]
≤
10
12
)
,依次描述每次收割。
数据保证
d[1]<d[2]<...<d[m]
d
[
1
]
<
d
[
2
]
<
.
.
.
<
d
[
m
]
,并且任何时刻没有任何一亩草的高度超过
1012
10
12
。
Output
输出 m 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,看来代码能力下降了啊。
这道题在思维上不难,就是在单调序列上的线段树。注意到把草按生长速度从小到大排序后高度总是不下降的,所以可以二分找到高度大于的最左端点,然后区间查询再赋值。至于草的生长,就是一个区间加,在实现上就是维护两个标记:区间赋值和区间加和区间和、区间最小高度。另外,注意一下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;
}