【NOIP2012借教室】（线段树区间操作）

最新推荐文章于 2022-06-24 14:48:25 发布

kaqiur

最新推荐文章于 2022-06-24 14:48:25 发布

阅读量565

点赞数

分类专栏： NOIP 文章标签：线段树 NOIP

本文链接：https://blog.csdn.net/kaqiur/article/details/52170560

版权

NOIP 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

1266. [NOIP2012] 借教室

★★☆ 输入文件： classrooms.in 输出文件： classrooms.out 简单对比
时间限制：1 s 内存限制：128 MB

【题目描述】

在大学期间，经常需要租借教室。大到院系举办活动，小到学习小组自习讨论，都需要向学校申请借教室。教室的大小功能不同，借教室人的身份不同，借教室的手续也不一样。

面对海量租借教室的信息，我们自然希望编程解决这个问题。

我们需要处理接下来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，第二行输出需要修改订单的申请人编号。

【样例输入】

【样例输出】

-1
2

【输入输出样例说明】

第 1 份订单满足后，4 天剩余的教室数分别为 0，3，2，3。第 2 份订单要求第 2 天到第 4 天每天提供 3 个教室，而第 3 天剩余的教室数为 2，因此无法满足。分配停止，通知第 2 个申请人修改订单。

【数据范围】

对于 10%的数据，有1 ≤ n,m ≤ 10；

对于 30%的数据，有1 ≤ n,m ≤ 1000；

对于 70%的数据，有1 ≤ n,m ≤ 10^5；

对于 100%的数据，有1 ≤ n,m ≤ 10^6,0 ≤ ri,dj ≤ 10^9,1 ≤ sj ≤ tj ≤ n。

思路：很容易想到线段树，裸的线段树区间操作。但是要稍微优化避免超时。其次线段树太容易写砸，代码中表明以警示。

各种小细节都要注意准确。

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#define maxn 4123456
#define inf 1000000009
using namespace std;
struct data{
    int l,r,minn,ada;
};
data tree[maxn];
int a[maxn],n,m,pos=0;
bool tmp=1;
void pushdown(int node){
    tree[node<<1].ada+=tree[node].ada;
    tree[(node<<1)+1].ada+=tree[node].ada;
    tree[node].ada=0;
}
void pushup(int node){
    tree[node].minn=min(tree[node<<1].minn-tree[node<<1].ada,tree[(node<<1)+1].minn-tree[(node<<1)+1].ada)；<span style="font-size: 16px; font-family: serif;">//tree[node<<1].minn-tree[node].ada  !!??</span>

}
void build(int node,int x,int y){
    if(x==y){tree[node].minn=a[x];tree[node].l=x;tree[node].r=y;tree[node].ada=0;return;}
    int mid=(x+y)>>1;
    build(node<<1,x,mid);
    build((node<<1)+1,mid+1,y);
    tree[node].minn=min(tree[node<<1].minn,tree[(node<<1)+1].minn);
    tree[node].l=x;tree[node].r=y;tree[node].ada=0;
}
void add(int node,int x,int y,int ad){
    if(x<=tree[node].l && y>=tree[node].r){tree[node].ada+=ad;if(tree[node].minn-tree[node].ada<0)tmp=0;return;}
    if(tree[node].minn-tree[node].ada<0){tmp=0;return;}
	if(tree[node].l>y || tree[node].r<x)return;
	pushdown(node);
    int mid=(tree[node].l+tree[node].r)>>1;
    if(x<=mid)add(node<<1,x,y,ad);
    if(y>mid)add((node<<1)+1,x,y,ad);   //查询和修改的区间不要改变！！！
    pushup(node);
    return;
}
int main(){
	freopen("classrooms.in","r",stdin);
	freopen("classrooms.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)scanf("%d",&a[i]);
    build(1,1,n);
    int d,s,t;
    for(int i=1;i<=m;++i){
        scanf("%d%d%d",&d,&s,&t);
        if(pos)continue;
        add(1,s,t,d);   //修改和判断和在一起写节约时间
        if(!tmp && !pos)pos=i;
    }
    if(pos)printf("-1\n%d\n",pos);
    else printf("0\n");
    return 0;
}

另一种方法（前缀和+二分）http://hzwer.com/2959.html