Skip to main content

D-QUERY

                                             D-Query

                                           segment tree + offline method

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <map>
#define sf(i) scanf("%d", &i)
#define pf(i) printf("%d\n", i)

using namespace std;

const int N = 30001;
const int Q = 200001;

struct node
{
    int l, r;
    int pos,
    int val;
};

int n, q;
node arr[N + Q];
int result[Q];
int seg[4 * N];
map<int, int> last_pos;

node makenode(int l, int r, int pos, int val)
{
    node temp;
    temp.l = l;
    temp.r = r;
    temp.pos = pos;
    temp.val = val;

    return temp;
}

void update(int l, int r, int index, int pos, int val)
{
    if(l == r)
    {
        seg[index] = val;
    }
    else
    {
        int mid = (r + l) / 2;
        if(pos <= mid)
        {
            update(l, mid, 2 * index, pos, val);
        }
        else
        {
            update(mid + 1, r, 2 * index + 1, pos, val);
        }
        seg[index] = seg[index << 1] + seg[(index << 1) + 1];
    }
}

int query(int l, int r, int index, int qs, int qe)
{
    if(qs > r || qe < l)
        return 0;
    if(qs <= l && qe >= r)
        return seg[index];

    int mid = (l + r) / 2;
    int a = query(l, mid, 2 * index, qs, qe);
    int b = query(mid + 1, r, 2 * index + 1, qs, qe);
    return a + b;
}

bool comp(node a, node b)
{
    if(a.pos == b.pos)
    {
        return a.r < b.r;
    }
    return (a.pos < b.pos);
}

int main()
{
    int l, r;
    sf(n);
    int index = 0;
    int temp;
    for(int i = 1; i <= n; i++)
    {
        sf(temp);
        arr[index++] = makenode(-1, -1, i, temp);
    }

    sf(q);
    for(int i = 1; i <= q; i++)
    {
        sf(l), sf(r);
        arr[index++] = makenode(l, r, r, i); // stores position as value in case of query
    }

    sort(arr, arr + index, comp);

    for(int i = 0; i < index; i++)
    {
        if(arr[i].l == -1 && arr[i].r == -1)
        {
            if(last_pos[arr[i].val] == 0)
            {
                update(1, n, 1, arr[i].pos, 1); //if value does not exist update the pos with 1
            }
            else
            {
                update(1, n, 1, last_pos[arr[i].val], 0); // if value already exist then update the old pos with 0
                update(1, n, 1, arr[i].pos, 1); // update the new pos with 1
            }
            last_pos[arr[i].val] = arr[i].pos; // update the last position array
        }
        else
        {
            int ans = query(1, n, 1, arr[i].l, arr[i].r);
            result[arr[i].val] = ans;
        }
    }

    for(int i = 1; i <= q; i++)
    {
        pf(result[i]);
    }

    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