BaoBao Loves Reading ZOJ - 4117 （思维/求区间不同种类数）

#include<bits/stdc++.h>

using namespace std;

const int maxn=2e5+10;
const double EPS=1e-12;

#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long

int tree[maxn];

int ans[maxn],pre[maxn];

int vis[maxn],arr[maxn];

int n;

int lowbit(int x)
{
    return x&(-x);
}

void add(int x,int val)
{
    while(x<=n)
    {
        tree[x]+=val;
        x+=lowbit(x);
    }
    //cout<<"hereadd"<<endl;
}

int sum(int x)
{
    int temp=0;
    while(x>0)
    {
        temp+=tree[x];
        x-=lowbit(x);
    }
    //cout<<"heresum"<<endl;
    return temp;
}

void init()
{
    mem(arr,0);
    mem(vis,0);
    mem(tree,0);
    mem(ans,0);
    mem(pre,0);
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        init();
        cin>>n;
        for(int i=1;i<=n;i++)
        {
            cin>>arr[i];
            pre[i]=vis[arr[i]];
            vis[arr[i]]=i;
        }

        //cout<<"here"<<endl;

        for(int i=1;i<=n;i++)
        {
            if(pre[i])
            {
                add(pre[i],-1);
                add(i,1);
                int temp=sum(i)-sum(pre[i]-1);
                ans[temp]++;
            }
            else add(i,1);
        }

        //cout<<"here"<<endl;

        int tot=0;
        for(int i=1;i<=n;i++)
        {
            tot+=ans[i];
            if(i!=1) cout<<" ";
            cout<<n-tot;
        }
        cout<<"\n";
    }


    return 0;
}