題意

詢問區間有多少不同的數。

題解

和Luogu 1903一樣，這道題也是用pre數組來求區間不同數的個數，這里pre[i]表示a[i]上一次出現的位置 +１，詢問相當于查詢區間內有多少pre小于等于左端點。

#include <cstdio> #include <cstring> #include <algorithm> #include <set> using namespace std; typedef long long ll; #define space putchar(' ') #define enter putchar('\n') template <class T> void read(T &x){char c;bool op = 0;while(c = getchar(), c < '0' || c > '9')if(c == '-') op = 1;x = c - '0';while(c = getchar(), c >= '0' && c <= '9')x = x * 10 + c - '0';if(op) x = -x; } template <class T> void write(T x){if(x < 0) x = -x, putchar('-');if(x >= 10) write(x / 10);putchar('0' + x % 10); } const int N = 50005; int n, m, l, r, ans, a[N], last[1000005], pre[N]; int idx, ls[N*50], rs[N*50], data[N*50], root[N]; void build(int &k, int l, int r){k = ++idx;if(l == r) return;int mid = (l + r) >> 1;build(ls[k], l, mid);build(rs[k], mid + 1, r); } void change(int old, int &k, int l, int r, int p, int x){k = ++idx;ls[k] = ls[old], rs[k] = rs[old], data[k] = data[old];if(l == r) return (void)(data[k] += x);int mid = (l + r) >> 1;if(p <= mid) change(ls[old], ls[k], l, mid, p, x);else change(rs[old], rs[k], mid + 1, r, p, x);data[k] = data[ls[k]] + data[rs[k]]; } int query(int old, int k, int l, int r, int ql, int qr){if(ql <= l && qr >= r) return data[k] - data[old];int mid = (l + r) >> 1, ret = 0;if(ql <= mid) ret += query(ls[old], ls[k], l, mid, ql, qr);if(qr > mid) ret += query(rs[old], rs[k], mid + 1, r, ql, qr);return ret; } int main(){read(n);build(root[0], 1, n);for(int i = 1; i <= n; i++){read(a[i]);pre[i] = last[a[i]] + 1;last[a[i]] = i;change(root[i - 1], root[i], 1, n, pre[i], 1);}read(m);while(m--){read(l), read(r);write(query(root[l - 1], root[r], 1, n, 1, l)), enter;}return 0; }

轉載于:https://www.cnblogs.com/RabbitHu/p/BZOJ1878.html

總結

以上是生活随笔為你收集整理的BZOJ 1878 HH的项链 | 主席树的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。