题目描述
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 <= 200000。
思路:离线加树状数组维护前缀和。
找出每个数第一次出现的位置,记为ihead[x],并且记录每个数的相同数的下一个位置,记为inext[i]。
for(i = n; i >= 0; i--) inext[i] = ihead[a[i]], ihead[a[i]] = i;
在数组a[i]中找到最大的那个数,从1~n将所有的ihead[i]加入到前缀和中去。
接下来将所有的询问排序
bool cmp1(const Q &a, const Q &b) { return a.l == b.l ? a.r < b.r : a.l < b.l; }
然后从小到大遍历,inext[i]有值就代表这个数之前已经出现过,所以我们要用树状数组去更新inext[i]这个位置。
for(i = 1; i <= m; ++i)
{
while(l < q[i].l)
{
if(inext[l])
add(inext[l], 1);
l++;
}
q[i].ans = sum(q[i].r) - sum(q[i].l-1);
}
最后再按照询问的id排一下序,输出答案就好了。我读入用了快读,可以当作板子记下来。
这题离线+树状数组和莫队都可以写,发现一个神奇的地方,离线+树状数组可以在洛谷AC但是在HYSBZ超时,莫队可以在HYSBZ AC但是在洛谷超时。233333...
#include <bits/stdc++.h>
using namespace std;
#define lowbit(x) (x&-x)
#define max(a,b) ((a)>(b)?(a):(b))
const int N = 5000005, M = 200005;
int n, m, bit[N], a[N], ihead[2*N], inext[N];
struct Q
{
int l, r, id, ans;
}q[M];
bool cmp1(const Q &a, const Q &b) { return a.l == b.l ? a.r < b.r : a.l < b.l; }
bool cmp2(const Q &a, const Q &b) { return a.id < b.id; }
inline int read()
{
char ch = getchar(); int x = 0, f = 1;
while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
while('0' <= ch && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
return x * f;
}
inline void add(int x, const int &y) { while(x <= n) bit[x] += y, x += lowbit(x);}
inline int sum(int x) { int ret = 0; while(x > 0) ret += bit[x], x -= lowbit(x); return ret;}
int main()
{
scanf("%d",&n);
int i, maxi = 0, l = 1;
for(i = 1; i <= n; i++) scanf("%d",&a[i]), maxi = max(maxi, a[i]);
for(i = n; i >= 0; i--) inext[i] = ihead[a[i]], ihead[a[i]] = i;
for(i = 1; i <= maxi; i++) if(ihead[i]) add(ihead[i], 1);
scanf("%d",&m);
for(i = 1; i <= m; i++) scanf("%d%d",&q[i].l, &q[i].r), q[i].id = i;
sort(q+1, q+1+m, cmp1);
for(i = 1; i <= m; ++i)
{
while(l < q[i].l)
{
if(inext[l])
add(inext[l], 1);
l++;
}
q[i].ans = sum(q[i].r) - sum(q[i].l-1);
}
sort(q+1, q+1+m, cmp2);
for(i = 1; i <= m; ++i) printf("%d\n", q[i].ans);
return 0;
}
莫队的写法:
#include<bits/stdc++.h>
using namespace std;
int sum,tot[1000010],ans[2000010],a[500010],pos[500010];
struct node
{
int l,r,id;
}q[200010];
inline int read()
{
char ch = getchar(); int x = 0, f = 1;
while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
while('0' <= ch && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
return x * f;
}
bool cmp(node aa,node bb)
{
if(pos[aa.l]==pos[bb.l]) return aa.r<bb.r;
return aa.l<bb.l;
}
void update(int x, int d)
{
sum -= (tot[x] == 0 ? 0 : 1);
tot[x] += d;
sum += (tot[x] == 0 ? 0 : 1);
}
int main()
{
int n,m,i,block,l,r;
n = read();
for(i = 1; i <= n; i++) a[i] = read();
block = (int)sqrt(n);
m = read();
for(i = 1; i <= m; i++) q[i].l = read(), q[i].r = read(), q[i].id = i;
for(i = 1; i <= n; i++) pos[i] = (int)(i-1)/block+1;
sort(q+1, q+m+1, cmp);
l = 1; r = 0;
memset(tot,0,sizeof(tot));
for(i = 1; i <= m; i++)
{
while(l < q[i].l) update(a[l++], -1);
while(l > q[i].l) update(a[--l], 1);
while(r < q[i].r) update(a[++r], 1);
while(r > q[i].r) update(a[r--], -1);
ans[q[i].id] = sum;
}
for(i = 1; i <= m; i++) printf("%d\n",ans[i]);
return 0;
}