我们需要处理接下来n天的借教室信息,其中第i天学校有ri个教室可供租借。共有m份订单,每份订单用三个正整数描述,分别为dj、sj、tj,表示某租借者需要从第sj天到第tj天租借教室(包括第sj天和第tj天),每天需要租借dj个教室。
我们假定,租借者对教室的大小、地点没有要求。即对于每份订单,我们只需要每天提供dj个教室,而它们具体是哪些教室,每天是否是相同的教室则不用考虑。
借教室的原则是先到先得,也就是说我们要按照订单的先后顺序依次为每份订单分配教室。如果在分配的过程中遇到一份订单无法完全满足,则需要停止教室的分配,通知当前申请人修改订单。这里的无法满足指从第sj天到第tj天中有至少一天剩余的教室数量不足dj个。
现在我们需要知道,是否会有订单无法完全满足。如果有,需要通知哪一个申请人修改订单。
第一行包含两个正整数n、m,表示天数和订单的数量;
第二行包含n个正整数,其中第i个数为ri,表示第i天可用于租借的教室数量;
接下来有m行,每行包含三个正整数dj、sj、tj,表示租借的数量,租借开始、结束分别在第几天。
每行相邻的两个数之间均用一个空格隔开。天数与订单均用从1开始的整数编号。
如果所有订单均可满足,则输出只有一行,包含一个整数0。
否则(订单无法完全满足)输出两行,第一行输出一个负整数-1,第二行输出需要修改订单的申请人编号。
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
-1
2
对于100%的数据,有:1 <= n,m <= 10^6,0 <= ri,dj <= 10^9,1 <= sj <= n。
谁说线段树不行,从闹红毛到洋枪队,他们不让线段树上,今天就问线段树威风不!
源代码: #include<cstdio> #include<cstring> #include<algorithm> using namespace std; int m,n,num(0),number(0),h[1000001]; struct tree { int left,right,lefts,rights,sum,sign; }i[2000200]; void x1(int t1,int t2) { int t=++num; i[t].left=t1; i[t].right=t2; if (t1!=t2-1) { i[t].lefts=num+1; x1(t1,(t1+t2)>>1); i[t].rights=num+1; x1((t1+t2)>>1,t2); i[t].sum=min(i[i[t].lefts].sum,i[i[t].rights].sum); } else i[t].sum=h[t1]; } void lazy(int t) { i[i[t].lefts].sum-=i[t].sign; i[i[t].rights].sum-=i[t].sign; i[i[t].lefts].sign+=i[t].sign; i[i[t].rights].sign+=i[t].sign; i[t].sign=0; } bool x2(int t,int s,int t1,int t2) { if (t1<=i[t].left&&t2>=i[t].right) { i[t].sum-=s; i[t].sign+=s; if (i[t].sum<0) //就是它,拯救了整个程序! return false; return true; } if (i[t].sign) lazy(t); bool ans=true; if (t1<(i[t].left+i[t].right)>>1) if (!x2(i[t].lefts,s,t1,t2)||!ans) ans=false; if (t2>(i[t].left+i[t].right)>>1) if (!x2(i[t].rights,s,t1,t2)||!ans) ans=false; i[t].sum=min(i[i[t].lefts].sum,i[i[t].rights].sum); return ans; } int main() //令人惊心动魄的线段树。 { memset(i,0,sizeof(i)); scanf("%d%d",&n,&m); for (int a=1;a<=n;a++) scanf("%d",&h[a]); x1(1,n+1); for (int a=1;a<=m;a++) { int t,t1,t2; scanf("%d%d%d",&t,&t1,&t2); if (!x2(1,t,t1,t2+1)) //结合了查找与判断。 { number=a; break; } } if (number) printf("-1\n%d",number); else printf("0"); return 0; }
要想赶尽洋鬼子呀,还得靠二分+前缀和。
源代码: #include<cstdio> int m,n,ans(0),h[1000001],i[1000001],X[1000001],L[1000001],R[1000001]; void Read(int &t) //读入优化。 { char T=getchar(); while (T<'0'||T>'9') T=getchar(); while (T>='0'&&T<='9') { t=t*10+T-'0'; T=getchar(); } } bool Check() //检验答案。 { int num(0); for (int a=1;a<=n;a++) //前缀和表示单点。 { num+=i[a]; if (num>h[a]) return false; } return true; } int main() { Read(n); Read(m); for (int a=1;a<=n;a++) Read(h[a]); for (int a=1;a<=m;a++) { Read(X[a]); Read(L[a]); Read(R[a]); } int left=1,right=m,mid=(left+right)>>1; for (int a=1;a<=mid;a++) //二分预处理。 { i[L[a]]+=X[a]; i[R[a]+1]-=X[a]; } while (left<=right) if (!Check()) //有答案就尽量前移。 { ans=mid; right=mid-1; mid=(left+right)>>1; for (int a=mid+1;a<=ans;a++) //注意边界。 { i[L[a]]-=X[a]; //割后。 i[R[a]]+=X[a]; } } else { left=mid+1; mid=(left+right)>>1; for (int a=left;a<=mid;a++) //补后。 { i[L[a]]+=X[a]; i[R[a]]-=X[a]; } } if (!ans) printf("0"); else printf("-1\n%d",ans); return 0; }