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;
}