[SDOI2009]HH的項鍊 樹狀陣列
阿新 • • 發佈:2018-12-29
題目背景
無
題目描述
HH 有一串由各種漂亮的貝殼組成的項鍊。HH 相信不同的貝殼會帶來好運,所以每次散步完後,他都會隨意取出一段貝殼,思考它們所表達的含義。HH 不斷地收集新的貝殼,因此,他的項鍊變得越來越長。有一天,他突然提出了一個問題:某一段貝殼中,包含了多少種不同的貝殼?這個問題很難回答……因為項鍊實在是太長了。於是,他只好求助睿智的你,來解決這個問題。
輸入輸出格式
輸入格式:第一行:一個整數N,表示項鍊的長度。
第二行:N 個整數,表示依次表示項鍊中貝殼的編號(編號為0 到1000000 之間的整數)。
第三行:一個整數M,表示HH 詢問的個數。
接下來M 行:每行兩個整數,L 和R(1 ≤ L ≤ R ≤ N),表示詢問的區間。
M 行,每行一個整數,依次表示詢問對應的答案。
輸入輸出樣例
輸入樣例#1: 複製6 1 2 3 4 3 5 3 1 2 3 5 2 6輸出樣例#1: 複製
2 2 4
說明
資料範圍:
對於100%的資料,N <= 500000,M <= 500000。
莫隊貌似過不了了;
luogu的資料加強了;
考慮樹狀陣列解決;
首先這麼考慮:如果右端點r相同的話,那麼區間出現了相同的數字,我們顯然只關心最右邊的那個;
那麼我們按 ask.r 排序;
那麼更新樹狀陣列的時候,如果該位置出現了,我們就 add( j,-1 )意思是消去;
否則 add( j,1 );
那麼query 的時候就字首和解決即可;
#include<iostream> #include<cstdio> #include<algorithm> #include<cstdlib> #include<cstring> #include<string> #include<cmath> #include<map> #include<set> #include<vector> #include<queue> #include<bitset> #include<ctime> #include<deque> #include<stack> #include<functional> #include<sstream> //#include<cctype> //#pragma GCC optimize(2) using namespace std; #define maxn 2000005 #define inf 0x7fffffff //#define INF 1e18 #define rdint(x) scanf("%d",&x) #define rdllt(x) scanf("%lld",&x) #define rdult(x) scanf("%lu",&x) #define rdlf(x) scanf("%lf",&x) #define rdstr(x) scanf("%s",x) typedef long long ll; typedef unsigned long long ull; typedef unsigned int U; #define ms(x) memset((x),0,sizeof(x)) const long long int mod = 1e9 + 7; #define Mod 1000000000 #define sq(x) (x)*(x) #define eps 1e-3 typedef pair<int, int> pii; #define pi acos(-1.0) //const int N = 1005; #define REP(i,n) for(int i=0;i<(n);i++) typedef pair<int, int> pii; inline ll rd() { ll x = 0; char c = getchar(); bool f = false; while (!isdigit(c)) { if (c == '-') f = true; c = getchar(); } while (isdigit(c)) { x = (x << 1) + (x << 3) + (c ^ 48); c = getchar(); } return f ? -x : x; } ll gcd(ll a, ll b) { return b == 0 ? a : gcd(b, a%b); } ll sqr(ll x) { return x * x; } /*ll ans; ll exgcd(ll a, ll b, ll &x, ll &y) { if (!b) { x = 1; y = 0; return a; } ans = exgcd(b, a%b, x, y); ll t = x; x = y; y = t - a / b * y; return ans; } */ int n; int ans[maxn]; int bt[maxn]; int chk[maxn]; int num[maxn]; struct node { int l, r; int pos; }ask[maxn]; inline int lowbit(int x) { return x & -x; } void add(int x, int val) { while (x <= n) { bt[x] += val; x += lowbit(x); } } inline int query(int x) { int ans = 0; while (x) { ans += bt[x]; x -= lowbit(x); } return ans; } bool cmp(node a, node b) { return a.r < b.r; } int main() { //ios::sync_with_stdio(0); cin >> n; for (int i = 1; i <= n; i++)rdint(num[i]); int q; rdint(q); for (int i = 1; i <= q; i++) { rdint(ask[i].l); rdint(ask[i].r); ask[i].pos = i; } sort(ask + 1, ask + 1 + q, cmp); int st = 1; for (int i = 1; i <= q; i++) { for (int j = st; j <= ask[i].r; j++) { if (chk[num[j]])add(chk[num[j]], -1); add(j, 1); chk[num[j]] = j; } st = ask[i].r + 1; ans[ask[i].pos] = query(ask[i].r) - query(ask[i].l - 1); } for (int i = 1; i <= q; i++) { printf("%d\n", ans[i]); } return 0; }