牛客网暑期ACM多校训练营(第一场)J : Different Integers

正好前几天学了一下莫队,看到这个题,就赶紧上去敲了个莫队,后来就。。。tle哭了
加了输入输出优化,还是没啥卵用。。。。可能是本人太菜了,别人都卡过去了。挣扎了半个小时,一直tle,后来就。。。。放弃莫队,并且放弃比赛(溜了溜了),去做杭电上一个树形莫队的题去了,也一直tle,心态崩了。。。
接下来就是补题阶段:
看了叉姐的直播,用树状数组来做的。
我也用树状数组维护区间不相同数的个数,我先把区间 [1,n]扩增了一倍,就是区间
[ 1,n] 后面再加一个区间[1,n]
这样询问[1,l]和[r,n]两个区间就相当于询问[r,n+l]一个区间
用树状数组来维护区间不相同数的个数,其实挺简单的
推荐博客:spoj D-query 区间不同数个数 主席树||离线+树状数组
看代码即可。

#include<cstdio>
#include<map>
#include<algorithm> 
using namespace std;

const int maxn=200000+100;

int bit[maxn];
int ans[maxn];
int Ans[maxn];
int n,m;

inline int lowbit(int x){

    return x&(-x);
}

inline void Add(int x,int v){

    while(x<=n){

        bit[x]+=v;
        x+=lowbit(x);
    }
} 

inline int getSum(int x){

    int sum=0;
    while(x>0){

        sum+=bit[x];
        x-=lowbit(x);
    }
    return sum;
}

struct Query{

    int l,r;
    int id;
}query[maxn];

inline bool cmp(Query a,Query b){

    return a.r<b.r;
}

int main(){

    map<int,int>M;
    while(scanf("%d%d",&n,&m)==2){

        M.clear();
        for(int i=1;i<=n;i++) scanf("%d",&ans[i]),ans[n+i]=ans[i],bit[i]=bit[n+i]=0;
        for(int i=1;i<=m;i++){

            scanf("%d%d",&query[i].r,&query[i].l);
            query[i].r+=n;
            query[i].id=i; 
        }
        n*=2;
        sort(query+1,query+m+1,cmp);
        int pre=1;
        for(int i=1;i<=m;i++){

            for(int j=pre;j<=query[i].r;j++){

                if(M[ans[j]]!=0) Add(M[ans[j]],-1);
                Add(j,1);
                M[ans[j]]=j;
            }
            pre=query[i].r+1;
            Ans[query[i].id]=getSum(query[i].r)-getSum(query[i].l-1);
        }
        for(int i=1;i<=m;i++) printf("%d\n",Ans[i]);
    }
} 

主席树,然而卡不过去,后面我把一些博客用主席书写的代码,重新交了一次,都没过,不知道是不是服务器的问题,还是数据问题。
虽然卡不过去,还是记录一下。

#include<bits/stdc++.h>
using namespace std;

const int maxn=200000+100;
struct Tree{

    int l,r;
    int sum;
}tree[maxn*40];

int Root[maxn];
int num[maxn];
int ans[maxn];
int tot;
int n,m;

inline void build(int &New,int l,int r){

    New=++tot;
    tree[New].l=0;
    tree[New].r=0;
    tree[New].sum=0;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(tree[New].l,l,mid);
    build(tree[New].r,mid+1,r);
}

inline void update(int l,int r,int &New,int Old,int ind,int v){

    New=++tot;
    tree[New]=tree[Old];
    if(l==r){

        tree[New].sum+=v;
        return ;
    }
    int mid=(l+r)>>1;
    if(mid>=ind) update(l,mid,tree[New].l,tree[Old].l,ind,v);
    else update(mid+1,r,tree[New].r,tree[Old].r,ind,v);
    tree[New].sum=tree[tree[New].l].sum+tree[tree[New].r].sum;
}

inline int query(int l,int r,int L,int R,int ind){

    if(L<=l && R>=r) return tree[ind].sum;
    int mid=(l+r)>>1;
    if(mid>=R) return query(l,mid,L,R,tree[ind].l);
    else if(L>mid) return query(mid+1,r,L,R,tree[ind].r);
    else return query(l,mid,L,R,tree[ind].l)+query(mid+1,r,L,R,tree[ind].r); 
}

inline void scan_d(int &res){

    res=0;
    char ch;
    ch=getchar();
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') res=res*10+(ch-'0'),ch=getchar();
}

inline void print_d(int res){

    if(res>9) print_d(res/10);
    putchar(res%10+'0');
}

int main(){

    while(scanf("%d%d",&n,&m)==2){
        if(!n) break;
        tot=0;
        for(int i=1;i<=n;i++) scan_d(ans[i]),ans[i+n]=ans[i],num[ans[i]]=-1;
        n*=2;
        build(Root[0],1,n);
        for(int i=1;i<=n;i++){

            if(num[ans[i]]==-1){

                update(1,n,Root[i],Root[i-1],i,1);
                num[ans[i]]=i;
            }
            else{

                int tmp;
                update(1,n,tmp,Root[i-1],num[ans[i]],-1);
                update(1,n,Root[i],tmp,i,1); 
                num[ans[i]]=i;
            }
        }
        for(int i=1;i<=m;i++){

            int l,r;
            scan_d(r);
            scan_d(l);
            r+=n/2;
            print_d(query(1,n,l,r,Root[r])),putchar('\n');
        }
    }
}

猜你喜欢

转载自blog.csdn.net/qq_37960603/article/details/81122979