P1972 [SDOI2009--HYSBZ - 1878--HH的项链(离线+树状数组or莫队)

题目描述

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...

题目链接:洛谷点我HYSBZ点我

#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;
}

猜你喜欢

转载自blog.csdn.net/sugarbliss/article/details/81146936