bzoj2762-[JLOI2011]不等式组
求 \(x=k\) 时满足一元一次不等式 \(ax+b<c\) 的个数.
解出 \(\frac{c-b}{a}\) 之后取整,得到合法区间,用树状数组维护.
注意 \(a\) 的值域是 \([-10^9,10^8]\) , \(k\) 的值域是 \([-10^6,10^6]\) . 需要特判范围外的部分.
#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
#define rep(i,l,r) for(register int i=(l);i<=(r);++i)
#define repdo(i,l,r) for(register int i=(l);i>=(r);--i)
#define il inline
typedef double db;
typedef long long ll;
//---------------------------------------
const int vsz=2e6+20;
int nmax=2e6+16;
int bit[vsz];
#define lb(v) (v&(-v))
void add(int p,int v){while(p<=nmax)bit[p]+=v,p+=lb(p);}
void addr(int l,int r,int v){add(l,v);add(r+1,-v);}
int que(int p){int res=0;while(p)res+=bit[p],p-=lb(p);return res;}
const int nsz=1e5+10;
int ninf=1e6+3,line[nsz][2],pl=0;
int tr(int v){return v+1e6+5;}
void insert(int a,int b,int c){
int l,r;
if(a==0){
if(b<=c)l=ninf,r=-ninf;
else l=-ninf,r=ninf;
}
else if(a>0){
l=floor((db)(c-b)/a)+1,r=ninf;
}
else{
l=-ninf,r=ceil((db)(c-b)/a)-1;
}
l=max(l,-ninf),r=min(r,ninf);
if(r<-ninf||l>ninf)l=ninf,r=-ninf;
line[++pl][0]=l,line[pl][1]=r;
if(l!=ninf)addr(tr(l),tr(r),1);
}
void del(int p){
if(line[p][0]==ninf&&line[p][1]==-ninf)return;
addr(tr(line[p][0]),tr(line[p][1]),-1);
line[p][0]=ninf,line[p][1]=-ninf;
}
int query(int p){
return que(tr(p));
}
int n;
char s[150];
int main(){
ios::sync_with_stdio(0),cin.tie(0);
cin>>n;
int a,b,c;
rep(i,1,n){
cin>>s;
if(s[0]=='A'){
cin>>a>>b>>c;
insert(a,b,c);
}
else if(s[0]=='D'){
cin>>a;
del(a);
}
else{
cin>>a;
cout<<query(a)<<'\n';
}
}
return 0;
}