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

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

#include
#include
#include 
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.rM;
	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
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');
		}
	}
}

你可能感兴趣的:(牛客网,ACM)