BZOJ3439:Kpm的MC密码 字典树+dfs序+主席树

标签: dfs序 主席树 字典树
12人阅读 评论(0) 收藏 举报
分类:

Description
系统将随机生成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)


Sample Input
3
cd
abcd
bcd
2
3
1


Sample Output
2
-1
2


你将字符串反着建一遍字典树,然后给每个串开头所属的编号打一个权值,由于可能会有多个串相等,考虑用链表。然后用这个dfs序建主席树,对于一个字符求它编号第k小的后缀串其实就是在它的子树里查询第k小的值,然后就乱水。


#include <cstdio>
#include <cstring>

using namespace std;

struct node {
    int s, v[30];
    node() {
        s = 0;
        memset(v, -1, sizeof(v));
    }
} t[310000]; int tot;
struct trnode {
    int lc, rc, c;
} tr[20 * 110000]; int cnt, rt[110000];
int n, id, pre[110000], a[110000], ll[310000], rr[310000];
char ss[310000];

void bt(int x, int now) {
    int len = strlen(ss + 1);
    for(int i = len; i >= 1; i--) {
        int y = ss[i] - 'a' + 1;
        if(t[x].v[y] == -1) t[x].v[y] = ++tot;
        x = t[x].v[y];
    }
    if(t[x].s != 0) {
        pre[now] = t[x].s;
        t[x].s = now;
    }
    else t[x].s = now;
    a[now] = x;
}

void Link(int &u, int l, int r, int p) {
    if(!u) u = ++cnt;
    tr[u].c++;
    if(l == r) return ;
    int mid = (l + r) / 2;
    if(p <= mid) Link(tr[u].lc, l, mid, p);
    else Link(tr[u].rc, mid + 1, r, p);
}

void Merge(int &u1, int u2) {
    if(!u1 || !u2) {u1 = u1 + u2; return ;}
    tr[u1].c += tr[u2].c;
    Merge(tr[u1].lc, tr[u2].lc);
    Merge(tr[u1].rc, tr[u2].rc);
}

int query(int u1, int u2, int l, int r, int k) {
    if(l == r) return l;
    int c = tr[tr[u2].lc].c - tr[tr[u1].lc].c;
    int mid = (l + r) / 2;
    if(k <= c) return query(tr[u1].lc, tr[u2].lc, l, mid, k);
    else return query(tr[u1].rc, tr[u2].rc, mid + 1, r, k - c);
}

void dfs(int x) {
    ll[x] = id++;
    int now = t[x].s;
    while(now != 0) {
        Link(rt[ll[x]], 1, n, now);
        now = pre[now];
    }
    for(int i = 1; i <= 26; i++) {
        if(t[x].v[i] != -1) {
            int y = t[x].v[i];
            dfs(y);
        }
    }
    rr[x] = id - 1;
}

int main() {
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) {
        scanf("%s", ss + 1);
        bt(0, i);
    }
    dfs(0);
    for(int i = 1; i <= tot; i++) Merge(rt[i], rt[i - 1]);
    for(int i = 1; i <= n; i++) {
        int x = a[i];
        int k; scanf("%d", &k);
        int g1 = rt[ll[x] - 1], g2 = rt[rr[x]];
        if(tr[g2].c - tr[g1].c < k) {
            printf("-1\n");
            continue;
        }
        printf("%d\n", query(g1, g2, 1, n, k));
    }
    return 0;
}
查看评论

【bzoj3439】Kpm的MC密码 trie树+主席树

裸题,倒着建trie树,所有符合要求的串都在它的子树里,然后就是查询子树第k小了,用dfs序+主席树就可以了。 #include #include #include #include #in...
  • u012288458
  • u012288458
  • 2015-11-25 18:02:41
  • 898

bzoj3439: Kpm的MC密码(主席树+DFS序+字典树)

题目传送门 做这题有人跟我说用链表。处理相同的串。 网上都说要。。 其实不用吧。。记录每个串的结尾是在字典树上哪个点就行啊。 然后一个一个插啊。 解法: 因为是后缀所以到这建字典树。 然...
  • Hanks_o
  • Hanks_o
  • 2018-04-18 10:12:56
  • 7

【BZOJ3439】Kpm的MC密码 Trie+dfs序+可持久化线段树

题解: 咳咳。首先补全数据范围:对于100%的数据,1 然后乍一看题(BZ少数据范围),直接排个序,然后插入点信息来一发可持久化线段树求区间第K大、 显然存不下,那么我们可以用Trie存一下这些...
  • Vmurder
  • Vmurder
  • 2015-01-15 22:05:07
  • 1381

【BZOJ3439】Kpm的MC密码,trie树+dfs序+主席树

博文格式有些小改动
  • xym_CSDN
  • xym_CSDN
  • 2016-05-07 21:56:27
  • 675

bzoj3439: Kpm的MC密码

SB题,网上都用了主席树……… 其实只需要每个trie节点开一个vector,记录一下这个节点的子树有哪些字符串, 按顺序加入每个字符串,就可以保证vector是有序的,完了。#include #...
  • DraZxlNDdt
  • DraZxlNDdt
  • 2017-09-18 15:30:25
  • 292

BZOJ3439 KPM的MC密码

传送门 题意:给出n个字符串,对于每一个字符串,输出以他为后缀的串中编号第k小串的编号. 不是很懂为什么网上这道题都是什么主席树,这题明明可以O(N)O(N)来实现啊.只需要Hash一下就好了. ...
  • geng4512
  • geng4512
  • 2016-05-18 21:42:12
  • 423

BZOJ3439: Kpm的MC密码

一开始想后缀数组+主席树…仔细想想…既然完全是后缀,那么直接将所有串reverse,塞到trie即可,现在问题就是查询子树里关键点第k大,在trie树上dfs,按照dfs序建一颗主席树即可。 (当模...
  • MirrorGray
  • MirrorGray
  • 2016-04-09 17:15:20
  • 628

bzoj3439: Kpm的MC密码(四种做法)

四种做法?
  • qq_36797743
  • qq_36797743
  • 2017-09-18 16:47:39
  • 267

[bzoj3439][Tire][主席树]Kpm的MC密码

Description 背景 想Kpm当年为了防止别人随便进入他的MC,给他的PC设了各种奇怪的密码和验证问题(不要问我他是怎么设的。。。),于是乎,他现在理所当然地忘记了密码,只能...
  • Rose_max
  • Rose_max
  • 2018-04-04 07:15:39
  • 61

BZOJ3439 KPM的MC密码 字符串哈希

题目大意:有n个字符串s1~sn,n个数k1~kn,求以si字符串为后缀的所有字符串中第ki个字符串的编号。看样例解释就很容易理解了,不做赘述。学习了大佬的字符串哈希解法。。原地址。。想法是把每一个后...
  • Desico
  • Desico
  • 2018-04-10 14:28:17
  • 5
    个人资料
    持之以恒
    等级:
    访问量: 8702
    积分: 683
    排名: 7万+
    文章存档
    最新评论