Cod sursa(job #2620640)

Utilizator CosminMorarMorar Cosmin Andrei CosminMorar Data 29 mai 2020 12:59:56
Problema Subsir crescator maximal Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.41 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("xormax.in");
ofstream fout("xormax.out");
int n, nr_max_biti, v[100100], xorpart[100100], sol_value = -1, sol_left, sol_right;

struct trie {
    trie *bit[2];
    int last_poz;

    trie() {
        bit[0] = bit[1] = nullptr;
    }
};

int calc_nr_max_biti(int x) {
    int nr_biti = 0;
    while (x) {
        nr_biti++;
        x >>= 1;
    }
    return max(1, nr_biti);
}

void add(trie *nod, int x, int nr_bit, int p) {
    if (nr_bit == 0) {
        nod->last_poz = p;
        return;
    }

    for (int i = 0; i <= 1; i++)
        if (((x & (1 << (nr_bit - 1))) > 0) == (bool)i) {
            /// bitul de pe pozitia "nr_bit - 1" este egal cu i (0 sau 1)
            if (nod->bit[i] == nullptr)
                nod->bit[i] = new trie();
            add(nod->bit[i], x, nr_bit - 1, p);
        }
}

int find_poz(trie *nod, int nr_bit, int want) {
    if (nr_bit == 0)
        return nod->last_poz;

    bool want_bit = (bool)((want & (1 << (nr_bit - 1))) > 0);

    if (nod->bit[want_bit] == nullptr) {
        if (nod->bit[!want_bit] == nullptr)
            return -1;
        return find_poz(nod->bit[!want_bit], nr_bit - 1, want);
    }
    return find_poz(nod->bit[want_bit], nr_bit - 1, want);
}

int main() {
    fin >> n;
    for (int i = 1; i <= n; i++) {
        fin >> v[i];
        xorpart[i] = xorpart[i - 1] ^ v[i];
        nr_max_biti = max(nr_max_biti, calc_nr_max_biti(v[i]));
    }

    trie *root = new trie();
    add(root, 0, nr_max_biti, 0);

    for (int i = 1; i <= n; i++) {
        /// incercam sa adaugam secventa [i, i]
        if (v[i] > sol_value || v[i] == sol_value && i == sol_right && i > sol_left) {
            sol_value = v[i];
            sol_left = sol_right = i;
        }

        /// incercam sa adaugam secventa [poz + 1, i]
        int poz = find_poz(root, nr_max_biti, (xorpart[i] ^ ((1 << nr_max_biti) - 1)));

        if (poz != -1) {
            int xor_secv = xorpart[i] ^ xorpart[poz];
            if (xor_secv > sol_value || xor_secv == sol_value && i == sol_right && poz + 1 > sol_left) {
                sol_value = xor_secv;
                sol_left = poz + 1;
                sol_right = i;
            }
        }

        add(root, xorpart[i], nr_max_biti, i);
    }

    fout << sol_value << ' ' << sol_left << ' ' << sol_right << '\n';
    return 0;
}