P4113 [HEOI2012]采花 【树状数组 AC】【莫队/主席树 TLE】

传送门

题意简述：给定一个长度为 n 的序列，有 m 次询问，每次询问一段区间，求区间中有多少个数出现次数超过 1 次

这道题真的有点恶心。。。在洛谷上属于莫队，但是莫队会T，数据居然有2e6。莫队的想法很简单就不多提了。

主席树：这道题和 HH的项链 有异曲同工之妙，后者这道题：将每个数右靠，保证区间中只存在一个相同的数。
前者：我们用i棵线段树，维护[1,i ]的信息。如果区间中相同的数的数量>=2，我们记录只每个数倒数第二次出现的位置。
询问[ l , r ]区间，在第r棵线段树上查找下标>=l 的数量。
对于这道题来讲：主席树占用的空间很大，时间效率也不高，和莫队一样过不了最后两组数据。

树状数组：如果用树状数组求解就需要将询问离线，按照右端点升序排列，思路和主席树一样(因为树状数组不能持久化，所以需要离线)，这样就能轻松的把这道题过掉。时间和空间真的是用的少啊。。。。。

树状数组代码（200分）：

#include<bits/stdc++.h>
#define lowbit(x)   (x&(-x))
using namespace std;
const int maxn = 2e6 + 5;

int n, c, q, num[maxn];
int pos[maxn], pre[maxn], ans[maxn];
///pos[x]  记录x出现最后的位置
///pre[i]  和num[i]相同的数的前一个位置
int low[maxn];
struct node {
    int l, id, p;
} load[maxn];
int head[maxn], sign;
void add_edge(int l, int r, int id) {
    load[++sign] = node{l, id, head[r]};
    head[r] = sign;
}
void add(int i, int x) {
    while(i <= n) {
        low[i] += x;
        i += lowbit(i);
    }
}
int sum(int i) {
    int ans = 0;
    while(i > 0) {
        ans += low[i];
        i -= lowbit(i);
    }
    return ans;
}
int main() {
    int l, r, x;
    memset(head, -1, sizeof(head));
    scanf("%d %d %d", &n, &c, &q);
    for(int i = 1; i <= n; i++)
        scanf("%d", &num[i]);
    for(int i = 1; i <= q; i++) {
        scanf("%d %d", &l, &r);
        add_edge(l, r, i);
    }
    for(int i = 1; i <= n; i++) {
        x = num[i];
        if(pos[x] != 0)     ///将倒数第二出现的位置标记
            add(pos[x], 1);
        if(pre[pos[x]] != 0)    ///将上一次标记给去掉
            add(pre[pos[x]], -1);
        for(int j = head[i]; ~j; j = load[j].p)
            ans[load[j].id] = sum(i) - sum(load[j].l - 1);
        pre[i] = pos[x];    ///记录每个颜色的信息
        pos[x] = i;
    }
    for(int i = 1; i <= q; i++)
        printf("%d\n", ans[i]);
    return 0;
}

主席树代码（133分）：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e6 + 5;

int pos[maxn], pre[maxn];
int n, c, q;
struct node {
    int ls, rs, cnt;
} p[maxn * 80];
int root[maxn], times;

void insert(int &now, int old, int l, int r, int pos, int x) {
    now = ++times;
    p[now] = p[old], p[now].cnt += x;
    if(l == r)
        return ;
    int mid = (l + r) >> 1;
    if(pos <= mid)
        insert(p[now].ls, p[old].ls, l, mid, pos, x);
    else
        insert(p[now].rs, p[old].rs, mid + 1, r, pos, x);
}
int query(int now, int l, int r, int x) {
    if(l >= x)
        return p[now].cnt;
    int mid = (l + r) >> 1;
    if(x <= mid)
        return p[p[now].rs].cnt + query(p[now].ls, l, mid, x);
    else
        return query(p[now].rs, mid + 1, r, x);
}
int main() {
    int x, l, r;
    scanf("%d %d %d", &n, &c, &q);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &x);
        root[i] = root[i - 1];
        if(pos[x] != 0)
            insert(root[i], root[i - 1], 1, n, pos[x], 1);
        if(pre[pos[x]] != 0)
            insert(root[i], root[i], 1, n, pre[pos[x]], -1);
        pre[i] = pos[x];
        pos[x] = i;
    }
    for(int i = 1; i <= q; i++) {
        scanf("%d %d", &l, &r);
        printf("%d\n", query(root[r], 1, n, l) );
    }
    return 0;
}