n天每天需要k个东西,有m种价格方案:
方案i [Li,Ri] 内每天有ci个,每个pi元
(如果某天所能获得的东西不足k个,全部借完)
求最小花费。
解:
按价格排序,线段树记录某天已经借了的数量,成段更新。
更新时判断,1.如果最小值 >= k,即该区间已经满足条件,不用再借,return
2.如果最大值 + c <= k,即该区间全部都可以借c个,利用lazy标记区间更新
3.如果最大值==最小值,说明该区间所有值相等,可以一起借,借的个数为min(c,k - max),区间更新
4.否则,向下pushdown
因此还需要l == r的递归出口
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
#define lc (rt << 1)
#define rc (rt << 1 | 1)
#define mid (l + r) / 2
const int maxn = 1e6 + 5;//
ll tr[maxn << 2];//记录第i天已经租的个数
int mn[maxn << 2],mx[maxn << 2];
int add[maxn << 2];
struct node{
int l,r,c,p;
bool operator < (const node &b) const{
return p < b.p;
}
}ns[maxn];
void init(int l,int r,int rt)
{
if(l == r){
tr[rt] = 0;
mn[rt] = 0;
mx[rt] = 0;
return ;
}
init(l,mid,lc);
init(mid + 1,r,rc);
//pushup(l,r,rt);
tr[rt] = tr[lc] + tr[rc];
mn[rt] = min(mn[lc],mn[rc]);
mx[rt] = max(mx[lc],mx[rc]);
}
int k;
ll ans = 0;
int c,p;
void pushdown(int l,int r,int rt)
{
if(add[rt]){
add[lc] += add[rt];add[rc] += add[rt];
tr[lc] += 1ll * add[rt] * (mid - l + 1);
tr[rc] += 1ll * add[rt] * (r - mid);
mn[lc] += add[rt];mn[rc] += add[rt];
mx[lc] += add[rt];mx[rc] += add[rt];
add[rt] = 0;
}
}
void update(int ql,int qr,int l,int r,int rt)
{
if(l == r){//??
int t = min(c,int(k - tr[rt]));
ans += 1ll * t * p;
tr[rt] = mx[rt] = mn[rt] = t + tr[rt];
return;
}
if(ql <= l && r <= qr){
if(mn[rt] >= k) return;
if(mx[rt] + c <= k){
add[rt] += c;
tr[rt] += 1ll * (r - l + 1) * c;
mn[rt] += c;mx[rt] += c;
ans += 1ll * c * p * (r - l + 1);
return;
}
else if(mx[rt] == mn[rt]){//mx[rt] + c > k
int t = min(c,k - mn[rt]);
add[rt] += t;
tr[rt] += 1ll * (r - l + 1) * t;
mn[rt] += t;mx[rt] += t;
ans += 1ll * t * p * (r - l + 1);
return;
}
}
pushdown(l,r,rt);
if(ql <= mid) update(ql,qr,l,mid,lc);
if(mid < qr) update(ql,qr,mid + 1,r,rc);
tr[rt] = tr[lc] + tr[rc];
mn[rt] = min(mn[lc],mn[rc]);
mx[rt] = max(mx[lc],mx[rc]);
}
int main()
{
int n,m;
scanf("%d%d%d",&n,&k,&m);
for(int i = 0;i < m;i ++){
scanf("%d%d%d%d",&ns[i].l,&ns[i].r,&ns[i].c,&ns[i].p);
}
sort(ns,ns + m);
init(1, n, 1);
for(int i = 0;i < m;i ++){
c = ns[i].c;p = ns[i].p;
update(ns[i].l,ns[i].r,1,n,1);
// for (int i = 0; i < 2 * n; i ++) {
// printf("%d %d %d %d %d\n",i,tr[i],mn[i],mx[i],add[i]);
// }
// printf("after %d th update ans = %lld \n",i,ans);
}
printf("%I64d\n",ans);
return 0;
}