树状数组-luogu4113-采花

题目大意

花园里有c种花共计n朵，公主要在一个区间[l,r]内采花，仅当该颜色的花必须至少可以采到两朵，公主才会采它，问公主可以才多少种花。此外，题目会给出m个这样的闭区间用来询问。

0<=c<n,m<=2e6

题目分析

1. 显然，这个问题和之前的luogu1972-HH的项链是很类似的，本质是在求区间内不同元素的数目，只是仅当该元素至少出现两次时才计入统计。所以处理该问题的基本方法与项链问题也应该基本一致，区别主要在于对二次出现统计的处理。

2. 在项链中，我们给上次出现的位置置零，给这次出现为位置置一，也就是只统计最后一次出现。类似地，为实现两次出现才统计的效果，我们可以在这次出现时，给上次出现置一，给上上次出现置零，也就是只统计倒数第二次出现。

3. 在实现只统计倒数第二次出现的过程中，由于需要记录上次、上上次的情况，所以需要设置

int last[maxn][2]//last[i][0]、last[i][1]分别为i的上上次、上次出现的位置

AC代码

#include <string>
#include <algorithm>
#include <cstdio>

struct Query
{
    int l,r,idx;
    friend bool operator<(const Query & a,const Query & b){return a.r<b.r;}
};


const int maxf=2e6+10;
const int maxn=maxf;

int last[maxf][2];
int c[maxn];
int a[maxn];
int ans[maxn];
Query queries[maxn];
int pos[maxn];
void add(int x,int t)
{
    if(!x) return;
    while(x<=maxn)
    {
        c[x]+=t;
        x+=x&(-x);
    }
}

int sum(int x)
{
    int ans=0;
    while(x>=1)
    {
        ans+=c[x];
        x-=x&(-x);
    }
    return ans;
}

using namespace std;

int main()
{
    int n,c,m;
    scanf("%d%d%d",&n,&c,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)
        scanf("%d%d",&queries[i].l,&queries[i].r),queries[i].idx=i;
    sort(queries+1,queries+m+1);
    int k=1;
    for(int i=1;i<=m;i++)
    {
        int l=queries[i].l,r=queries[i].r;
        for(;k<=r;k++)
        {
            if(last[a[k]][1])
                add(last[a[k]][1],1);
            if(last[a[k]][0])
                add(last[a[k]][0],-1);
            last[a[k]][0]=last[a[k]][1],last[a[k]][1]=k;
        }
        ans[queries[i].idx]=sum(r)-sum(l-1);
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}