BZOJ1513: [POI2006]Tet-Tetris 3D

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
#include <set>
#include <map>
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
const int MAXN=2e3+10;
const double eps=1e-8;
#define ll long long
using namespace std;
struct edge{int t;edge*next;}e[MAXN<<1],*h[MAXN],*o=e;
void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;}
ll read(){
	ll x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch==‘-‘)f=-1;ch=getchar();}
	while(isdigit(ch))x=x*10+ch-‘0‘,ch=getchar();
	return x*f;
}
typedef struct node{
	int l,r,maxx,tag,rt,tag1,tag2;
}node;
node d[MAXN*MAXN];
int cnt,L,R,D,S,vul,n,rt;
void update1(int &x,int l,int r,int ql,int qr){
	if(!x)x=++cnt;
	if(ql<=l&&r<=qr){d[x].maxx=max(d[x].maxx,vul);d[x].tag1=max(d[x].tag1,vul);return ;}
	int mid=(l+r)>>1;
	if(ql<=mid)update1(d[x].l,l,mid,ql,qr);
	if(qr>mid)update1(d[x].r,mid+1,r,ql,qr);
	d[x].maxx=max(d[d[x].l].maxx,d[d[x].r].maxx);
}
void update2(int &x,int l,int r,int ql,int qr){
	if(!x)x=++cnt;
	if(ql<=l&&r<=qr){d[x].tag=max(d[x].tag,vul);d[x].tag2=max(d[x].tag2,vul);return ;}
	int mid=(l+r)>>1;
	if(ql<=mid)update2(d[x].l,l,mid,ql,qr);
	if(qr>mid)update2(d[x].r,mid+1,r,ql,qr);
	d[x].tag=max(d[d[x].l].tag,d[d[x].r].tag);
}
void update(int &x,int l,int r,int ql,int qr){
	if(!x)x=++cnt;
	update1(d[x].rt,1,S,L,R);
	if(ql<=l&&r<=qr){update2(d[x].rt,1,S,L,R);return ;}
	int mid=(l+r)>>1;
	if(ql<=mid)update(d[x].l,l,mid,ql,qr);
	if(qr>mid)update(d[x].r,mid+1,r,ql,qr);
}
int ans,ans1;
void querty1(int x,int l,int r,int ql,int qr){
	ans1=max(ans1,d[x].tag1);
	if(ql<=l&&r<=qr){ans1=max(ans1,d[x].maxx);return ;}
	int mid=(l+r)>>1;
	if(ql<=mid)querty1(d[x].l,l,mid,ql,qr);
	if(qr>mid)querty1(d[x].r,mid+1,r,ql,qr);
}
void querty2(int x,int l,int r,int ql,int qr){
	ans1=max(ans1,d[x].tag2);
	if(ql<=l&&r<=qr){ans1=max(ans1,d[x].tag);return ;}
	int mid=(l+r)>>1;
	if(ql<=mid)querty2(d[x].l,l,mid,ql,qr);
	if(qr>mid)querty2(d[x].r,mid+1,r,ql,qr);
}
void querty(int x,int l,int r,int ql,int qr){
	if(ql<=l&&r<=qr){ans1=0;querty1(d[x].rt,1,S,L,R);ans=max(ans,ans1);return ;}
	int mid=(l+r)>>1;ans1=0;querty2(d[x].rt,1,S,L,R);ans=max(ans1,ans);
	if(ql<=mid)querty(d[x].l,l,mid,ql,qr);
	if(qr>mid)querty(d[x].r,mid+1,r,ql,qr);
}
int main(){
    D=read();S=read();n=read();
    int dd,s,w,x,y;D++;S++;
    while(n--){
        dd=read();s=read();w=read();x=read();y=read();x++;y++;
        //cout<<x<<"---------------"<<x+dd-1<<" "<<y<<" "<<y+s-1<<endl;
        ans=0;L=y;R=y+s-1;querty(rt,1,D,x,x+dd-1);
        vul=w+ans;update(rt,1,D,x,x+dd-1);
    }
    ans=0;
    printf("%d\n",d[d[1].rt].maxx);
}