Skip to main content

Sereja and Brackets

                                Sereja and Brackets

                                  segment tree

#include <iostream>
#include <cmath>
#include <cstring>

using namespace std;

string s;

struct node
{

    long int len; //max length of matched brackets
    int tn, tp; // tp = total number of '(' brackets

                //tn = total number of ')' brackets
};

node func(node a, node b)
{
    node temp;

    int len = min(a.tp, b.tn);

    temp.len = a.len + b.len + 2 * len;
    temp.tp = a.tp + b.tp - len;
    temp.tn = a.tn + b.tn - len;

    return temp;
}

void buildTree(node seg[], int l, int r, int index)
{
    if(l == r)
    {
        seg[index].len = seg[index].tp = seg[index].tn = 0;
        if(s[l] == ')')
        {
            seg[index].tn = 1;
        }
        else
        {
            seg[index].tp = 1;
        }
     }
     else
     {
         int mid = (l + r) / 2;
         buildTree(seg, l, mid, 2 * index + 1);
         buildTree(seg, mid + 1, r, 2 * index + 2);

         seg[index] = func(seg[2 * index + 1], seg[2 * index + 2]);
     }
}

node query(node seg[], int l, int r, int qs, int qe, int index)
{
    if((qs > r || qe < l))
    {
        node temp;
        temp.len = temp.tp = temp.tn = 0;

        return temp;
    }

    else if(qs <= l && qe >= r)
        return seg[index];

    else
    {
        int mid = (l + r ) / 2;

        node p1 = query(seg, l, mid, qs, qe, 2 * index + 1);
        node p2 = query(seg, mid + 1, r, qs, qe, 2 * index + 2);

        node temp = func(p1, p2);
        return temp;
    }

}

int main()
{
    int m, l, r;

    cin >> s;

    long int size = 4 * s.size();

    node seg[size];

    buildTree(seg, 0, s.size() - 1, 0);

    cin >> m;

    while(m--)
    {
        cin >> l >> r;

        node temp = query(seg, 0, s.size() - 1, l - 1, r - 1, 0);
        cout << temp.len << endl;
    }
    return 0;
}

Comments

Post a Comment

Popular posts from this blog

MAIN72

MAIN72 - Subset sum #include <iostream> #include <cstring> using namespace std; bool dp[100001][1001]; int arr[1001]; int main() {     int t, n;     long long int sum;     cin >> t;     while(t--)     {         cin >> n;         memset(dp, false, sizeof(dp));         sum = 0;         for(int i = 0; i < n; i++)         {             cin >> arr[i];             sum += arr[i];         }         for(int i = 0; i < n; i++)             dp[0][i] = true; // 0 sum         for(int i = 1; i < n; i++)             dp[i][0] = false; // sum is i but 0 element         for(long int i = 1; i <= sum; i++)         {             for(int j = 1; j <= n; j ++)             {                 dp[i][j] = dp[i][j - 1];                 if(i >= arr[j - 1])                     dp[i][j] = dp[i][j] || dp[i - arr[j - 1]][j - 1];             }         }         long int result = 0;         for(int i = 1; i <= sum; i++)  
Post a Comment
Read more

War of XORs- XORIER

Problem Link #include <iostream> using namespace std; int main() { int t, n, odd, even; cin >> t; while(t--) { cin >> n; int i,arr[n],freq[1100001]={0}; long res = 0; odd = even = 0; for(int i = 0; i < n; i++) { cin >> arr[i]; freq[arr[i]]++; } for(int i = 0; i < n; i++) { if(arr[i] & 1) { odd++; } else { even++; } } for(int i = 0; i < n; i++) { if(arr[i] % 2) { res += odd; } else { res += even; } res -= freq[arr[i] ^ 2]; res -= freq[arr[i]]; } cout << res / 2 << endl; } }
Post a Comment
Read more

Chef and Easy Problem- XXOR

Problem Link #include <iostream> #include <vector> using namespace std; int main() { int n, q, l, r; cin >> n >> q; long arr[n]; vector<long> b[n]; for(int i = 0; i < n; i++) { cin >> arr[i]; long temp = arr[i]; int x = 0; for(int j = 0; j < 32; j++) { b[i].push_back(0); } while(temp) { b[i][x] = temp % 2; temp /= 2; x++; } } for(int i = 1; i < n; i++) { for(int j = 0; j < 32; j++) { b[i][j] += b[i - 1][j]; } } while(q--) { cin >> l >> r; l --; r--; long res = 0; for(int i = 0; i < 31; i++) { int setBits, unsetBits; if(l > 0) { setBits = b[r][i] - b[l - 1][i]; unsetBits = r - l + 1 - setBits; } else { setBits = b[r][i]; unsetBits = r - l + 1 - setBits; } if(setBits < unsetBits) { res |= (1 << i); } } cout << res << endl;
Post a Comment
Read more