tien_noob

Segment Tree - find maxsum between a[x] -> a[y]

May 6th, 2021 (edited)
131
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.76 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. const long long N = 5e4, lim = -1e9;
  4. struct T
  5. {
  6.     long long prefix, sum, suffix, best;
  7. };
  8. T Tree[4*N+1];
  9. long long n, m, a[N+1];
  10. T Connect(const T & u, const T & v)
  11. {
  12.     T last;
  13.     last.sum = u.sum + v.sum;
  14.     last.prefix = max(u.prefix, u.sum + v.prefix);
  15.     last.suffix = max(v.suffix, v.sum + u.suffix);
  16.     last.best = max(u.best, max(v.best, u.suffix + v.prefix));
  17.     return last;
  18. }
  19. void BuildTree(int id, int TreeL, int TreeR)
  20. {
  21.     if (TreeL == TreeR)
  22.     {
  23.         Tree[id].prefix = Tree[id].sum = Tree[id].suffix = Tree[id].best = a[TreeL];
  24.         return ;
  25.     }
  26.     int mid = (TreeL + TreeR)/2;
  27.     BuildTree(id * 2, TreeL, mid);
  28.     BuildTree(id * 2 + 1, mid + 1, TreeR);
  29.     Tree[id] = Connect(Tree[id * 2], Tree[id * 2 + 1]);
  30. }
  31. T Get(int id, int TreeL, int TreeR, int l, int r)
  32. {
  33.     if (l > r)
  34.     {
  35.         return {0, lim, lim, lim};
  36.     }
  37.     if (l == TreeL && r == TreeR)
  38.     {
  39.         return Tree[id];
  40.     }
  41.     int mid = (TreeL + TreeR)/2;
  42.     return Connect(Get(id * 2, TreeL, mid, l, min(mid, r)), Get(id * 2 + 1, mid + 1, TreeR, max(l, mid + 1), r));
  43. }
  44. void read()
  45. {
  46.     cin >> n;
  47.     for (long long i = 1; i <= n; ++ i)
  48.     {
  49.         cin >> a[i];
  50.     }
  51.     BuildTree(1, 1, n);
  52. }
  53. void solve()
  54. {
  55.     cin >> m;
  56.     long long x, y;
  57.     while (m--)
  58.     {
  59.         cin >> x >> y;
  60.         cout << Get(1, 1, n, x, y).best << '\n';
  61.     }
  62. }
  63. int main()
  64. {
  65.    ios_base::sync_with_stdio(false);
  66.    cin.tie(nullptr);
  67.    //freopen(task".INP", "r", stdin);
  68.    //freopen(task".OUT", "w", stdout);
  69.    long long t = 1;
  70.    bool typetest = false;
  71.    if (typetest)
  72.    {
  73.       cin >> t;
  74.    }
  75.    for (long long __ = 1; __ <= t; ++ __)
  76.    {
  77.       read();
  78.       solve();
  79.    }
  80. }
  81.  
Add Comment
Please, Sign In to add comment