Thursday, January 24, 2019

[toph.co] Sofdor Ali and Bracket Sequence

Author            : Dipu Kumar Mohanto 
                    CSE, Batch - 6
                    BRUR.
Problem Statement : Sofdor Ali and Bracket Sequence
Source            : toph.co
Category          : String, Data Structure
Algorithm         : Suffix Array, Segment Tree
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. #define REP(i, a, b)    for (int i = a; i <= b; i++)  
  15.   
  16. #define ll              long long  
  17. #define ull             unsigned long long  
  18. #define vi              vector <int>  
  19. #define vl              vector <ll>  
  20. #define pii             pair <int, int>  
  21. #define pll             pair <ll, ll>  
  22. #define mk              make_pair  
  23. #define ff              first  
  24. #define ss              second  
  25. #define eb              emplace_back  
  26. #define em              emplace  
  27. #define pb              push_back  
  28. #define ppb             pop_back  
  29. #define All(a)          a.begin(), a.end()  
  30. #define memo(a, b)      memset(a, b, sizeof a)  
  31. #define Sort(a)         sort(All(a))  
  32. #define ED(a)           Sort(a), a.erase(unique(All(a)), a.end())  
  33. #define rev(a)          reverse(All(a))  
  34. #define sz(a)           (int)a.size()  
  35. #define max3(a, b, c)   max(a, max(b, c))  
  36. #define min3(a, b, c)   min(a, min(b, c))  
  37. #define maxAll(a)       *max_element(All(a))  
  38. #define minAll(a)       *min_element(All(a))  
  39. #define allUpper(a)     transform(All(a), a.begin(), :: toupper)  
  40. #define allLower(a)     transform(All(a), a.begin(), :: tolower)  
  41. #define endl            '\n'  
  42. #define nl              puts("")  
  43. #define ub              upper_bound  
  44. #define lb              lower_bound  
  45. #define Exp             exp(1.0)  
  46. #define PIE             2*acos(0.0)  
  47. #define Sin(a)          sin(((a)*PIE)/180.0)  
  48. #define EPS             1e-9  
  49.   
  50. #include "ext/pb_ds/assoc_container.hpp"  
  51. #include "ext/pb_ds/tree_policy.hpp"  
  52. using namespace __gnu_pbds;  
  53.   
  54. template <typename T> using orderset = tree <T, null_type, less <T>, rb_tree_tag, tree_order_statistics_node_update>;  
  55.   
  56. #include "ext/rope"  
  57. using namespace __gnu_cxx;  
  58.   
  59. // rope <int> Rope;  
  60.   
  61. // int dr[] = {1, -1, 0, 0}; // 4 Direction  
  62. // int dc[] = {0, 0, 1, -1};  
  63. // int dr[] = {0, 0, 1, -1, 1, 1, -1, -1}; // 8 Direction  
  64. // int dc[] = {1, -1, 0, 0, 1, -1, 1, -1};  
  65. // int dr[] = {-1, 1, -2, -2, -1, 1, 2, 2}; // knight Moves  
  66. // int dc[] = {-2, -2, -1, 1, 2, 2, 1, -1};  
  67.   
  68. #define trace1(x)                           cerr << #x << ": " << x << endl;  
  69. #define trace2(x, y)                        cerr << #x << ": " << x << " | " << #y << ": " << y << endl;  
  70. #define trace3(x, y, z)                     cerr << #x << ": " << x << " | " << #y << ": " << y << " | " << #z << ": " << z << endl;  
  71. #define trace4(a, b, c, d)                  cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << endl;  
  72. #define trace5(a, b, c, d, e)               cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << " | " << #e << ": " << e << endl;  
  73. #define trace6(a, b, c, d, e, f)            cerr << #a << ": " << a << " | " << #b << ": " << b << " | " << #c << ": " << c << " | " << #d << ": " << d << " | " << #e << ": " << e << " | " << #f << ": " << f << endl;  
  74.   
  75. inline int setbit(int mask, int pos)        { return mask |= (1 << pos); }  
  76. inline int resetbit(int mask, int pos)      { return mask &= ~(1 << pos); }  
  77. inline int togglebit(int mask, int pos)     { return mask ^= (1 << pos); }  
  78. inline bool checkbit(int mask, int pos)     { return (bool)(mask & (1 << pos)); }  
  79.   
  80. #define popcount(mask)                       __builtin_popcount(mask) // count set bit  
  81. #define popcountLL(mask)                     __builtin_popcountll(mask) // for long long  
  82.   
  83. inline int read()                           { int a; scanf("%d", &a); return a; }  
  84. inline ll readLL()                          { ll a; scanf("%lld", &a); return a; }  
  85. inline double readDD()                      { double a; scanf("%lf", &a); return a; }  
  86.   
  87. template <typename T> string toString(T num) { stringstream ss; ss << num; return ss.str(); }  
  88. int toInt(string s)                          { int num; istringstream iss(s); iss >> num; return num;  }  
  89. ll toLLong(string s)                         { ll num; istringstream iss(s); iss >> num; return num; }  
  90.   
  91. #define inf             1e7  
  92. #define mod             1000000007  
  93.   
  94. static const int maxn = 2e5 + 5;  
  95. static const int logn = 21;  
  96.   
  97. struct segmentTree  
  98. {  
  99.       int open, close;  
  100.       segmentTree(int open = 0, int close = 0) :  
  101.             open(open), close(close) {}  
  102.   
  103.       inline void assignLeaf(char ch)  
  104.       {  
  105.             open = close = 0;  
  106.             if (ch == '(') ++open;  
  107.             else ++close;  
  108.       }  
  109.       inline void marge(const segmentTree &lft, const segmentTree &rgt)  
  110.       {  
  111.             int mini = min(lft.open, rgt.close);  
  112.             open = lft.open + rgt.open - mini;  
  113.             close = lft.close + rgt.close - mini;  
  114.       }  
  115.       inline bool balanced()  
  116.       {  
  117.             return (open == 0 && close == 0);  
  118.       }  
  119. } Tree[maxn << 2];  
  120.   
  121. string s;  
  122. bool balancedPos[maxn];  
  123.   
  124. inline void build(int node, int a, int b)  
  125. {  
  126.       if (a == b)  
  127.       {  
  128.             Tree[node].assignLeaf(s[a-1]);  
  129.             return;  
  130.       }  
  131.       int lft = node << 1;  
  132.       int rgt = lft | 1;  
  133.       int mid = (a + b) >> 1;  
  134.       build(lft, a, mid);  
  135.       build(rgt, mid+1, b);  
  136.       Tree[node].marge(Tree[lft], Tree[rgt]);  
  137. }  
  138.   
  139. inline segmentTree query(int node, int a, int b, int i, int j)  
  140. {  
  141.       if (a == i && b == j) return Tree[node];  
  142.       int lft = node << 1;  
  143.       int rgt = lft | 1;  
  144.       int mid = (a + b) >> 1;  
  145.       segmentTree res = segmentTree();  
  146.       if (j <= mid)  
  147.       {  
  148.             segmentTree p = query(lft, a, mid, i, j);  
  149.             res.marge(res, p);  
  150.       }  
  151.       else if (i > mid)  
  152.       {  
  153.             segmentTree q = query(rgt, mid+1, b, i, j);  
  154.             res.marge(res, q);  
  155.       }  
  156.       else  
  157.       {  
  158.             segmentTree p = query(lft, a, mid, i, mid);  
  159.             segmentTree q = query(rgt, mid+1, b, mid+1, j);  
  160.             res.marge(res, p);  
  161.             res.marge(res, q);  
  162.       }  
  163.       return res;  
  164. }  
  165.   
  166. struct entry  
  167. {  
  168.       int first, second;  
  169.       int p;  
  170.       entry(int ff = 0, int ss = 0, int pp = 0) : first(ff), second(ss), p(pp) {}  
  171.       friend bool operator < (const entry &A, const entry &B)  
  172.       {  
  173.             if (A.first == B.first) return A.second < B.second;  
  174.             return A.first < B.first;  
  175.       }  
  176. } L[maxn];  
  177.   
  178. int P[logn][maxn], B[maxn];  
  179. int p, q;  
  180. int step, lcp[maxn];  
  181.   
  182. inline int comp2(const int &a, const int &b)  
  183. {  
  184.       return P[step-1][a] < P[step-1][b];  
  185. }  
  186.   
  187. inline void suffixArray(const string &s)  
  188. {  
  189.       int len = s.size();  
  190.       for (int i = 0; i < len; i++) P[0][i] = s[i] - '0';  
  191.       step = 1;  
  192.       int cnt = 1;  
  193.       for ( ; cnt>>1 < len; step++, cnt <<= 1)  
  194.       {  
  195.             for (int i = 0; i < len; i++)  
  196.             {  
  197.                   L[i].first = P[step-1][i];  
  198.                   L[i].second = (i + cnt) < len ? P[step-1][i+cnt] : -1;  
  199.                   L[i].p = i;  
  200.             }  
  201.             sort(L, L+len);  
  202.             for (int i = 0; i < len; i++)  
  203.             {  
  204.                   if (i > 0 && L[i].first == L[i-1].first && L[i].second == L[i-1].second)  
  205.                         P[step][ L[i].p ] = P[step][ L[i-1].p ];  
  206.                   else  
  207.                         P[step][ L[i].p ] = i;  
  208.             }  
  209.       }  
  210.       for (int i = 0; i < len; i++) B[i] = i;  
  211.       sort(B, B+len, comp2);  
  212. }  
  213.   
  214. int main()  
  215. {  
  216.       //FI;  
  217.       FAST;  
  218.       cin >> s;  
  219.       int len = sz(s);  
  220.       if (len & 1)  
  221.       {  
  222.             cout << "-1\n";  
  223.             exit(0);  
  224.       }  
  225.       s += s;  
  226.       int n = sz(s);  
  227.       build(1, 1, n);  
  228.       bool exist = 0;  
  229.       for (int i = 1; i <= len; i++)  
  230.       {  
  231.             segmentTree q = query(1, 1, n, i, i+len-1);  
  232.             if (q.balanced()) exist = 1, balancedPos[i-1] = 1; 
  233.       }  
  234.       if (exist == 0)  
  235.       {  
  236.             cout << "-1\n";  
  237.             exit(0);  
  238.       }  
  239.       suffixArray(s);  
  240.       for (int i = 0; i < n; i++)  
  241.       {  
  242.             if (balancedPos[ B[i] ])  
  243.             {  
  244.                   cout << s.substr(B[i], len) << endl;  
  245.                   break;  
  246.             }  
  247.       }  
  248. }  

No comments:

Post a Comment

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