ZSTU OJ 4272 最佳淘汰算法

线段树。

处理出每个位置下一个位置是哪里。然后搞个线段树找一下最大值就可以了。

#include<map>
#include<set>
#include<ctime>
#include<cmath>
#include<queue>
#include<string>
#include<stack>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
 
int n,m,a[];
bool In[];
 
int Seg[],nx[],p[];
int POS,sx[],A[];
 
void update(int p,int val,int l,int r,int rt)
{
    if(l==r)
    {
        Seg[rt]=val;
        return ;
    }
 
    int m=(l+r)/;
 
    if(p>=l&&p<=m) update(p,val,l,m,*rt);
    else update(p,val,m+,r,*rt+);
 
    Seg[rt] = max(Seg[*rt],Seg[*rt+]);
}
 
void get(int l,int r,int rt)
{
    if(l==r)
    {
        POS=l;
        return ;
    }
 
    int m=(l+r)/;
 
    if(Seg[*rt]>Seg[*rt+]) get(l,m,*rt);
    else get(m+,r,*rt+);
}
 
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=;i<=n;i++) scanf("%d",&a[i]);
 
        for(int i=;i<=;i++) p[i]=n+;
        for(int i=n;i>=;i--) nx[i] = p[a[i]], p[a[i]]=i;
 
        memset(In,,sizeof In);
        memset(Seg,,sizeof Seg);
        memset(sx,,sizeof sx);
        memset(A,,sizeof A);
 
        int sz=;
 
        for(int i=;i<=n;i++)
        {
            if(In[a[i]])
            {
                update(a[i],nx[i],,,);
                continue;
            }
            if(sz<m)
            {
                sz++;
                A[sz]=a[i]; sx[a[i]]=sz;
                update(a[i],nx[i],,,);
            }
 
            else
            {
                get(,,);
                update(POS,,,,);
 
                int t = sx[POS]; sx[POS]=; In[POS]=;
 
                update(a[i],nx[i],,,);
                sx[a[i]]=t; A[t]=a[i];
            }
 
            In[a[i]]=;
        }
 
        for(int i=;i<=sz;i++)
        {
            printf("%d",A[i]);
            if(i<sz) printf(" ");
            else printf("\n");
        }
    }
    return ;
}