牛客多校第一场Points Division(线段树)
题目大意
给出一些二维平面上的一些点,每个点有两种权值a,b,现在要将所有的点分成两部分,分别归入A集合和B集合,并且要求不存在 i ∈ A , j ∈ B i\in A,j\in B i∈A,j∈B使得 x i ≥ x j x_i\ge x_j xi≥xj且 y i ≤ y j y_i\le y_j yi≤yj。对于每种划分,都有一个值其大小为A中的点的a值加上B中电的b值。现在求值尽可能大划分
解题思路
题目中要求的划分即画一条单调不减直线,在这条直线上方的点即归入A集合,下方的点就归入B集合。故从左向右枚举这条线上的折点,并用线段树维护当前横坐标前所有纵坐标的最大的划分,则可以得出最大答案。当存在横坐标相等时从上向下枚举则可以避免重复枚举
AC代码
#include<bits/stdc++.h>
#define int long long
#define lson(x) ((x)<<1)
#define rson(x) ((x)<<1|1)
using namespace std;
typedef long long LL;
const int inf=0x3f3f3f3f3f3f3f3f;
const int size=1e5+5;
struct Point{
int x,y;
int yid;
int a,b;
friend bool operator<(Point j,Point k)
{
if(j.x==k.x) return j.y>k.y;
return j.x<k.x;
}
Point(){}
Point(int x,int y,int a,int b):x(x),y(y),a(a),b(b){}
}p[size];
struct node{
LL mx,tag;
node():mx(0),tag(0){}
node(LL mx,LL tag):mx(mx),tag(tag){}
}tree[size<<2];
vector<int> yv;
void build(int id,int l,int r)
{
tree[id]=node();
if(l==r)
{
return ;
}
int mid=(l+r)/2;
build(lson(id),l,mid);
build(rson(id),mid+1,r);
}
void updown(int id)
{
if(tree[id].tag==0) return ;
tree[lson(id)].mx=tree[lson(id)].mx+tree[id].tag;
tree[lson(id)].tag=tree[lson(id)].tag+tree[id].tag;
tree[rson(id)].mx=tree[rson(id)].mx+tree[id].tag;
tree[rson(id)].tag=tree[rson(id)].tag+tree[id].tag;
tree[id].tag=0;
}
void update(int id,int l,int r,int ql,int qr,LL val)
{
if(l>=ql&&r<=qr)
{
tree[id].mx+=val;
tree[id].tag+=val;
return ;
}
updown(id);
int mid=(l+r)/2;
if(ql<=mid)update(lson(id),l,mid,ql,qr,val);
if(mid<qr) update(rson(id),mid+1,r,ql,qr,val);
tree[id].mx=max(tree[lson(id)].mx,tree[rson(id)].mx);
}
void update2(int id,int l,int r,int loc,LL val)
{
if(l==r)
{
tree[id].mx=max(tree[id].mx,val);
return ;
}
updown(id);
int mid=(l+r)/2;
if(mid>=loc) update2(lson(id),l,mid,loc,val);
if(mid<loc) update2(rson(id),mid+1,r,loc,val);
tree[id].mx=max(tree[lson(id)].mx,tree[rson(id)].mx);
}
LL query(int id,int l,int r,int ql,int qr)
{
if(l>=ql&&r<=qr)
{
return tree[id].mx;
}
int mid=(l+r)/2;
updown(id);
LL ans=0;
if(mid>=ql) ans=max(ans,query(lson(id),l,mid,ql,qr));
if(mid<qr) ans=max(ans,query(rson(id),mid+1,r,ql,qr));
return ans;
}
int32_t main()
{
int n;
while(~scanf("%lld",&n))
{
yv.clear();
int x,y,a,b;
for(int i=1;i<=n;i++)
{
scanf("%lld%lld%lld%lld",&x,&y,&a,&b);
p[i]=Point(x,y,a,b);
yv.push_back(y);
}
sort(p+1,p+1+n);
yv.push_back(-inf);
yv.push_back(inf);
sort(yv.begin(),yv.end());
yv.erase(unique(yv.begin(),yv.end()),yv.end());
for(int i=1;i<=n;i++)
{
p[i].yid=lower_bound(yv.begin(),yv.end(),p[i].y)-yv.begin()+1;
}
int ns=yv.size();
build(1,1,ns);
for(int i=1;i<=n;i++)
{
LL g = query(1, 1, ns, 1, p[i].yid);
update2(1, 1, ns, p[i].yid, g + p[i].b);
update(1, 1, ns, p[i].yid + 1, ns, p[i].b);
update(1, 1, ns, 1, p[i].yid - 1, p[i].a);
}
printf("%lld\n",tree[1].mx);
}
}