BZOJ 2743 [HEOI2012]采花

Description

萧芸斓是Z国的公主，平时的一大爱好是采花。

今天天气晴朗，阳光明媚，公主清晨便去了皇宫中新建的花园采花。花园足够大，容纳了n朵花，花有c种颜色（用整数1-c表示），且花是排成一排的，以便于公主采花。公主每次采花后会统计采到的花的颜色数，颜色数越多她会越高兴！同时，她有一癖好，她不允许最后自己采到的花中，某一颜色的花只有一朵。为此，公主每采一朵花，要么此前已采到此颜色的花，要么有相当正确的直觉告诉她，她必能再次采到此颜色的花。由于时间关系，公主只能走过花园连续的一段进行采花，便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了m个行程，然后一一向你询问公主能采到多少朵花（她知道你是编程高手，定能快速给出答案！），最后会选择令公主最高兴的行程（为了拿到更多奖金！）。

【题目分析】

树状数组

【代码】

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int next[1000001];
int last[1000001];
int first[1000001];
int a[1000001];
int t[1000001];
int l[1000001];
int r[1000001];
int rank[1000001];
int ans[1000001];
int n,c,m;
inline void add(int k,int f)
{
//  cout<<"add in "<<k<<" "<<f<<endl;
for (;k<=n;k+=k&(-k)) t[k]+=f;
}
inline int gs(int k)
{
int ret=0;
for (;k;k-=k&(-k)) ret+=t[k];
return ret;
}
inline bool cmp(int a,int b)
{return l[a]<l[b];}
int main()
{
scanf("%d%d%d",&n,&c,&m);
for (int i=1;i<=n;++i) scanf("%d",&a[i]);
for (int i=n;i;--i)
{
//      scanf("%d",&a[i]);
//      if (!last[a[i]]) first[a[i]]=i;
next[i]=last[a[i]];
last[a[i]]=i;
}
//  for (int i=1;i<=n;++i) printf("%d ",next[i]);printf("\n");
for (int i=1;i<=c;++i)
if (next[last[i]]) add(next[last[i]],1);
for (int i=1;i<=m;++i)
{
scanf("%d%d",&l[i],&r[i]);
rank[i]=i;
}
sort(rank+1,rank+m+1,cmp);
int ll=1;
for (int i=1;i<=m;++i)
{
int nl=l[rank[i]],nr=r[rank[i]];
//      cout<<nl<<" "<<nr<<endl;
while (ll<nl)
{
if (next[ll]) add(next[ll],-1);
if (next[next[ll]]) add(next[next[ll]],1);
ll++;
}
//      cout<<"ans "<<gs(nr)-gs(nl-1)<<endl;
ans[rank[i]]=gs(nr)-gs(nl-1);
}
for (int i=1;i<=m;++i) printf("%d\n",ans[i]);
}