Advertisement
El_GEMMY

wavio sequence (segment tree)

Jun 4th, 2022
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 5.70 KB | None | 0 0
  1. // Those who cannot remember the past are
  2. // condemned to repeat it (use DP -_-)
  3. // - George Santayana
  4.  
  5. #include <bits/stdc++.h>
  6. #include <ext/pb_ds/assoc_container.hpp>
  7. #include <ext/pb_ds/tree_policy.hpp>
  8.  
  9. using namespace std;
  10. using namespace __gnu_pbds;
  11.  
  12. #define all(v) v.begin(), v.end()
  13. #define rall(v) v.rbegin(), v.rend()
  14. #define ll long long
  15. #define ull unsigned long long
  16. #define MOD 1000000007
  17. #define PI 3.14159265
  18. #define ceil(a, b) (((a) / (b)) + ((a) % (b) ? 1 : 0))
  19. #define imin INT_MIN
  20. #define imax INT_MAX
  21. #define llmax LLONG_MAX
  22. #define llmin LLONG_MIN
  23. #define inf 2000000000
  24. #define nl '\n'
  25. #define ppcnt __builtin_popcount
  26. #define ppcntll __builtin_popcountll
  27. #define clz __builtin_clz
  28. #define clzll __builtin_clzll
  29. #define ctz __builtin_ctz
  30. #define ctzll __builtin_ctzll
  31. #define modulo(a, b, mod) ((((a) % (mod)) + ((b) % (mod))) % (mod))
  32. #define cnte(v, x) count(all(v), (x))
  33. #define mine(v) min_element(all(v))
  34. #define maxe(v) max_element(all(v))
  35. #define updmin(a, b) a = min(a, b)
  36. #define updmax(a, b) a = max(a, b)
  37. #define findmod(x, m) x = ((x) % (m) + (m)) % m
  38. #define getmod(x, m) ((x) % (m) + (m)) % (m)
  39. #define debug(x) cout << "x: " << (x) << nl;
  40. #define debug2(x, y) cout << "x: " << (x) << " y: " << y << nl;
  41. #define ordered_set tree<int, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update>
  42. #define ordered_map tree<int, int, less<>, rb_tree_tag, tree_order_statistics_node_update>
  43.  
  44. //vector<int> dx = {0, 0, 1, -1, 1, 1, -1, -1}, dy = {1, -1, 0, 0, 1, -1, 1, -1};
  45. //vector<int> dx = {0, 0, 1, -1}, dy = {1, -1, 0, 0};
  46.  
  47. template<typename T = int> istream& operator>>(istream& in, vector<pair<int, int>>& v){
  48.     for (auto& [x, y] : v) in >> x >> y;
  49.     return in;
  50. }
  51.  
  52. template<typename T = int> istream& operator>>(istream& in, vector<T>& v){
  53.     for (T& i : v) in >> i;
  54.     return in;
  55. }
  56.  
  57. template<typename T = int> ostream& operator<<(ostream& out, const vector<T>& v){
  58.     for (const T& x : v)
  59.         out << x << ' ';
  60.     return out;
  61. }
  62.  
  63. template<typename T = pair<int, int>> ostream& operator << (ostream& out, const vector<pair<int, int>>& v){
  64.     for(auto& [x, y] : v){
  65.         out << x << ' ' << y << nl;
  66.     }
  67.     return out;
  68. }
  69.  
  70. void Start_Crushing() {
  71.     ios_base::sync_with_stdio(false);
  72.     cin.tie(nullptr);
  73.     cout.tie(nullptr);
  74. #ifndef ONLINE_JUDGE
  75.     freopen("input.txt", "r", stdin);
  76.     freopen("output.txt", "w", stdout);
  77. #endif
  78. }
  79.  
  80. struct segmentTree{
  81.     vector<int> tree;
  82.     int size, n;
  83.  
  84.     explicit segmentTree(int n){
  85.         this -> n = n;
  86.         size = 1;
  87.         while(size <= n)
  88.             size *= 2;
  89.         tree.assign(2 * size, 0);
  90.     }
  91.  
  92.     void assign(int new_n, int value){
  93.         this -> n = new_n;
  94.         size = 1;
  95.         while(size <= n)
  96.             size *= 2;
  97.         tree.assign(2 * size, value);
  98.     }
  99.  
  100.     int merge(int a, int b){
  101.         return max(a, b);
  102.     }
  103.  
  104.     void build(vector<int>& v, int idx, int tl, int tr){
  105.         if(tl >= n) return;
  106.         if(tl == tr){
  107.             tree[idx] = v[tl];
  108.             return;
  109.         }
  110.         int mid = tl - (tl - tr) / 2;
  111.         build(v, idx * 2 + 1, tl, mid);
  112.         build(v, idx * 2 + 2, mid + 1, tr);
  113.         tree[idx] = merge(tree[idx * 2 + 1], tree[idx * 2 + 2]);
  114.     }
  115.  
  116.     void build(vector<int>& v){
  117.         this -> n = int(v.size());
  118.         for(size = 1; size < n; size *= 2);
  119.         tree.assign(size * 2, 0);
  120.         build(v, 0, 0, size - 1);
  121.     }
  122.  
  123.     void set(int i, int val, int idx, int tl, int tr){
  124.         if(i > tr or i < tl) return;
  125.         if(tl == tr){
  126.             tree[idx] = val;
  127.             return;
  128.         }
  129.         int mid = tl - (tl - tr) / 2;
  130.         set(i, val, idx * 2 + 1, tl, mid);
  131.         set(i, val, idx * 2 + 2, mid + 1, tr);
  132.         tree[idx] = merge(tree[idx * 2 + 1], tree[idx * 2 + 2]);
  133.     }
  134.  
  135.     void set(int i, int val){
  136.         set(i, val, 0, 0, size - 1);
  137.     }
  138.  
  139.     int query(int l, int r, int idx, int tl, int tr){
  140.         if(tr < l or tl > r) return 0;
  141.         if(tl >= l and tr <= r){
  142.             return tree[idx];
  143.         }
  144.  
  145.         int mid = tl - (tl - tr) / 2;
  146.         int left = query(l, r, idx * 2 + 1, tl, mid);
  147.         int right = query(l, r, idx * 2 + 2, mid + 1, tr);
  148.         return merge(left, right);
  149.     }
  150.  
  151.     int query(int l, int r){
  152.         return query(l, r, 0, 0, size - 1);
  153.     }
  154. };
  155.  
  156. vector<int> co;
  157.  
  158. int idx(int x){
  159.     return int(upper_bound(all(co), x) - co.begin()) + 1;
  160. }
  161.  
  162. void solve(){
  163.     int n;
  164.     while(cin >> n) {
  165.         vector<int> v(n);
  166.         cin >> v;
  167.  
  168.         co = v;
  169.         sort(all(co));
  170.         co.erase(unique(all(co)), co.end());
  171.  
  172.         int ans = 1;
  173.         segmentTree st(n);
  174.  
  175.         vector<int> lis(n), lds(n);
  176.  
  177.         for (int i = 0; i < n; i++) {
  178.             int curr = st.query(1, idx(v[i]) - 1) + 1;
  179.             st.set(idx(v[i]), curr);
  180.             lis[i] = curr;
  181.         }
  182.         st.assign(n, 0);
  183.         reverse(all(v));
  184.  
  185.         for (int i = 0; i < n; i++) {
  186.             int curr = st.query(1, idx(v[i]) - 1) + 1;
  187.             st.set(idx(v[i]), curr);
  188.             lds[n - i - 1] = curr;
  189.         }
  190.  
  191.         for (int i = 0; i < n; i++) {
  192.             int len = min(lis[i], lds[i]);
  193.             updmax(ans, len * 2 - 1);
  194.         }
  195.         cout << ans << nl;
  196.     }
  197. }
  198.  
  199. int main(){
  200.     Start_Crushing();
  201. //    freopen("perm.in", "r", stdin);
  202.  
  203.     int t = 1;
  204. //    /*Multiple test cases?*/ cin >> t;
  205.     while (t--) {
  206.         solve();
  207. //        if(!t)
  208. //            break;
  209. //        cout << nl;
  210.     }
  211.  
  212. //    for(int tc = 1; tc <= t; tc++){
  213. //        cout << "Case #" << tc << ": ";
  214. //        solve();
  215. //        if(tc != t)
  216. //            cout << nl;
  217. //    }
  218.  
  219.     return 0;
  220. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement