Codeforces Round #548 (Div. 2) F splay(新坑) + 思维

https://codeforces.com/contest/1139/problem/F

题意

有m个人,n道菜,每道菜有(p_i),(s_i),(b_i),每个人有(inc_j),(pref_j),一个人可以买一道菜的条件是
1. (p_i leq inc_j leq s_i)
2. (|b_i - pref_j| leq inc_j-p_i)
,问每个人分别能买多少道菜

题解

• 转化一下公式
  1. (p_i leq inc_j leq s_i)
  • 下面两个满足其一即可
  2. (b_i + p_i leq inc_j + pref_j)
  3. (p_i - b_i leq inc_j - pref_j)
• 对于每个j只需要找出有多少个i满足要求就行
• 按(p_i leq inc_j leq s_i)建立时间节点并排序,然后从前往后扫一遍,用两颗splay维护后两个式子并计数

代码

#include<bits/stdc++.h>
#define ls(x) tr[x].son[0]
#define rs(x) tr[x].son[1]
#define MAXN 200005
using namespace std;
struct SPLAY{
	int root=0,tot=0;
	struct node{
		int son[2],par;
		int sz,tsz,v;
		void init(int _v=0,int _sz=0){
			v=_v;
			tsz=sz=_sz;
			son[0]=son[1]=par=0;
		}
	}tr[MAXN];
	void push_up(int x){
		tr[x].tsz=tr[x].sz+tr[ls(x)].tsz+tr[rs(x)].tsz;
	}
	int chk(int x){
		return rs(tr[x].par)==x;
	}
	void rot(int x){
		int y=tr[x].par,z=tr[y].par,k=chk(x),w=tr[x].son[k^1];
		tr[y].son[k]=w;tr[w].par=y;
		tr[z].son[chk(y)]=x;tr[x].par=z;
		tr[x].son[k^1]=y;tr[y].par=x;
		push_up(y);
		push_up(x);
	}
	void splay(int x,int goal){
		while(tr[x].par!=goal){
			int y=tr[x].par,z=tr[y].par;
			if(z!=goal){
				if(chk(x)==chk(y))rot(y);
				else rot(x);
			}
			rot(x);
		}
		if(!goal)root=x;
	}
	void insert(int v,int sz){
		int p=root,ff=0;
		while(p&&tr[p].v!=v){
			ff=p;
			p=tr[p].son[v>tr[p].v];
		}
		if(p){
			tr[p].sz+=sz;
		}else{
			p=++tot;
			if(ff)tr[ff].son[v>tr[ff].v]=p;
			tr[p].init(v,sz);
			tr[p].par=ff;
		}
		splay(p,0);
	}
	int count(int v){
		insert(v,1);
		int re=tr[rs(root)].tsz;
		insert(v,-1);
		return re;
	}
}V[2];
int n,m,p[MAXN],s[MAXN],b[MAXN],inc[MAXN],pref[MAXN],ans[MAXN];
int cnt=0,tot=0;

struct node{
	int v,id,kd;
}A[MAXN*4];
bool cmp(node x,node y){
	if(x.v==y.v)return x.kd<y.kd;
	return x.v<y.v;
}
void add(int v,int id,int kd){
	A[++tot].v=v;
	A[tot].id=id;
	A[tot].kd=kd;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++)scanf("%d",&p[i]),add(p[i],i,1);	
	for(int i=1;i<=n;i++)scanf("%d",&s[i]),add(s[i],i,3);
	for(int i=1;i<=n;i++)scanf("%d",&b[i]);
	for(int i=1;i<=m;i++)scanf("%d",&inc[i]),add(inc[i],i,2);
	for(int i=1;i<=m;i++)scanf("%d",&pref[i]);
	sort(A+1,A+tot+1,cmp);
	for(int i=1;i<=tot;i++){
		int id=A[i].id;
		if(A[i].kd==1){
			cnt++;
			V[0].insert(b[id]+p[id],1);
			V[1].insert(p[id]-b[id],1);	
		}else if(A[i].kd==2){
			ans[id]=cnt-V[0].count(inc[id]+pref[id])-V[1].count(inc[id]-pref[id]);	
		}else{
			cnt--;
			V[0].insert(b[id]+p[id],-1);
			V[1].insert(p[id]-b[id],-1);
		}
	}
	for(int i=1;i<=m;i++)printf("%d ",ans[i]);
}