Thursday, January 17, 2019

[Codeforces] D. The Child and Sequence

Author            : Dipu Kumar Mohanto 
                    CSE, Batch - 6
                    BRUR.
Problem Statement : D. The Child and Sequence
Source            : Codeforces
Category          : Data Structure
Algorithm         : Segment Tree Beats
Verdict           : Accepted

  1. #include "bits/stdc++.h"  
  2.   
  3. using namespace std;  
  4.   
  5. #define FI              freopen("in.txt", "r", stdin)  
  6. #define FO              freopen("out.txt", "w", stdout)  
  7. #define FAST            ios_base::sync_with_stdio(false), cin.tie(NULL), cout.tie(NULL)  
  8.   
  9. #define FOR(i, n)       for (int i = 1; i <= n; i++)  
  10. #define For(i, n)       for (int i = 0; i < n; i++)  
  11. #define ROF(i, n)       for (int i = n; i >= 1; i--)  
  12. #define Rof(i, n)       for (int i = n-1; i >= 0; i--)  
  13. #define FORI(i, n)      for (auto i : n)  
  14.   
  15. #define ll              long long  
  16. #define ull             unsigned long long  
  17. #define vi              vector <int>  
  18. #define vl              vector <ll>  
  19. #define pii             pair <int, int>  
  20. #define pll             pair <ll, ll>  
  21. #define mk              make_pair  
  22. #define ff              first  
  23. #define ss              second  
  24. #define eb              emplace_back  
  25. #define em              emplace  
  26. #define pb              push_back  
  27. #define ppb             pop_back  
  28. #define All(a)          a.begin(), a.end()  
  29. #define memo(a, b)      memset(a, b, sizeof a)  
  30. #define Sort(a)         sort(All(a))  
  31. #define ED(a)           Sort(a), a.erase(unique(All(a)), a.end())  
  32. #define rev(a)          reverse(All(a))  
  33. #define sz(a)           (int)a.size()  
  34. #define max3(a, b, c)   max(a, max(b, c))  
  35. #define min3(a, b, c)   min(a, min(b, c))  
  36. #define maxAll(a)       *max_element(All(a))  
  37. #define minAll(a)       *min_element(All(a))  
  38. #define allUpper(a)     transform(All(a), a.begin(), :: toupper)  
  39. #define allLower(a)     transform(All(a), a.begin(), :: tolower)  
  40. #define endl            '\n'  
  41. #define nl              puts("")  
  42. #define ub              upper_bound  
  43. #define lb              lower_bound  
  44. #define Exp             exp(1.0)  
  45. #define PIE             2*acos(0.0)  
  46. #define Sin(a)          sin(((a)*PIE)/180.0)  
  47. #define EPS             1e-9  
  48.   
  49. #include "ext/pb_ds/assoc_container.hpp"  
  50. #include "ext/pb_ds/tree_policy.hpp"  
  51. using namespace __gnu_pbds;  
  52.   
  53. template <typename T> using orderset = tree <T, null_type, less <T>, rb_tree_tag, tree_order_statistics_node_update>;  
  54.   
  55. #include "ext/rope"  
  56. using namespace __gnu_cxx;  
  57.   
  58. // rope <int> Rope;  
  59.   
  60. // int dr[] = {1, -1, 0, 0}; // 4 Direction  
  61. // int dc[] = {0, 0, 1, -1};  
  62. // int dr[] = {0, 0, 1, -1, 1, 1, -1, -1}; // 8 Direction  
  63. // int dc[] = {1, -1, 0, 0, 1, -1, 1, -1};  
  64. // int dr[] = {-1, 1, -2, -2, -1, 1, 2, 2}; // knight Moves  
  65. // int dc[] = {-2, -2, -1, 1, 2, 2, 1, -1};  
  66.   
  67. #define trace1(x)                           cerr << #x << ": " << x << endl;  
  68. #define trace2(x, y)                        cerr << #x << ": " << x << " | " << #y << ": " << y << endl;  
  69. #define trace3(x, y, z)                     cerr << #x << ": " << x << " | " << #y << ": " << y << " | " << #z << ": " << z << endl;  
  70. #define trace4(a, b, c, d)                  cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << endl;  
  71. #define trace5(a, b, c, d, e)               cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << " | " << #e << ": " << e << endl;  
  72. #define trace6(a, b, c, d, e, f)            cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << " | " << #e << ": " << e << " | " << #f << ": " << f << endl;  
  73.   
  74. inline int setbit(int mask, int pos)        { return mask |= (1 << pos); }  
  75. inline int resetbit(int mask, int pos)      { return mask &= ~(1 << pos); }  
  76. inline int togglebit(int mask, int pos)     { return mask ^= (1 << pos); }  
  77. inline bool checkbit(int mask, int pos)     { return (bool)(mask & (1 << pos)); }  
  78.   
  79. #define popcount(mask)                       __builtin_popcount(mask) // count set bit  
  80. #define popcountLL(mask)                     __builtin_popcountll(mask) // for long long  
  81.   
  82. inline int read()                           { int a; scanf("%d", &a); return a; }  
  83. inline ll readLL()                          { ll a; scanf("%lld", &a); return a; }  
  84. inline double readDD()                      { double a; scanf("%lf", &a); return a; }  
  85.   
  86. template <typename T> string toString(T num) { stringstream ss; ss << num; return ss.str(); }  
  87. int toInt(string s)                          { int num; istringstream iss(s); iss >> num; return num;  }  
  88. ll toLLong(string s)                         { ll num; istringstream iss(s); iss >> num; return num; }  
  89.   
  90. #define inf             1e17  
  91. #define mod             1000000007  
  92.   
  93. static const int maxn = 2e5 + 5;  
  94. static const int logn = 18;  
  95.   
  96. struct segmentTreeBeats  
  97. {  
  98.       int pos;  
  99.       ll maxnum, sum;  
  100.       segmentTreeBeats(int pos = 0, ll maxnum = 0, ll sum = 0) : pos(pos), maxnum(maxnum), sum(sum) {}  
  101.   
  102.       inline void assignLeaf(int pos, ll val)  
  103.       {  
  104.             this->pos = pos;  
  105.             this->maxnum = val;  
  106.             this->sum = val;  
  107.       }  
  108.       inline void marge(const segmentTreeBeats &lft, const segmentTreeBeats &rgt)  
  109.       {  
  110.             if (lft.maxnum > rgt.maxnum) maxnum = lft.maxnum, pos = lft.pos;  
  111.             else maxnum = rgt.maxnum, pos = rgt.pos;  
  112.             sum = lft.sum + rgt.sum;  
  113.       }  
  114. } Tree[maxn << 2];  
  115.   
  116. int n, q;  
  117. ll arr[maxn];  
  118.   
  119. inline void update(int node, int a, int b, int pos, ll val)  
  120. {  
  121.       if (a == pos && b == pos)  
  122.       {  
  123.             Tree[node].assignLeaf(pos, val);  
  124.             return;  
  125.       }  
  126.       int lft = node << 1;  
  127.       int rgt = lft | 1;  
  128.       int mid = (a + b) >> 1;  
  129.       if (pos <= mid) update(lft, a, mid, pos, val);  
  130.       else update(rgt, mid+1, b, pos, val);  
  131.       Tree[node].marge(Tree[lft], Tree[rgt]);  
  132. }  
  133.   
  134. inline segmentTreeBeats query(int node, int a, int b, int i, int j)  
  135. {  
  136.       if (a == i && b == j) return Tree[node];  
  137.       int lft = node << 1;  
  138.       int rgt = lft | 1;  
  139.       int mid = (a + b) >> 1;  
  140.       segmentTreeBeats res = segmentTreeBeats();  
  141.       if (j <= mid)  
  142.       {  
  143.             res.marge(res, query(lft, a, mid, i, j));  
  144.       }  
  145.       else if (i > mid)  
  146.       {  
  147.             res.marge(res, query(rgt, mid+1, b, i, j));  
  148.       }  
  149.       else  
  150.       {  
  151.             segmentTreeBeats p = query(lft, a, mid, i, mid);  
  152.             segmentTreeBeats q = query(rgt, mid+1, b, mid+1, j);  
  153.             res.marge(res, p);  
  154.             res.marge(res, q);  
  155.       }  
  156.       return res;  
  157. }  
  158.   
  159. int main()  
  160. {  
  161.       n = read();  
  162.       q = read();  
  163.       FOR(i, n)  
  164.       {  
  165.             arr[i] = readLL();  
  166.             update(1, 1, n, i, arr[i]);  
  167.       }  
  168.       FOR(_q, q)  
  169.       {  
  170.             int type = read();  
  171.             if (type == 1)  
  172.             {  
  173.                   int l = read();  
  174.                   int r = read();  
  175.                   segmentTreeBeats res = query(1, 1, n, l, r);  
  176.                   printf("%lld\n", res.sum);  
  177.             }  
  178.             else if (type == 2)  
  179.             {  
  180.                   int l = read();  
  181.                   int r = read();  
  182.                   ll x = readLL();  
  183.                   while (1)  
  184.                   {  
  185.                         segmentTreeBeats res = query(1, 1, n, l, r);  
  186.                         if (res.maxnum < x) break;  
  187.                         update(1, 1, n, res.pos, res.maxnum % x);  
  188.                   }  
  189.             }  
  190.             else  
  191.             {  
  192.                   int k = read();  
  193.                   ll x = readLL();  
  194.                   update(1, 1, n, k, x);  
  195.             }  
  196.       }  
  197. }  

No comments:

Post a Comment

Note: Only a member of this blog may post a comment.