PTA森森快递 线段树维护区间最小值+贪心

参考了一下这位博主的文章，自己在此基础上稍做了修改
大概思路体现在代码注释中了

//贪心加线段树，每次选择订单都将订单所经过路上的最小重量作为运输重量
//选择重量后将所涉及的区间做lazy标记，减去相应的重量
//并且优先选择区间小的，保证结果的和最大
#include <bits/stdc++.h>
#define inf 0x3f3f3f3f3f
using namespace std;
typedef long long ll;
inline void scf(int &a) { scanf("%d", &a); }
inline void scf(int &a, int &b) { scanf("%d%d", &a, &b); }

struct point
{
    int x, y;//表示起始和末尾地点
} p[100005];

//lazy[i]标记表示该段已经有重量为lazy[i]的货物会经过，那么最小值就要相应的减去货物的重量
int num[100005], tree[400005], lazy[400005];//为了保证在建树的时候不会下标越界，开四倍大小的数组比较保险

bool cmp(point a, point b) //区间小的排在前面
{
    if (a.y == b.y)
        return a.x < b.x;
    return a.y < b.y;
}

void build(int l, int r, int i = 1) //建立线段树，指定左右范围
{
    lazy[i] = 0;
    if (l == r)
    {
        tree[i] = num[l];
        return;
    }
    int mid = (l + r) >> 1;
    build(l, mid, i << 1);
    build(mid + 1, r, i << 1 | 1);
    tree[i] = min(tree[i << 1], tree[i << 1 | 1]);
}

inline void pushdown(int i) //下传标记，i是tree数组的下标
{
    if (lazy[i] == 0)
        return;
    tree[i << 1] -= lazy[i];
    tree[i << 1 | 1] -= lazy[i];
    lazy[i << 1] += lazy[i];
    lazy[i << 1 | 1] += lazy[i];
    lazy[i] = 0;
}

void updata(int L, int R, int l, int r, int d, int i = 1) //更新区间，大写LR表示当前区间，小写lr表示目标区间，d表示要增加的值
{
    if (l <= L && r >= R)
    {
        lazy[i] += d;
        tree[i] -= d;
        return;
    }
    pushdown(i);
    int mid = (L + R) >> 1;
    if (mid >= l)
        updata(L, mid, l, r, d, i << 1);
    if (mid < r)
        updata(mid + 1, R, l, r, d, i << 1 | 1);
    tree[i] = min(tree[i << 1], tree[i << 1 | 1]);
}

ll query(int L, int R, int l, int r, int i = 1)//返回查询区间的最小值
{
    if (l <= L && r >= R) return tree[i];
    int mid = (L + R) >> 1;
    ll ret = inf;
    pushdown(i);
    if (mid >= l) ret = min(ret, query(L, mid, l, r, i << 1));
    if (mid < r)  ret = min(ret, query(mid + 1, R, l, r, i << 1 | 1));
    return ret;
}

int main()
{
    int m, n;
    ll ans = 0;
    scf(m, n);
    for (int i = 0; i < m - 1; i++)
    {
        scf(num[i]); //num[i]表示从i到i+1的最大重量
    }
    build(0, m - 2);
    for (int i = 0; i < n; i++)
    {
        scf(p[i].x, p[i].y);
        if (p[i].x > p[i].y)
            swap(p[i].x, p[i].y);
        p[i].y--; //为了跟num的表示方式同步,原来的p[i].x->p[i].y涉及num中的num[x]->num[y-1]
    }
    sort(p, p + n, cmp);
    for (int i = 0; i < n; i++)
    {
        ll tmp = query(0, m - 2, p[i].x, p[i].y);
        ans += tmp;//如果这段区间最小值为0，即不能再运输任何重量的货物了，tmp=0，相加不会影响结果
        updata(0, m - 2, p[i].x, p[i].y, tmp);
    }
    printf("%lld", ans);
    return 0;
}

输入样例：

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

输出样例：

7