【第6题】又没长记性：P3130 [USACO15DEC] Counting Haybale P

题目

题目原文请移步下面的链接

• https://www.luogu.com.cn/problem/P3130
  • 参考题解：https://www.luogu.com.cn/problem/solution/P3130
• 标签：OI、USACO、数据结构、线段树

题解

思路

• 本题是线段树裸题，思路就不赘述了；
• 坑点：
  • 我第一次提交时，华丽丽挂了，二进宫，我这次又把scanf和cin混用，后来统一修成改cin就AC掉了。
    • **谨记：**再犯同样的错误，准被收拾

代码

#include <bits/stdc++.h>
using namespace std;
#define endl '\n';

typedef long long LL;

struct edge {
    int l, r;
    LL sum, add, min;
};

struct Segment_Tree {
    vector<edge> t;
    vector<LL> a;

    Segment_Tree(int n) : t(4 * n), a(n + 1){}

    void push_up(int u) {
        t[u].sum = t[u << 1].sum + t[u << 1 | 1].sum;
        t[u].min = min(t[u << 1].min, t[u << 1 | 1].min);
    }

    void push_down(int u) {
        auto &root = t[u], &left = t[u << 1], &right = t[u << 1 | 1];
        if (root.add) {
            left.add += root.add;
            left.sum += (left.r - left.l + 1) * root.add;
            left.min += root.add;
            right.add += root.add;
            right.sum += (right.r - right.l + 1) * root.add;
            right.min += root.add;
            root.add = 0;
        }
    }

    void build(int u, int l, int r) {
        t[u].l = l;
        t[u].r = r;
        if (l == r) {
            t[u].sum = a[l];
            t[u].min = a[l];
            return;
        }
        int mid = l + r >> 1;
        build(u << 1, l, mid);
        build(u << 1 | 1, mid + 1, r);
        push_up(u);
    }

    LL query(int u, int l, int r) {
        if (t[u].l >= l && t[u].r <= r) {
            return t[u].sum;
        }
        push_down(u);
        int mid = t[u].l + t[u].r >> 1;
        LL s = 0;
        if (l <= mid) {
            s = query(u << 1, l, r);
        }
        if (r > mid) {
            s += query(u << 1 | 1, l, r);
        }
        return s;
    }

    LL minnumber(int u, int l, int r) {
        if (t[u].l >= l && t[u].r <= r) {
            return t[u].min;
        }
        push_down(u);
        int mid = t[u].l + t[u].r >> 1;
        LL s = 2147483647;
        if (l <= mid) {
            s = min(s, minnumber(u << 1, l, r));
        }
        if (r > mid) {
            s = min(s, minnumber(u << 1 | 1, l, r));
        }
        return s;
    }

    void cnt(int u, int l, int r,  LL v) {
        if (t[u].l >= l && t[u].r <= r) {
            t[u].sum += (t[u].r - t[u].l + 1) * v;
            t[u].add += v;
            t[u].min += v;
        } else {
            push_down(u);
            int mid = t[u].l + t[u].r >> 1;
            if (l <= mid) {
                cnt(u << 1, l, r, v);
            }
            if (r > mid) {
                cnt(u << 1 | 1, l, r, v);
            }
            push_up(u);
        }
    }
};
void best_coder() {
    int n, m;
    cin >> n >> m;
    Segment_Tree st(n);
    for (int i = 1; i <= n; ++i) {
        cin >> st.a[i];
    }
    st.build(1, 1, n);
    for (int i = 0; i < m; ++i) {
        char c;
        int l, r;
        cin >> c >> l >> r;
        if (c == 'M') {
            cout << st.minnumber(1, l, r) << endl;
        } else if (c == 'S') {
            cout << st.query(1, l, r) << endl;
        } else {
            LL v;
            cin >> v;
            st.cnt(1, l, r, v);
        }
    }
}

void happy_coder() {
}

int main() {
    // 提升cin、cout效率
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    // 小码匠
    best_coder();

    // 最优解
    // happy_coder();

    // 返回
    return 0;
}

小码匠上初中后，平时作业比较多，课余时间都奉献给一方键盘和数学公式。

有时间我会把文章发布到CSDN，和更多的小伙伴分享我的学习心路。