[题解]P1083 [NOIP2012 提高组] 借教室
解法\(1\):线段树 - \(O((n+m)\log n)\)
比较直观的一种做法,但是可能需要卡一下输入(这里没卡也过了,但要注意输入是\(10^6\)级的,为了保险一定要加)。
#include<bits/stdc++.h>
#define lc (x<<1)
#define rc ((x<<1)|1)
#define int long long
using namespace std;
int n,m,a[1000010],minn[4000010],tag[4000010];
void update(int x){minn[x]=min(minn[lc],minn[rc]);
}
void ch(int x,int v){tag[x]+=v;minn[x]+=v;
}
void pushdown(int x){if(tag[x]){ch(lc,tag[x]);ch(rc,tag[x]);tag[x]=0;}
}
void build(int l,int r,int x){if(l==r){minn[x]=a[l];return;}int mid=(l+r)>>1; build(l,mid,lc);build(mid+1,r,rc);update(x);
}
void modify(int a,int b,int v,int l,int r,int x){pushdown(x);if(a<=l&&r<=b){ch(x,v);return;}int mid=(l+r)>>1;if(a<=mid) modify(a,b,v,l,mid,lc);if(b>mid) modify(a,b,v,mid+1,r,rc);update(x);
}
int query(int a,int b,int l,int r,int x){pushdown(x);if(a<=l&&r<=b) return minn[x];int mid=(l+r)>>1,ans=LLONG_MAX;if(a<=mid) ans=min(ans,query(a,b,l,mid,lc));if(b>mid) ans=min(ans,query(a,b,mid+1,r,rc));return ans;
}
signed main(){cin>>n>>m;for(int i=1;i<=n;i++){cin>>a[i];}build(1,n,1);for(int i=1;i<=m;i++){int v,l,r;cin>>v>>l>>r;modify(l,r,-v,1,n,1);if(query(1,n,1,n,1)<0){cout<<"-1\n"<<i<<"\n";return 0;}}cout<<"0\n";return 0;
}
解法\(2\):二分答案 - \(O((n+m)\log m)\)
二分枚举申请的编号,对于枚举出的编号。通过差分,计算出处理完当前编号后,每一天剩余的教室数量。如果最终结果存在负数,则r=mid
,否则l=mid+1
。因为我们要找的是使存在负数的最小编号。
#include<bits/stdc++.h>
#define int long long
#define N 1000010
#define M 1000010
using namespace std;
int n,m,a[N],b[N],tb[N],c[M],ld[M],rd[M];
bool check(int d){for(int i=1;i<=n;i++) tb[i]=b[i];for(int i=1;i<=d;i++) tb[ld[i]]-=c[i],tb[rd[i]+1]+=c[i];for(int i=1;i<=n;i++) tb[i]+=tb[i-1];for(int i=1;i<=n;i++) if(tb[i]<0) return 1;return 0;
}
signed main(){cin>>n>>m;for(int i=1;i<=n;i++){cin>>a[i];b[i]=a[i]-a[i-1];}for(int i=1;i<=m;i++) cin>>c[i]>>ld[i]>>rd[i];int l=1,r=m;bool flag=0;while(l<r){int mid=(l+r)>>1;if(check(mid)){//不合法 flag=1;r=mid;}else{l=mid+1;}}if(!flag) cout<<"0\n";else cout<<"-1\n"<<l<<"\n";return 0;
}
解法\(3\) - \(O(n+m)\)
总体思路就是用一个指针\(j\),初始为\(m\)。对于每一天,判断完成操作\(1\sim j\)后是否仍然合法。如果不合法了,就把\(j\)往前挪,同时撤回操作,直到该天合法为止。
最后,如果\(j=m\),则说明所有操作都是合法的,输出0
。
否则,我们需要输出导致不合法的第一个操作。而\(j\)表示的是合法的最后一个操作,输出\(j+1\)即可。
此算法的优势在于,枚举的每一天不需要从最后开始移动指针,而是从上一天的位置开始移动,大大减少了冗余操作。
注意到\(j\)是只减不增的,所以while
循环一共最多执行\(m\)次。而for
一共执行\(n\)次,所以复杂度是\(O(n+m)\)。
#include<bits/stdc++.h>
#define N 1000010
#define M 1000010
#define int long long
using namespace std;
int n,m,a[N],c[M],l[M],r[M];
int cf[N];
signed main(){cin>>n>>m;for(int i=1;i<=n;i++) cin>>a[i];for(int i=1;i<=m;i++){cin>>c[i]>>l[i]>>r[i];cf[l[i]]+=c[i],cf[r[i]+1]-=c[i];}int j=m,sum=0;//sum表示完成操作1~j后,第i天一共用去多少教室for(int i=1;i<=n;i++){sum+=cf[i];//cf[i]表示完成操作j之后,第i天一共用去多少个教室while(sum>a[i]){//用去的>已有的,不合法,需要撤回操作cf[l[j]]-=c[j],cf[r[j]+1]+=c[j];if(l[j]<=i&&i<=r[j]) sum-=c[j];j--;}}if(j==m) cout<<"0\n";else cout<<"-1\n"<<j+1<<"\n";return 0;
}