区间更新,区间查询
区间更新用的是差分的做法实现的,理论时间复杂度应该是最快的?
不明白为什么有大佬能跑到几百ms,好像用的是整体二分。。。
详解以后有时间单独贴,理解起来不是很容易
差分做法:
#include<bits/stdc++.h>
#define ls ls_[i]
#define rs rs_[i]
#define mid (l+r)/2
using namespace std;
typedef long long ll;
const int NN=100100;
ll n,m,k,x,y,z,q,W,T,N,S,op,cnt,tmp,dst,cas,sum,val,tot,idx,test;
ll c[NN*40],d[NN*40];
int ls_[NN*40],rs_[NN*40];
int rot[NN],now_rotx[NN],now_roty[NN];
void update(int l,int r,int pos,int val,int x,int &i){
if(!i) i=++tot;
c[i]+=val;d[i]+=val*x;
if(l==r) return ;
if(mid>=pos) update(l,mid,pos,val,x,ls);
else update(mid+1,r,pos,val,x,rs);
}
#define rx now_rotx[i]
#define ry now_roty[i]
int main(){
ios::sync_with_stdio(false);
N=NN;
cin>>n>>m;
while(m--){
cin>>op>>x>>y>>k;
if(op==1){
y++;
for(int j=x;j<=n;j+=j&-j)
update(0,N,k,1,x-1,rot[j]);
for(int j=y;j<=n;j+=j&-j)
update(0,N,k,-1,y-1,rot[j]);
}
else{
int l=0,r=N;x--;
for(int i=y;i;i-=i&-i) ry=rot[i];
for(int i=x;i;i-=i&-i) rx=rot[i];
//查询的非递归写法,递归式的我没想到该怎么写
//因为树状数组节点的有序性决定它只适合做外壳
//也许记录每个lowbit位置存前向星能放进内层?可以一试
while(l!=r){
ll c_rs=0;
for(int i=y;i;i-=i&-i){
c_rs+=c[rs_[ry]]*y;
c_rs-=d[rs_[ry]];
}
for(int i=x;i;i-=i&-i){
c_rs-=c[rs_[rx]]*x;
c_rs+=d[rs_[rx]];
}
if(c_rs>=k){
l=mid+1;
for(int i=y;i;i-=i&-i) ry=rs_[ry];
for(int i=x;i;i-=i&-i) rx=rs_[rx];
}
else{
r=mid;k-=c_rs;
for(int i=y;i;i-=i&-i) ry=ls_[ry];
for(int i=x;i;i-=i&-i) rx=ls_[rx];
}
}
cout<<l<<endl;
}
}
}
lazy数组做法(权值线段树做外壳):
#include<bits/stdc++.h>
#define mid (l+r)/2
using namespace std;
typedef long long ll;
ll n,m,k,x,y,z,q,W,T,N,S,op,cnt,tmp,dst,cas,sum,val,tot,idx,test;
const int NN=50100;
int rot[NN*4];
int ls_[NN*320],rs_[NN*320],c[NN*320],lazy[NN*320];
void push_down(int ,int ,int );
void update(int ,int ,int ,int ,int& );
ll get_sum(int ,int ,int ,int ,int );
void update(int ,int ,int ,int ,int ,int );
ll get_kth(int ,int ,int ,int ,int ,int );
inline int read() {
int x=0,f=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1;
for(;isdigit(ch);x=(x<<3)+(x<<1)+(ch^48),ch=getchar());
return x*f;
}
int main(){
n=read();m=read();N=NN;
for(int i=1;i<=m;i++){
op=read();x=read();y=read();k=read();
if(op==1) update(0,N,k,x,y,1);
else printf("%d\n",get_kth(0,N,k,x,y,1));
}
}
#define ls ls_[i]
#define rs rs_[i]
void push_down(int l,int r,int i){
if(ls==0) ls=++tot;
if(rs==0) rs=++tot;
lazy[ls]+=lazy[i];
lazy[rs]+=lazy[i];
c[ls]+=lazy[i]*(mid-l+1);
c[rs]+=lazy[i]*(r-mid);
lazy[i]=0;
}
void update(int l,int r,int L,int R,int &i){
if(i==0) i=++tot;
if(L<=l&&r<=R){
lazy[i]++;
c[i]+=r-l+1;
return ;
}
if(lazy[i]) push_down(l,r,i);
if(mid>=L) update(l,mid,L,R,ls);
if(mid<R) update(mid+1,r,L,R,rs);
c[i]=c[ls]+c[rs];
}
ll get_sum(int l,int r,int L,int R,int i){
if(i==0) return 0;
if(L<=l&&r<=R) return c[i];
if(lazy[i]) push_down(l,r,i);
ll sum=0;
if(mid>=L) sum+=get_sum(l,mid,L,R,ls);
if(mid<R) sum+=get_sum(mid+1,r,L,R,rs);
return sum;
}
#undef ls
#undef rs
#define ls i<<1
#define rs i<<1|1
void update(int l,int r,int pos,int L,int R,int i){
update(0,N,L,R,rot[i]);
if(l==r) return ;
if(mid>=pos) update(l,mid,pos,L,R,ls);
else update(mid+1,r,pos,L,R,rs);
}
ll get_kth(int l,int r,int k,int L,int R,int i){
if(l==r) return l;
ll r_ls=get_sum(0,N,L,R,rot[rs]);
if(r_ls>=k) return get_kth(mid+1,r,k,L,R,rs);
return get_kth(l,mid,k-r_ls,L,R,ls);
}