【线段树】 L3-017 森森快递 (30分) 27分WA代码

L3-017 森森快递 (30分)

森森开了一家快递公司，叫森森快递。因为公司刚刚开张，所以业务路线很简单，可以认为是一条直线上的N个城市，这些城市从左到右依次从0到(N−1)编号。由于道路限制，第i号城市（i=0,⋯,N−2）与第(i+1)号城市中间往返的运输货物重量在同一时刻不能超过C
​i
​​ 公斤。

公司开张后很快接到了Q张订单，其中j张订单描述了某些指定的货物要从S
​j
​​ 号城市运输到T
​j
​​ 号城市。这里我们简单地假设所有货物都有无限货源，森森会不定时地挑选其中一部分货物进行运输。安全起见，这些货物不会在中途卸货。

为了让公司整体效益更佳，森森想知道如何安排订单的运输，能使得运输的货物重量最大且符合道路的限制？要注意的是，发货时间有可能是任何时刻，所以我们安排订单的时候，必须保证共用同一条道路的所有货车的总重量不超载。例如我们安排1号城市到4号城市以及2号城市到4号城市两张订单的运输，则这两张订单的运输同时受2-3以及3-4两条道路的限制，因为两张订单的货物可能会同时在这些道路上运输。

输入格式：

输入在第一行给出两个正整数N和Q（2≤N≤10
​5
​​ , 1≤Q≤10
​5
​​ ），表示总共的城市数以及订单数量。

第二行给出(N−1)个数，顺次表示相邻两城市间的道路允许的最大运货重量C
​i
​​ （i=0,⋯,N−2）。题目保证每个C
​i
​​ 是不超过2
​31
​​ 的非负整数。

接下来Q行，每行给出一张订单的起始及终止运输城市编号。题目保证所有编号合法，并且不存在起点和终点重合的情况。

输出格式：

在一行中输出可运输货物的最大重量。

输入样例：

10 6
0 7 8 5 2 3 1 9 10
0 9
1 8
2 7
6 3
4 5
4 2

输出样例：

7

样例提示：

我们选择执行最后两张订单，即把5公斤货从城市4运到城市2，并且把2公斤货从城市4运到城市5，就可以得到最大运输量7公斤。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

const int maxn=1e5+5;

int n,q;
int vv1,vv2;
int ans;

struct Node
{
    int val,lz;
}tree[maxn<<2];


int getMid(int a,int b){ return (a+b)>>1; }



void build(int l,int r,int rt)
{
    tree[rt].lz=0;
    if(l==r)
    {
        std::cin >> tree[rt].val;
        return ;
        }
    else
    {
        int mid=getMid(l,r);
        build(l,mid,rt<<1);
        build(mid+1,r,rt<<1|1);
    }
    tree[rt].val=std::min(tree[rt<<1].val,tree[rt<<1|1].val);
}

void pushDown(int rt)
{
    if(!tree[rt].lz) return ;
    tree[rt<<1].val-=tree[rt].lz;
    tree[rt<<1|1].val-=tree[rt].lz;
    tree[rt<<1].lz+=tree[rt].lz;
    tree[rt<<1|1].lz+=tree[rt].lz;
    tree[rt].lz=0;
}

void update(int sl,int sr,int l,int r,int rt,int val)
{
    if(l>=sl && r<=sr)
    {
        tree[rt].val-=val;
        tree[rt].lz+=val;
        return ;
    }
    pushDown(rt);
    int mid=getMid(l,r);
    if(sr<=mid) update(sl,sr,l,mid,rt<<1,val);
    else if(sl>mid) update(sl,sr,mid+1,r,rt<<1|1,val);
    else
    {
        update(sl,sr,l,mid,rt<<1,val);
        update(sl,sr,mid+1,r,rt<<1|1,val);
    }
//子树update完之后，根还还还还还还还还要取子树最小值
    tree[rt].val=std::min(tree[rt<<1].val,tree[rt<<1|1].val);
}

int query(int sl,int sr,int l,int r,int rt)
{
    if(l>=sl && r<=sr) return tree[rt].val;
    pushDown(rt);
    int mn=2147483647,mid=getMid(l,r);
    if(sr<=mid) mn=std::min(mn,query(sl,sr,l,mid,rt<<1));
    else if(sl>mid) mn=std::min(mn,query(sl,sr,mid+1,r,rt<<1|1));
    else
    {
        mn=std::min(mn,query(sl,sr,l,mid,rt<<1));
        mn=std::min(mn,query(sl,sr,mid+1,r,rt<<1|1));
    }
    return mn;
}

bool cmp(std::pair<int ,int > a,std::pair<int ,int > b)
{
    if(a.second-a.first==b.second-b.first) return a.first<b.first;
    return a.second-a.first<b.second-b.first;
}

int main()
{
    std::cin >> n >> q;
    std::pair<int ,int > pa[q];
    build(1,n-1,1);
    ans=0;
    for(int i=0; i<q; i++)
    {
        std::cin >> pa[i].first >> pa[i].second;
        if(pa[i].first>pa[i].second) std::swap(pa[i].first,pa[i].second);
//线段树的一个点表示相邻的一段路，所以要second-1
//但由于题目的城市编号从0开始，这里默认从1开始，所以变成first+1
        pa[i].first+=1;
    }
    std::sort(pa,pa+q,cmp);
    for(int i=0; i<q; i++)
    {
        int ch=query(pa[i].first,pa[i].second,1,n-1,1);
        ans+=ch;
        update(pa[i].first,pa[i].second,1,n-1,1,ch);
    }
    std::cout << ans;

    return 0;
}