【NOI2018模拟3.10】子序列

Description

求一个序列的所有2^n-1个子序列的字典序前k小的哈希值。
n,k<=1e5

Solution

这道题专门狙杀我这种老年菜鸡
我们考虑一种从旧的子序列扩展出新的子序列的方法。
对于一个子序列，我们肯定是找到它后面第一个比它小的位置来扩展。
我们把一样的子序列放在一起考虑，维护所有的结束位置。
然后我们枚举所有可能的更新位置，暴力判断哪些子序列可以用这些位置进行更新。
这样子可以通过优美的实现达到O(n log n)

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

int read() {
    char ch;
    for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
    int x=ch-'0';
    for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x;
}

void write(int x) {
    if (!x) {puts("0");return;}
    char ch[20];int tot=0;
    for(;x;x/=10) ch[++tot]=x%10+'0';
    fd(i,tot,1) putchar(ch[i]);
    puts("");
}

const int N=1e5+5;

int n,k,seed,p,tmp,mx,a[N],nxt[N],lst[N],rt[N],ans[N],L[N];

int tr[N<<5],ls[N<<5],rs[N<<5],id[N<<5],size[N<<5],tot;

void insert(int &v,int l,int r,int x,int y) {
    tr[++tot]=tr[v];
    ls[tot]=ls[v];rs[tot]=rs[v];
    size[tot]=size[v];v=tot;
    if (l==r) {size[v]=1;id[v]=y;return;}
    int mid=l+r>>1;
    if (x<=mid) insert(ls[v],l,mid,x,y);
    else insert(rs[v],mid+1,r,x,y);
    size[v]=size[ls[v]]+size[rs[v]];
}

int find(int v,int l,int r,int x) {
    if (l==r) return id[v];
    int mid=l+r>>1;
    if (size[ls[v]]>=x) return find(ls[v],l,mid,x);
    else return find(rs[v],mid+1,r,x-size[ls[v]]);
}

void dfs(int len,int ha,int l,int r) {
    if (tmp==k||l>r) return;
    fo(i,1,size[rt[len+1]]) {
        int nl=tmp+1;
        int now=find(rt[len+1],1,mx,i);
        int H=((ll)ha*seed%p+a[now])%p;
        for(int j=lst[a[now]];j;j=nxt[j]) {
            fo(w,l,r) {
                if (j<=L[w]) break;
                ans[++tmp]=H;L[tmp]=j;
                if (tmp==k) return;
            }
        }
        int nr=tmp;
        reverse(L+nl,L+nr+1);
        dfs(now,H,nl,nr);
        if (tmp==k) return;
    }
}

int main() {
    freopen("sequence.in","r",stdin);
    freopen("sequence.out","w",stdout);
    n=read();k=read();seed=read();p=read();
    fo(i,1,n) {
        mx=max(mx,a[i]=read());
        nxt[i]=lst[a[i]];
        lst[a[i]]=i;
    }
    fd(i,n,1) insert(rt[i]=rt[i+1],1,mx,a[i],i);
    dfs(0,0,0,0);
    fo(i,1,k) write(ans[i]);
    return 0;
}