[SDOI2009]HH的项链 树状数组 BZOJ 1878
题目背景
无
题目描述
HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答……因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。
输入输出格式
输入格式:第一行:一个整数N,表示项链的长度。
第二行:N 个整数,表示依次表示项链中贝壳的编号(编号为0 到1000000 之间的整数)。
第三行:一个整数M,表示HH 询问的个数。
接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。
输出格式:M 行,每行一个整数,依次表示询问对应的答案。
输入输出样例
说明
数据范围:
对于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; }
EPFL - Fighting