[数据结构]火车线路

题目传送门:http://codevs.cn/problem/1291/


题目描述 Description
某列火车行使在C个城市之间(出发的城市编号为1,结束达到的城市的编号为C),假设该列火车有S个座位,现在有R笔预订票的业务。现在想对这R笔业务进行处理,看哪些预定能满足,哪些不能满足。

一笔预定由O、D、N三个整数组成,表示从起点站O到目标站D需要预定N个座位。一笔预定能满足是指该笔业务在行程范围内有能满足的空座位,否则就不能满足。一笔业务不能拆分,也就是起点和终点站不能更改,预定的座位数目也不能更改。所有的预定需求按给出先后顺序进行处理。

请你编写程序,看那些预定业务能满足,那些不能满足。

输入描述 Input Description
输入文件中的第一行为三个整数C、S、R,(1<=c<=60 000, 1<=s<=60 000, 1<=r<=60 000)他们之间用空隔分开。接下来的R行每行为三个整数O、D、N,(1<=o<=d<=c, 1<=n<=s),分别表示每一笔预定业务。

输出描述 Output Description
对第I笔业务,如果能满足,则在输出文件中的第I行输出“T”,否则输出“N”


题解:

一般解决区间维护问题,首选方法都是线段树树状数组
在这一道题里,我们可以维护一个线段树,存储列车在每个站点时的载客量。

之后便要分两种情况讨论:(i,j分别表示当前讨论的两个站点)
1.如果当前情况可以满足(额载人数>=FindMax[i,j-1]+订单人数),那么就对线段树进行区间修改,将[i,j-1]这个区间的值+订单人数,并输出T。
2.如果不能满足,就输出N(什么也不做)。

线段树可以使用延迟标记进行优化,防止超时。


代码:

#include
#include
#include
#define maxn 60005
using namespace std;
int c,s,r;
struct node
{
    int l,r,lazy,max;
};
node tree[maxn*10];
void build(int p,int x,int y)
{
    int mid=(x+y)>>1;
    tree[p].l=x;
    tree[p].r=y;
    tree[p].lazy=0;
    tree[p].max=0;
    if(x1,x,mid);
        build(p<<1|1,mid+1,y);
    }
}
void putDown(int p)
{
    tree[p<<1].lazy+=tree[p].lazy;
    tree[p<<1].max+=tree[p].lazy;
    tree[p<<1|1].lazy+=tree[p].lazy;
    tree[p<<1|1].max+=tree[p].lazy;
    tree[p].lazy=0;
}
void change(int p,int x,int y,int val)
{
    if(tree[p].lazy!=0) putDown(p);
    if(x<=tree[p].l&&tree[p].r<=y)
    {
        tree[p].lazy+=val;
        tree[p].max+=val;
        return;
    }
    int mid=(tree[p].l+tree[p].r)>>1;
    if(x<=mid&&y>=tree[p].l)
    {
        change(p<<1,x,y,val);
    }
    if(y>mid&&x<=tree[p].r)
    {
        change(p<<1|1,x,y,val);
    }
    tree[p].max=max(tree[p<<1].max,tree[p<<1|1].max);
}
int query(int p,int x,int y)
{
    if(tree[p].lazy!=0) putDown(p);
    if(x<=tree[p].l&&tree[p].r<=y)
    {
        return tree[p].max;
    }
    int mid=(tree[p].l+tree[p].r)>>1;
    int lmax=0,rmax=0;
    if(x<=mid&&y>=tree[p].l)
    {
        lmax=query(p<<1,x,y);
    }
    if(y>mid&&x<=tree[p].r)
    {
        rmax=query(p<<1|1,x,y);
    }
    return max(lmax,rmax);
}
int main()
{
    scanf("%d%d%d",&c,&s,&r);
    build(1,1,c);
    while(r--)
    {
        int o,d,n;
        scanf("%d%d%d",&o,&d,&n);
        if(s1,o,d-1)+n) 
        {
            printf("N\n");
        }
        else
        {
            change(1,o,d-1,n);
            printf("T\n");
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构,数据结构,线段树,code-vs)