bzoj 2743: [HEOI2012]采花(树状数组)附加HH项链

首先先做HH项链
题目描述
HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。

有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答…… 因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。

输入格式
一行一个正整数 n,表示项链长度。
第二行 n 个正整数 ai,表示项链中第 i 个贝壳的种类。

第三行一个整数 m,表示询问的个数。
接下来 m 行,每行两个整数 l,r,表示询问的区间。

输出格式
输出 m 行,每行一个整数,依次表示询问对应的答案。

输入输出样例
输入 #1 复制
6
1 2 3 4 3 5
3
1 2
3 5
2 6
输出 #1 复制
2
2
4
我们思考一下,如果我们对所有的右端点区间进行排序,然后按顺序来计算。然后每一次询问维持一个线段树,这样做的目的就是可以按照顺序维护线段树。**把线段数组当做普通的数组来看待其实非常好处理!**我们可以用book[i]表示i这朵花最后一次出现的位置,为什么这么做?因为我们从位置pos1到n遍历,对于出现过的花,其实只有最后一次出现位置的花有意义,就好比数据12412,对于右区间为4的询问来说,此时对于花1来说,我们只需要把1处的花的标记消掉(怎么消掉,只需在这个位置加上-1就能消掉了),然后在4这个位置标记一下这里出现了一朵花就可以了。
在这里插入图片描述

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n;
int top[1000010];
int num[1000010];
int book[1000010];
int answer[1000010];
struct node
{
    int l,r,pos;
}data[1000010];
bool cmp(struct node a,struct node b)
{
    return a.r<b.r;
}
int lowbit(int x)
{
    return x&(-x);
}
void updata(int i,int num)
{
    for(int j=i;j<=n;j=j+lowbit(j))
    {
        top[j]+=num;
    }
    return;
}
int getsum(int i)
{
    int ans=0;
    for(int j=i;j>0;j=j-lowbit(j))
    {
        ans+=top[j];
    }
    return ans;
}
int main ()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
    }
    int m;
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&data[i].l,&data[i].r);
        data[i].pos=i;
    }
    sort(data+1,data+1+m,cmp);
    int j=1;
    for(int i=1;i<=n;i++)
    {
        if(!book[num[i]])//没有出现过
        {
            book[num[i]]=i;
            updata(i,1);
        }
        else
        {
            updata(book[num[i]],-1);
            updata(i,1);
            book[num[i]]=i;
        }
        while(data[j].r==i)
        {
            answer[data[j].pos]=getsum(data[j].r)-getsum(data[j].l-1);
            j++;
        }
    }
    for(int i=1;i<=m;i++)
    {
        printf("%d\n",answer[i]);
    }
    return 0;
}

萧芸斓是Z国的公主,平时的一大爱好是采花。今天天气晴朗,阳光明媚,公主清晨便去了皇宫中新建的花园采花
。花园足够大,容纳了n朵花,花有c种颜色(用整数1-c表示),且花是排成一排的,以便于公主采花。公主每次
采花后会统计采到的花的颜色数,颜色数越多她会越高兴!同时,她有一癖好,她不允许最后自己采到的花中,某
一颜色的花只有一朵。为此,公主每采一朵花,要么此前已采到此颜色的花,要么有相当正确的直觉告诉她,她必
能再次采到此颜色的花。由于时间关系,公主只能走过花园连续的一段进行采花,便让女仆福涵洁安排行程。福涵
洁综合各种因素拟定了m个行程,然后一一向你询问公主能采到多少朵花(她知道你是编程高手,定能快速给出答
案!),最后会选择令公主最高兴的行程(为了拿到更多奖金!)。
Input
第一行四个空格隔开的整数n、c以及m。
接下来一行n个空格隔开的整数,每个数在[1, c]间,第i个数表示第i朵花的颜色。
接下来m行每行两个空格隔开的整数l和r(l ≤ r),表示女仆安排的行程为公主经过第l到第r朵花进行采花。
Output
共m行,每行一个整数,第i个数表示公主在女仆的第i个行程中能采到的花的颜色数。
Sample Input
5 3 5
1 2 2 3 1
1 5
1 2
2 2
2 3
3 5
Sample Output
2
0
0
1
0
【样例说明】
询问[1, 5]:公主采颜色为1和2的花,由于颜色3的花只有一朵,公主不采;询问[1, 2]:颜色1和颜色2的花均只有一朵,公主不采;
询问[2, 2]:颜色2的花只有一朵,公主不采;
询问[2, 3]:由于颜色2的花有两朵,公主采颜色2的花;
询问[3, 5]:颜色1、2、3的花各一朵,公主不采。
Hint
【数据范围】

对于100%的数据,1 ≤ n ≤ 10^6,c ≤ n,m ≤10^6。
基本思路都是一样的,这次在book数组基础上加一个pre数组就可以了。book[i]表示i这种花最后出现的位置,pre[i]表示i这种花倒数第二次出现的位置,这样就可以处理了。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n;
int top[1000010];
int num[1000010];
int book[1000010];//记录数字num[i]的前一个位置
int pre[1000010];//记录num[i]这个数的前一个位置
int answer[1000010];
struct node
{
    int l,r,pos;
}data[1000010];
bool cmp(struct node a,struct node b)
{
    return a.r<b.r;
}
int lowbit(int x)
{
    return x&(-x);
}
void updata(int i,int num)
{
    for(int j=i;j<=n;j=j+lowbit(j))
    {
        top[j]+=num;
    }
    return;
}
int getsum(int i)
{
    int ans=0;
    for(int j=i;j>0;j=j-lowbit(j))
    {
        ans+=top[j];
    }
    return ans;
}
int main ()
{
    int temp,m;
    scanf("%d%d%d",&n,&temp,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&data[i].l,&data[i].r);
        data[i].pos=i;
    }
    sort(data+1,data+1+m,cmp);
    int j=1;
    for(int i=1;i<=n;i++)
    {
        if(pre[num[i]])//出现过
        {
            updata(pre[num[i]],-1);
            updata(book[num[i]],1);
            pre[num[i]]=book[num[i]];
            book[num[i]]=i;
        }
        else if(book[num[i]])
        {
            updata(book[num[i]],1);
            pre[num[i]]=book[num[i]];
            book[num[i]]=i;
        }
        else book[num[i]]=i;
        while(data[j].r==i)
        {
            answer[data[j].pos]=getsum(data[j].r)-getsum(data[j].l-1);
            j++;
        }
    }
    for(int i=1;i<=m;i++)
    {
        printf("%d\n",answer[i]);
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章