1. 程式人生 > >【NOIp 2012】【線段樹】借教室

【NOIp 2012】【線段樹】借教室

描述

在大學期間,經常需要租借教室。大到院系舉辦活動,小到學習小組自習討論,都需要向學校申請借教室。教室的大小功能不同,借教室人的身份不同,借教室的手續也不一樣。
面對海量租借教室的資訊,我們自然希望程式設計解決這個問題。我們需要處理接下來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; }