【NOIp 2012】【線段樹】借教室
阿新 • • 發佈:2019-02-05
描述
在大學期間,經常需要租借教室。大到院系舉辦活動,小到學習小組自習討論,都需要向學校申請借教室。教室的大小功能不同,借教室人的身份不同,借教室的手續也不一樣。
面對海量租借教室的資訊,我們自然希望程式設計解決這個問題。我們需要處理接下來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,第二行輸出需要修改訂單的申請人編號。
程式碼
線段樹模板套一套
#include <cstdio>
#include <cstring>
#include <algorithm>
#define L(a) (a<<1)
#define R(a) (a<<1|1)
#define REP(a,b) for(int a=1;a<=b;a++)
#define maxn 1000030
using namespace std;
struct node{
int lson,rson,val,lazy;
}t[maxn*3];
int n,m,r[maxn],ans=0;
bool flag=1;
int read(){
int x=0,f=1; char ch=getchar();
while (ch<'0'||ch>'9'){
if(ch=='-') f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9'){
x=x*10+ch-'0'; ch=getchar();
}
return x*f;
}
void build_tree(int p,int l,int r){
t[p].lson=l; t[p].rson=r; t[p].lazy=0;
if(l==r) {
t[p].val=read(); return;
} else{
int mid=(l+r)>>1;
build_tree(L(p),l,mid);
build_tree(R(p),mid+1,r);
t[p].val=min(t[L(p)].val,t[R(p)].val);
return;
}
}
void pushdown(int p){
if(t[p].lson==t[p].rson||t[p].lazy==0) return;
t[L(p)].lazy+=t[p].lazy; t[R(p)].lazy+=t[p].lazy;
t[L(p)].val-=t[p].lazy; t[R(p)].val-=t[p].lazy;
t[p].lazy=0;
t[p].val=min(t[L(p)].val,t[R(p)].val);
return;
}
void modify(int p,int l,int r,int d){
pushdown(p);
if(t[p].lson==l&&t[p].rson==r){
t[p].lazy+=d;
t[p].val-=d;
if(t[p].val<0) flag=0;
return;
} else {
int mid=(t[p].lson+t[p].rson)>>1;
if(r<=mid) modify(L(p),l,r,d);
else if(l>mid) modify(R(p),l,r,d);
else{
modify(L(p),l,mid,d);
modify(R(p),mid+1,r,d);
}
t[p].val=min(t[L(p)].val,t[R(p)].val);
//if(t[p].val<0) flag=0;
return;
}
}
int query(int p,int l,int r){
if(t[p].lson==l&&t[p].rson==r) return t[p].val;
else {
int mid=(l+r)>>1;
if(r<=mid) return query(L(p),l,r);
else if(l>mid) return query(R(p),l,r);
else return min(query(L(p),l,mid),query(R(p),mid+1,r));
}
}
int main(){
scanf("%d%d",&n,&m);
build_tree(1,1,n);
while((++ans)<=m){
int d=read(),s=read(),T=read();
modify(1,s,T,d);
if(!flag) break;
}
if(flag) puts("0");
else printf("-1\n%d\n",ans);
return 0;
}