BZOJ 3439: Kpm的MC密码( trie + DFS序 + 主席树 )

把串倒过来插进trie上, 那么一个串的kpm串就是在以这个串最后一个为根的子树, 子树k大值的经典问题用dfs序+可持久化线段树就可以O(NlogN)解决

------------------------------------------------------------------

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<cctype>

using namespace std;

#define c(x) ((x) - 'a')

#define chk(x) ((x) >= 'a' && (x) <= 'z')

const int maxn = 100009;

const int maxl = 300009;

const int C = 26;

int V[maxn], seq[maxl], L[maxl], R[maxl];

int N, node_n, Val, n;

char s[maxl];

inline int read() {

char c = getchar();

int ret = 0;

for(; !isdigit(c); c = getchar());

for(; isdigit(c); c = getchar())

ret = ret * 10 + c - '0';

return ret;

}

struct O {

O* next;

int v;

} Opool[maxn], *Opt = Opool, *head[maxn];

inline O* AddO(int x, int v) {

Opt->v = v;

Opt->next = head[x];

return head[x] = Opt++;

}

struct node {

node *ch[C], *f;

O* t;

int n;

node() {

memset(ch, 0, sizeof ch);

f = 0;

t = 0;

}

} Pool[maxl], *Pt, *Rt;

inline node* newNode() {

Pt->n = node_n++;

return Pt++;

}

void Init_trie() {

node_n = 0;

Pt = Pool;

Rt = newNode();

}

void Insert(int x, int len) {

node* t = Rt;

while(len--) {

if(!t->ch[c(s[len])])

t->ch[c(s[len])] = newNode();

t = t->ch[c(s[len])];

}

t->t = AddO(V[x] = t->n, x + 1);

}

void Init() {

Init_trie();

n = 1;

N = read();

for(int i = 0; i < N; i++) {

int len = 0;

for(s[len] = getchar(); !chk(s[len]); s[len] = getchar());

for(s[++len] = getchar(); chk(s[len]); s[++len] = getchar());

Insert(i, len);

}

void DFS(node* t) {

L[t->n] = n + 1;

for(O* x = head[t->n]; x; x = x->next)

seq[++n] = x->v;

for(int i = 0; i < C; i++)

if(t->ch[i]) DFS(t->ch[i]);

R[t->n] = n;

}

struct Node {

Node *lc, *rc;

int v;

Node() {

lc = rc = 0;

}

} pool[maxl * 40], *pt, *Root[maxl];

void Init_sgt() {

Root[0] = (pt = pool)++;

Root[0]->lc = Root[0]->rc = Root[0];

Root[0]->v = 0;

}

Node* Modify(Node* t, int l, int r) {

Node* h = pt++;

h->v = t->v + 1;

if(l != r) {

int m = (l + r) >> 1;

if(Val <= m) {

h->lc = Modify(t->lc, l, m);

h->rc = t->rc;

} else {

h->lc = t->lc;

h->rc = Modify(t->rc, m + 1, r);

}

return h;

}

void Build() {

for(int i = 1; i <= n; i++) if(seq[i]) {

Val = seq[i];

Root[i] = Modify(Root[i - 1], 1, N);

} else

Root[i] = Root[i - 1];

}

int Query(int x, int Rk) {

Node *L = Root[::L[x] - 1], *R = Root[::R[x]];

if(R->v - L->v < Rk)

return -1;

int l = 1, r = N;

while(l < r) {

int m = (l + r) >> 1;

if(R->lc->v - L->lc->v < Rk) {

Rk -= R->lc->v - L->lc->v;

L = L->rc;

R = R->rc;

l = m + 1;

} else {

L = L->lc;

R = R->lc;

r = m;

}

return l;

}

void Work() {

for(int i = 0; i < N; i++)

printf("%d\n", Query(V[i], read()));

}

int main() {

Init();

DFS(Rt);

Init_sgt();

Build();

Work();

return 0;

}

------------------------------------------------------------------

3439: Kpm的MC密码

Time Limit: 15 Sec Memory Limit: 256 MB
Submit: 311 Solved: 153
[Submit][Status][Discuss]

Description

背景

想Kpm当年为了防止别人随便进入他的MC，给他的PC设了各种奇怪的密码和验证问题（不要问我他是怎么设的。。。），于是乎，他现在理所当然地忘记了密码，只能来解答那些神奇的身份验证问题了。。。

描述

Kpm当年设下的问题是这样的：

现在定义这么一个概念，如果字符串s是字符串c的一个后缀，那么我们称c是s的一个kpm串。

系统将随机生成n个由a…z组成的字符串，由1…n编号（s1,s2…,sn），然后将它们按序告诉你，接下来会给你n个数字，分别为k1…kn，对于每一个ki，要求你求出列出的n个字符串中所有是si的kpm串的字符串的编号中第ki小的数，如果不存在第ki小的数，则用-1代替。（比如说给出的字符串是cd,abcd,bcd,此时k1=2，那么”cd”的kpm串有”cd”,”abcd”,”bcd”,编号分别为1,2,3其中第2小的编号就是2）（PS：如果你能在相当快的时间里回答完所有n个ki的查询，那么你就可以成功帮kpm进入MC啦~~）

Input

第一行一个整数 n 表示字符串的数目

接下来第二行到n+1行总共n行，每行包括一个字符串，第i+1行的字符串表示编号为i的字符串

接下来包括n行，每行包括一个整数ki，意义如上题所示

Output

包括n行，第i行包括一个整数，表示所有是si的kpm串的字符串的编号中第ki小的数

Sample Input

3
cd
abcd
bcd
2
3
1

Sample Output

2
-1
2

样例解释

“cd”的kpm 串有”cd”,”abcd”,”bcd”,编号为1,2,3，第2小的编号是

2，”abcd”的kpm串只有一个，所以第3小的编号不存在，”bcd”的kpm

串有”abcd”,”bcd”,第1小的编号就是2。

数据范围与约定

设所有字符串的总长度为len

对于100%的数据，1<=n<=100000,0

HINT

Source

Kpmcup#0 By Greens