牛客网暑期ACM多校训练营（第一场）J Different Integers（树状数组离线做法）

题目连接 : https://www.nowcoder.com/acm/contest/139/J

题意：给出长为n的数组a1,a2...an，有q次查询，每次查询给出Li,Ri，让求数组中去掉开区间(aLi,aRi)后剩余部分中不同元素的个数的和。

思路 : 树状数组有求区间不同数的做法，现在分开求2个区间，所以考虑拼接一个区间，加在后边就可以树状数组解决问题了。

 

#include<bits/stdc++.h>
using namespace std;
int a[200005];
struct node{
	int l;
	int r;
	int id;
}no[100005];
int ans[200005];
int Ans[200005]; 
map<int,int>M; 
int n,m;
int lowbit(int x)
{
	return x&(-x);	
}
void upda(int x,int val)
{
	while(x<=2*n)
	{
		ans[x]+=val;
		x=x+lowbit(x);
	} 
}
int sum(int x)
{
	int ANS=0;
	while(x>0)
	{
		ANS+=ans[x];
		x=x-lowbit(x);
	}
	return ANS;
}
int cmp(node a,node b)
{
	if(a.r == b.r)
		return a.l < b.l;
	return a.r < b.r; 
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		memset(ans,0,sizeof(ans));
		M.clear();
		for(int i = 1;i <= n;i++)
		{
			scanf("%d",&a[i]);
			a[i+n] = a[i];
		}
		for(int i = 0;i < m; i++)
		{
			scanf("%d%d",&no[i].l,&no[i].r);
			no[i].id = i;
			int temp = no[i].r;
			no[i].r = n + no[i].l;
			no[i].l = temp; 
		}		
		sort(no,no+m,cmp);
		int l=1;
		for(int i = 0;i < m;i++)
		{
			for(int j = l ;j <= no[i].r;j++)
			{
				if(M.find(a[j])!=M.end())
				{
					upda(M[a[j]],-1);
				}
				upda(j,1);
				M[a[j]]=j; 
			}
			l=no[i].r+1; 
			Ans[no[i].id]=sum(no[i].r)-sum(no[i].l-1);
		}
		for(int i = 0 ;i < m ;i++)
			printf("%d\n",Ans[i]);
	}
	return 0;
}

莫队也可以过，但是我T到怀疑人生，后来加了输入挂 更改了分块的大小 才过。但是是当时1000ms过了，后来交了1400ms。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
struct node{
    int l,r,id;
}que[maxn];
int ans[maxn];
int pos[maxn];
int a[maxn];
int vis[maxn];
int sum;
inline bool scan_d(int &num)
{
        char in;bool IsN=false;
        in=getchar();
        if(in==EOF) return false;
        while(in!='-'&&(in<'0'||in>'9')) in=getchar();
        if(in=='-'){ IsN=true;num=0;}
        else num=in-'0';
        while(in=getchar(),in>='0'&&in<='9'){
                num*=10,num+=in-'0';
        }
        if(IsN) num=-num;
        return true;
}
int cmp(node a,node b)//右端点
{
    if(pos[a.l] == pos[b.l])
        return a.r<b.r;
    return pos[a.l] < pos[b.l];
}
    
void add(int x)
{
    if(vis[a[x]] == 0)
        sum++;
    vis[a[x]]++;
}
    
void del(int x)
{
    if(vis[a[x]] == 1)
        sum--;
    vis[a[x]]--;
}
    
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        sum=0;
        int block = (int)pow(n,0.57);
        for(int i=1;i<=n;i++)
        {
            vis[i]=0;
            ans[i]=0;
        }
        for(int i=1;i<=n;i++)
        {
            scan_d(a[i]);
            pos[i]=i/block;
            vis[a[i]]++;
            if(vis[a[i]] == 1)
                sum++;
        }
        for(int i=1;i<=m;i++)
        {
            scan_d(que[i].l);
            scan_d(que[i].r);
            que[i].id=i;
            if(que[i].r - que[i].l <= 1)//挨着
                ans[i] = sum;
            else
            {
                que[i].l++;
                que[i].r--;
            }
        }
        sort(que+1,que+m+1,cmp);
        int l=1,r=0;
        for(int i=1;i<=m;i++)
        {
            if(ans[que[i].id])  continue;
            while(l<que[i].l)   add(l++);
            while(l>que[i].l)   del(--l);
            while(r>que[i].r)   add(r--);
            while(r<que[i].r)   del(++r);
            ans[que[i].id] = sum;
        }
        for(int i=1;i<=m;i++)   printf("%d\n",ans[i]);
    }
    return 0;
}