首页 > 代码库 > BZOJ 3110 [Zjoi2013]K大数查询 ——整体二分
BZOJ 3110 [Zjoi2013]K大数查询 ——整体二分
【题目分析】
整体二分显而易见。
自己YY了一下用树状数组区间修改,区间查询的操作。
又因为一个字母调了一下午。
貌似树状数组并不需要清空,可以用一个指针来维护,可以少一个log
懒得写了。
【代码】
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; #define maxn 50005 #define inf 0x3f3f3f3f #define ll long long ll n,m,cnt=0,tot=0; struct Bit_Tree{ ll a[maxn],b[maxn]; void add(ll x,ll y,ll z) { // cout<<"Add "<<x<<" "<<y<<" "<<z<<endl; for (ll i=x;i<=n;i+=i&(-i)) b[i]+=z; for (ll i=y+1;i<=n;i+=i&(-i)) b[i]-=z; for (ll i=x;i<=n;i+=i&(-i)) a[i]+=(n-x)*z; for (ll i=y+1;i<=n;i+=i&(-i)) a[i]-=(n-y-1)*z; } void init() { memset(a,0,sizeof a); memset(b,0,sizeof b); } ll getsum(ll x) { ll ret=0,tmp=0; for (ll i=x;i;i-=i&(-i)) ret+=a[i],tmp+=b[i]; return ret-(n-x-1)*tmp; } ll get(ll x,ll y) { return getsum(y)-getsum(x-1); } }t; struct data{ ll opt,id; ll x,y,z; }q[maxn<<1],q1[maxn<<1],q2[maxn<<1]; ll ans[maxn<<1],tag[maxn<<1]; void solve(ll ql,ll qr,ll l,ll r) { // cout<<"solve"<<ql<<" "<<qr<<" "<<l<<" "<<r<<endl; // cout<<"In Que"<<endl; if (ql>qr) return; if (l==r) { for (ll i=ql;i<=qr;++i) ans[q[i].id]=l; return ; } ll mid=l+r>>1,p1=0,p2=0,cnt=0; // cout<<"Mid is "<<mid<<endl; for (ll i=ql;i<=qr;++i) { // cout<<q[i].opt<<" "<<q[i].x<<" "<<q[i].y<<" "<<q[i].z<<" "<<endl; if (q[i].opt==1) { if (q[i].z<=mid) { t.add(q[i].x,q[i].y,1); cnt+=q[i].y-q[i].x+1; q1[++p1]=q[i]; } else q2[++p2]=q[i]; } else { ll tmp=t.get(q[i].x,q[i].y); // cout<<"Tmp is "<<tmp<<endl; if (q[i].z<=tmp) q1[++p1]=q[i]; else q[i].z-=tmp,q2[++p2]=q[i]; } } for (ll i=1;i<=p1;++i) { if (q1[i].opt==1) t.add(q1[i].x,q1[i].y,-1); q[ql+i-1]=q1[i]; } for (ll i=1;i<=p2;++i) q[ql+p1+i-1]=q2[i]; solve(ql,ql+p1-1,l,mid); solve(ql+p1,qr,mid+1,r); } int main() { // freopen("in.txt","r",stdin); // freopen("wa.txt","w",stdout); scanf("%lld",&n); n=maxn-1; scanf("%lld",&m); cnt=m; for (ll i=1;i<=m;++i) { scanf("%lld%lld%lld%lld",&q[i].opt,&q[i].x,&q[i].y,&q[i].z); q[i].id=i; if (q[i].opt==1) t.add(q[i].x,q[i].y,1); if (q[i].opt==2) { tag[i]=1; tot=t.get(q[i].x,q[i].y); q[i].z=tot-q[i].z+1; } } for (ll i=1;i<=m;++i) if (q[i].opt==1) t.add(q[i].x,q[i].y,-1); solve(1,cnt,-inf,inf); for (ll i=1;i<=m;++i) if (tag[i]) printf("%lld\n",ans[i]); }
BZOJ 3110 [Zjoi2013]K大数查询 ——整体二分
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。