codeforces C. Cloud Computing(線段樹)

題目鏈接:https://codeforces.com/contest/1070/problem/C

思路:線段樹水題一道,本來不想寫什麼題解的,結果發現別人幾乎全是維護價格的,我這篇是維護天數的,首先離線m個詢問,按照價格排序,那麼對於每個詢問這明顯就是一個區間加減的操作而已,可以發現每一天最多隻有一次會減小到0,那麼維護一下區間最大值,最大值爲0表示不需要更新,在維護一下區間最小值,最小值大於當前詢問的個數y那麼區間減y,否則直接遞歸下去,最後直接輸出tr[1].su就好了

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include<iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define pb push_back
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define lc (d<<1)
#define rc (d<<1|1)
#define P pair<int,int>
#define pi acos(-1)
int n,k,m;
struct as{
    int l,r,ma,lz,mi;
    ll su,lzz;
}tr[1000007<<2];
struct ass{int l,r,y,x;}a[1000008];
bool cmp(ass a,ass b){return a.x<b.x;}
void push(int d){tr[d].ma=max(tr[lc].ma,tr[rc].ma);tr[d].su=tr[lc].su+tr[rc].su;tr[d].mi=min(tr[lc].mi,tr[rc].mi);}
void build(int d,int l,int r)
{
    tr[d].l=l,tr[d].r=r,tr[d].lz=tr[d].lzz=0;
    if(l==r)  {tr[d].ma=tr[d].mi=k,tr[d].su=0;return;}
    int mid=(l+r)>>1;
    build(lc,l,mid);
    build(rc,mid+1,r);
    push(d);
}
void pushh(int d)
{
    if(!tr[d].lz&&!tr[d].lzz) return;
    tr[lc].ma-=tr[d].lz;tr[rc].ma-=tr[d].lz;
    tr[lc].mi-=tr[d].lz;tr[rc].mi-=tr[d].lz;
    tr[lc].lz+=tr[d].lz,tr[rc].lz+=tr[d].lz;
    tr[lc].su+=(ll)(tr[lc].r-tr[lc].l+1)*tr[d].lzz;tr[rc].su+=(ll)(tr[rc].r-tr[rc].l+1)*tr[d].lzz;
    tr[lc].lzz+=tr[d].lzz,tr[rc].lzz+=tr[d].lzz;
    tr[d].lz=tr[d].lzz=0;
}
void add(int d,int l,int r,int x,int y)
{
    if(tr[d].ma==0) return;
    if(tr[d].l==l&&tr[d].r==r)
    {
        if(tr[d].mi>=y)
        {
            tr[d].mi-=y;tr[d].ma-=y;tr[d].lz+=y;
            tr[d].su+=(ll)(r-l+1)*(ll)x*(ll)y;tr[d].lzz+=(ll)x*(ll)y;
            return;
        }
        else if(tr[d].l==tr[d].r)
        {
            tr[d].su+=(ll)x*(ll)(tr[d].ma);
            tr[d].ma=tr[d].mi=0;
            return;
        }
    }
    int mid=(tr[d].l+tr[d].r)>>1;
    pushh(d);
    if(mid>=r)  add(lc,l,r,x,y);
    else if(l>mid)  add(rc,l,r,x,y);
    else  add(lc,l,mid,x,y),add(rc,mid+1,r,x,y);
    push(d);
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n>>k>>m;
    FOR(i,1,m) si(a[i].l),si(a[i].r),si(a[i].y),si(a[i].x);
    sort(a+1,a+m+1,cmp);build(1,1,n);
    FOR(i,1,m) add(1,a[i].l,a[i].r,a[i].x,a[i].y);
    cout<<tr[1].su<<endl;
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章