【第3题】P3372 【模板】线段树 1

最新推荐文章于 2025-01-31 14:12:47 发布

原创最新推荐文章于 2025-01-31 14:12:47 发布 · 156 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

图论专栏收录该内容

2 篇文章

订阅专栏

文章讲述了在解决洛谷P3372题目时遇到的C++程序运行错误，问题出在线段树的实现上，由于开销导致的越界。作者通过调试和他人帮助定位到问题代码，并成功修复，使得程序能在Windows环境下正常运行。

题目

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

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

坑点

错误

CLion C++ Process finished with exit code -1073740940 (0xC0000374)

现象
- Step1: 调试的时候，执行到第12行代码就会退出，最初我是用scanf读取数据，后来改成cin一样不行。
- Step2: 后来求助老码农，帮我看看，他在mac本上执行没问题，可以正常跑，在我的windows上就挂了，后来他尝试注释掉第8行，就能正常读取数据。所以断定是第8行方法又问题。
- Step3: 问题找到了，接下来就是我小码匠闪亮登场时间，自己挖的坑，自己还是要填了。是因为树组开销了，越界导致。
```
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);
    int q, x, y;
    long long v;
    for (int i = 0; i < m; ++i) {
        cin >> q;
        if (q == 1) {
            cin >> x >> y >> v;
            st.cnt(1,x, y, v);

        } else {
            cin >> x >> y;
            cout << st.query(1, x, y) << endl;
        }
    }
}
```

题解

思路

思路在这就不分享了，网络上很多，我也是看资料，边学边调试。

代码

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;

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

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

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

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

    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;
            right.add += root.add;
            right.sum += (right.r - right.l + 1) * 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];
            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;
    }

    void modify(int u, int x, LL v) {
        if (t[u].l == x && t[u].r == x) {
            t[u].sum = v;
        } else {
            int mid = (t[u].l + t[u].r) >> 1;
            if (x <= mid) {
                modify(u << 1, x, v);
            } else {
                modify(u << 1 | 1, x, v);
            }
            push_up(u);
        }
    }

    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;
        } 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);
    int q, x, y;
    long long v;
    for (int i = 0; i < m; ++i) {
        cin >> q;
        if (q == 1) {
            cin >> x >> y >> v;
            st.cnt(1,x, y, v);

        } else {
            cin >> x >> y;
            cout << st.query(1, x, y) << endl;
        }
    }
}

void happy_coder() {
}

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

    // 小码匠
    best_coder();

    // 最优解
    // happy_coder();

    // 返回
    return 0;
}