首页 > 代码库 > BZOJ 2006 NOI2010 超级钢琴 划分树+堆
BZOJ 2006 NOI2010 超级钢琴 划分树+堆
题目大意:给定一个序列,找到k个长度在[l,r]之间的序列,使得和最大
暴力O(n^2logn),肯定过不去
看到这题的第一眼我OTZ了一下午。。。后来研究了很久别人的题解才弄明白怎么回事。。。蒟蒻果然不能理解大神的思路啊0.0
首先维护前缀和,那么以第i个元素结尾的和最大的序列自然就是sum[i]-min{sum[j]}(i-r<=j<=i-l)
然后我们维护一个大根堆,每取走一个以i为结尾的元素,加入sum[i]-2thmin{sum[j]},再取走这个元素就加入sum[i]-3thmin{sum[j]},以此类推
维护区间第k小,划分树,不说啥了吧 尼玛本大爷的划分树居然还写挂了0.0
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 500500 using namespace std; typedef pair<int,int> abcd; typedef long long ll; int n,k,l,r;ll ans; int sum[M],a[M],b[M],c[M],s[22][M],now[M]; abcd heap[M];int top; void insert(abcd x) { heap[++top]=x; int t=top; while(t>1&&heap[t]>heap[t>>1]) swap(heap[t],heap[t>>1]),t>>=1; } void pop() { heap[1]=heap[top--]; int t=2; while(t<=top) { if(t<top&&heap[t+1]>heap[t]) t++; if(heap[t]>heap[t>>1]) swap(heap[t],heap[t>>1]),t<<=1; else break; } } void Build_Tree(int l,int r,int dpt) { int i,mid=l+r>>1; int l1=l,l2=mid+1; int left=mid-l+1; if(l==r) return ; for(i=l;i<=r;i++) left-=a[i]<c[mid]; for(i=l;i<=r;i++) { if(a[i]<c[mid]||a[i]==c[mid]&&left) b[l1++]=a[i],s[dpt][i]=i==l?1:s[dpt][i-1]+1,left-=a[i]==c[mid]; else b[l2++]=a[i],s[dpt][i]=i==l?0:s[dpt][i-1]; } memcpy(a+l,b+l,r-l+1<<2); Build_Tree(l,mid,dpt+1); Build_Tree(mid+1,r,dpt+1); } int Get_Ans(int l,int r,int dpt,int x,int y,int k) { int mid=l+r>>1; int l1=x==l?0:s[dpt][x-1],l2=s[dpt][y]; if(l==r) return c[mid]; if(k<=l2-l1) return Get_Ans(l,mid,dpt+1,l+l1,l+l2-1,k); else return Get_Ans(mid+1,r,dpt+1,(mid+1)+(x-l-l1),(mid+1)+(y-l+1-l2)-1,k-l2+l1); } int main() { //freopen("piano.in","r",stdin); //freopen("piano.out","w",stdout); int i; cin>>n>>k>>l>>r; for(i=1;i<=n;i++) scanf("%d",&sum[i]),sum[i]+=sum[i-1]; memcpy(a,sum,n+1<<2); memcpy(c,sum,n+1<<2); sort(c,c+n+1); Build_Tree(0,n,0); for(i=l;i<=n;i++) { now[i]=1; int temp=Get_Ans(0,n,0,max(i-r,0),i-l,1); insert( make_pair(sum[i]-temp,i) ); } for(i=1;i<=k;i++) { abcd temp=heap[1];pop(); if( now[temp.second]!=(temp.second-l)-( max(temp.second-r,0) )+1 ) insert( make_pair(sum[temp.second]-Get_Ans(0,n,0,max(temp.second-r,0),temp.second-l,++now[temp.second]),temp.second) ); ans+=temp.first; } cout<<ans<<endl; }
BZOJ 2006 NOI2010 超级钢琴 划分树+堆
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。