书上例题,二维线段树,一开始照着书上打了一遍,发现书上的代码有点封装过度了而且特别长,跑的还慢,而且很多地方看起来就不不好(不是不换行的大括号/笑哭),又臭又长,于是照着黄学长的码风改了一遍,毕竟本来线段树就是跟着黄学长学的,递归+位运算,感觉自己跟hzwer码风好像= =。
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
inline int F()
{
register int a;register char ch;
while(ch=getchar(),ch<'0'||ch>'9'&&ch!='-');a=ch-'0';
while(ch=getchar(),ch>='0'&&ch<='9')a=(a<<3)+(a<<1)+ch-'0';
return a;
}
int D,S,N;
int ql,qr,qd,qu;
struct segx{
int v[3005],tag[3005];
void change(int k,int l,int r,int x,int y,int val) //x,y是查询区间
{
v[k]=max(v[k],val);
if (l==x&&y==r){tag[k]=max(tag[k],val);return;}
int mid=(l+r)>>1;
if (x<=mid)change(k<<1,l,mid,x,min(mid,y),val);
if (y>mid)change(k<<1|1,mid+1,r,max(x,mid+1),y,val);
}
int query(int k,int l,int r,int x,int y)
{
if (l==x&&y==r)return v[k];
int mid=(l+r)>>1,ans=tag[k];
if (x<=mid)ans=max(ans,query(k<<1,l,mid,x,min(mid,y)));
if (y>mid)ans=max(ans,query(k<<1|1,mid+1,r,max(x,mid+1),y));
return ans;
}
};
struct segy{
segx v[3005],tag[3005];
void change(int k,int l,int r,int x,int y,int val)
{
v[k].change(1,1,S,qd,qu,val);
if (l==x&&y==r){tag[k].change(1,1,S,qd,qu,val);return;}
int mid=(l+r)>>1;
if (x<=mid)change(k<<1,l,mid,x,min(mid,y),val);
if (y>mid)change(k<<1|1,mid+1,r,max(x,mid+1),y,val);
}
int query(int k,int l,int r,int x,int y)
{
if (l==x&&y==r)return v[k].query(1,1,S,qd,qu);
int mid=(l+r)>>1,ans=tag[k].query(1,1,S,qd,qu);
if (x<=mid)ans=max(ans,query(k<<1,l,mid,x,min(mid,y)));
if (y>mid)ans=max(ans,query(k<<1|1,mid+1,r,max(x,mid+1),y));
return ans;
}
}T;
int main()
{
D=F(),S=F(),N=F();
int d,s,w,x,y;
for (int i=1;i<=N;i++)
{
d=F(),s=F(),w=F(),x=F(),y=F();
ql=x+1,qr=x+d,qd=y+1,qu=y+s;
int ans=T.query(1,1,D,ql,qr);
T.change(1,1,D,ql,qr,ans+w);
}
qd=1;qu=S;
cout<<T.query(1,1,D,1,D);
return 0;
}