bzoj2743: [HEOI2012]采花(树状数组)

题目传送门

解法:
想了蛮久。

首先这道题肯定不能用前缀和相减什么的。
那么只能用区间1~l来表示答案。
因为1~r就没有了l的限制。

记录每个点上一个出现的位置。
last[i]表示i上一个出现的位置。
能采到i这个位置上的花是从last[last[i]]+1这个位置开始的。
然后last[i]+1开始就采不了这朵花了。

树状数组维护下

代码实现:

#include
#include
#include
#include
#include
#include
#include
using namespace std;
int lowbit(int x) {return x&-x;}
int n,s[1100000],a[1100000];
void change(int x,int c) {while(x<=n) {s[x]+=c;x+=lowbit(x);}}
int find_sum(int x) {
    int ans=0;
    while(x>0){ans+=s[x];x-=lowbit(x);}
    return ans;
}
int last[1100000],next[1100000];
struct node {int l,r,id,ans;}Q[1100000];
bool cmp(node n1,node n2) {return n1.rbool cmp1(node n1,node n2) {return n1.idint main() {
    int c,m;scanf("%d%d%d",&n,&c,&m);
    memset(last,0,sizeof(last));memset(next,0,sizeof(next));
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);last[i]=next[a[i]];next[a[i]]=i;
    }
    for(int i=1;i<=m;i++){scanf("%d%d",&Q[i].l,&Q[i].r);Q[i].id=i;}
    sort(Q+1,Q+1+m,cmp);int now=0;memset(s,0,sizeof(s));
    for(int i=1;i<=m;i++) {
        while(nowif(last[now]!=0) {
                change(last[last[now]]+1,1);
                change(last[now]+1,-1);
            }
        }Q[i].ans=find_sum(Q[i].l);
    }sort(Q+1,Q+1+m,cmp1);
    for(int i=1;i<=m;i++)printf("%d\n",Q[i].ans);
    return 0;
}

你可能感兴趣的:(BZOJ,树状数组)