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

原题
思路:
首先求得所有合法的三元组( ai,ai+1,ai+2 ),这样的三元组数量是O(n)的,然后离散化成1-n的序列,不合法的用0表示,得到新的序列 bi 。令 pi bi 这个数上一次出现的位置,则求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
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值