bzoj2743 [HEOI2012]采花

  做法是每个询问先算出询问区间中花的种类减去区间中只有一朵花的花的种类,这两个子问题都不算难,具体看代码吧。询问可以离线处理,用树状数组维护,复杂度O(nlogn)。

  不知道是想的复杂了还是打的太low,运行时间有点久。。

  代码

  

 #include<cstdio>
#include<algorithm>
#include<vector>
#define pb push_back
#define mp make_pair
#define fi first
#define sc second
#define lb(x) (x&-x)
#define N 1000010
#define P 1000000007
using namespace std;
int n,c[N],C,m,i,a,j,b,ans[N],e[N];
int pos[N],next[N];
vector<int> vec[N];
vector<pair<int,int> > q[N],vec2[N];
void cc(int x,int w)
{
while (x<=n)
{
c[x]+=w;
x+=lb(x);
}
}
int sum(int x)
{
int ans=;
while (x)
{
ans+=c[x];
x-=lb(x);
}
return ans;
}
int main()
{
scanf("%d%d%d",&n,&C,&m);
for (i=;i<=n;i++)
{
scanf("%d",&a);
e[i]=a;
vec[pos[a]].pb(i);
next[pos[a]]=i;
pos[a]=i;
}
for (i=;i<=C;i++)
pos[i]=;
for (i=;i<=n;i++)
{
vec2[pos[e[i]]].pb(mp(i,next[i]));
pos[e[i]]=i;
} for (i=;i<=m;i++)
{
scanf("%d%d",&a,&b);
q[a].pb(mp(b,i));
}
for (i=;i<=n;i++)
{
if (i)
for (j=;j<q[i].size();j++)
ans[q[i][j].sc]=sum(q[i][j].fi)-sum(i-);
for (j=;j<vec[i].size();j++) cc(vec[i][j],);
}
for (i=;i<=n;i++) c[i]=;
for (i=;i<vec2[].size();i++) cc(vec2[][i].fi,-); for (i=;i<=n;i++)
{
if (i)
for (j=;j<q[i].size();j++)
ans[q[i][j].sc]-=sum(q[i][j].fi)-sum(i-);
for (j=;j<vec2[i].size();j++)
{
cc(vec2[i][j].fi,);
if (vec2[i][j].sc)
cc(vec2[i][j].sc,-);
}
}
for (i=;i<=m;i++)
printf("%d\n",ans[i]);
}
上一篇:关于java中异常机制


下一篇:BZOJ-2743: [HEOI2012]采花 前缀和 树状数组