洛谷P1972 [SDOI2009]HH的項鍊(莫隊)44分做法
阿新 • • 發佈:2021-08-01
題目描述
HH 有一串由各種漂亮的貝殼組成的項鍊。HH 相信不同的貝殼會帶來好運,所以每次散步完後,他都會隨意取出一段貝殼,思考它們所表達的含義。HH 不斷地收集新的貝殼,因此,他的項鍊變得越來越長。
有一天,他突然提出了一個問題:某一段貝殼中,包含了多少種不同的貝殼?這個問題很難回答…… 因為項鍊實在是太長了。於是,他只好求助睿智的你,來解決這個問題。
輸入格式
一行一個正整數 nn,表示項鍊長度。
第二行 nn 個正整數 aia**i,表示項鍊中第 ii 個貝殼的種類。
第三行一個整數 mm,表示 HH 詢問的個數。
接下來 mm 行,每行兩個整數 l,rl,r,表示詢問的區間。
輸出格式
輸出 mm 行,每行一個整數,依次表示詢問對應的答案。
輸入輸出樣例
輸入 #1複製
6
1 2 3 4 3 5
3
1 2
3 5
2 6
輸出 #1複製
2
2
4
這個題可以從 [l, r]的答案能夠 O(1)擴充套件到 [l − 1, r], [l + 1, r], [l, r + 1], [l, r − 1]的答案,因此可以用莫隊寫。可惜的是資料卡掉了莫隊,只能用樹狀陣列、線段樹等。
#include <bits/stdc++.h> using namespace std; int n, a[10000005], m, ans[10000005], block; inline int read() { char x; while((x = getchar()) > '9' || x < '0') ; int u = x - '0'; while((x = getchar()) <= '9' && x >= '0') u = (u << 3) + (u << 1) + x - '0'; return u; } int buf[105]; inline void write(int i) { int p = 0; if(i == 0) p++; else while(i) { buf[p++] = i % 10; i /= 10; } for(int j = p-1; j >= 0; j--) putchar('0' + buf[j]); } struct node { int l, r, id; } q[10000005]; bool cmp(node a, node b) { return (a.l / block) ^ (b.l / block)? (a.l / block) < (b.l / block) : (a.l / block) & 1? a.r < b.r: a.r > b.r; } int Ans = 0, cnt[10000001]; void add(int x) { if(!cnt[a[x]]) Ans++; cnt[a[x]]++; } void del(int x) { cnt[a[x]]--; if(!cnt[a[x]]) Ans--; } int main() { n = read(); for (int i = 1; i <= n; i++) { a[i] = read(); } m = read(); block = n / sqrt(m * 2 / 3); for (int i = 1; i <= m; i++) { q[i].l = read(); q[i].r = read(); q[i].id = i; } sort(q + 1, q + m + 1, cmp); int l = 0, r = 0; for(int i = 1; i <= m; i++) { int ql = q[i].l, qr = q[i].r; while(l < ql) del(l++); while(l > ql) add(--l); while(r < qr) add(++r); while(r > qr) del(r--); ans[q[i].id] = Ans; } for(int i = 1; i <= m; i++) { write(ans[i]); putchar('\n'); } return 0; }