[hdu5654 xiaoxin and his watermelon candy]区间内不同数的个数

来源:互联网 发布:stm32f103tbu6编程 编辑:程序博客网 时间:2024/05/19 22:48

原题
思路:
首先求得所有合法的三元组(ai,ai+1,ai+2),这样的三元组数量是O(n)的,然后离散化成1-n的序列,不合法的用0表示,得到新的序列bi。令pibi这个数上一次出现的位置,则求a的[L,R]之间有多少个不同的合法三元组等价于求p的[L,R-2]之间有多少个值小于L。考虑到有不合法的三元组,需要对它们的p做一些特殊处理,即如果bi等于0,则令pi=i。而[L,R-2]可以写成[1,R-2]-[1,L-1],主席树搞之。
回过头来看,这个题就是求“区间内不同的数的个数“,经典的离线做法也是可以的。
坑点:RL<2时不特判可能出现负数;节点数目开到2e520居然存不下,开到2e521就好了-_-|||

#include <bits/stdc++.h>using namespace std;const int N = 2e5 + 7;typedef long long ll;map<ll, int> mp;int n, m, l, r, T, cnt, a[N], b[N], root[N], last[N], p[N];struct SegTree {    struct Node {        int lp, rp, sum;    };    int c;    Node tree[N * 21];    void clear() {        c = 0;        memset(tree, 0, sizeof(tree));    }    void build(int l, int r, int &rt) {        rt = c++;        if (l == r) return;        int m = l + r >> 1;        build(l, m, tree[rt].lp);        build(m + 1, r, tree[rt].rp);    }    void update(int x, int v, int last, int l, int r, int &rt) {        rt = c++;        tree[rt] = tree[last];        tree[rt].sum += v;        if (l == r) return;        int m = l + r >> 1;        if (x <= m) update(x, v, tree[last].lp, l, m, tree[rt].lp);        else update(x, v, tree[last].rp, m + 1, r, tree[rt].rp);    }    int query(int R, int l, int r, int rt) {        if (R >= r) return tree[rt].sum;        int m = (l + r) >> 1;        if (R <= m) return query(R, l, m, tree[rt].lp);        else return query(R, l, m, tree[rt].lp) + query(R, m + 1, r, tree[rt].rp);    }};SegTree st;int main() {    cin >> T;    while (T--) {        cin >> n;        for (int i = 1; i <= n; i++) {            scanf("%d", a + i);            p[i] = i;        }        cnt = 0;        memset(b, 0, sizeof(b));        memset(last, 0, sizeof(last));        mp.clear();        for (int i = 1; i <= n - 2; i++) {            if (a[i] > a[i + 1] || a[i + 1] > a[i + 2]) continue;            ll x = ((ll)a[i] * 1000000ll + a[i + 1]) * 1000000ll + a[i + 2];            if (!mp[x]) b[i] = mp[x] = ++cnt;            else b[i] = mp[x];            p[i] = last[b[i]];            last[b[i]] = i;        }        st.clear();        st.build(0, n, root[0]);        for (int i = 1; i <= n; i++) st.update(p[i], 1, root[i - 1], 0, n, root[i]);        cin >> m;        while (m--) {            scanf("%d%d", &l, &r);            if (r - l < 2) puts("0");            else printf("%d\n", st.query(l - 1, 0, n, root[r - 2]) - l + 1);        }    }    return 0;}
0 0