티스토리 뷰

알고리즘 관련/BOJ

BOJ)10256 돌연변이

JASON 자손9319 2017. 8. 3. 02:53

문제: icpc.me/10256


N^3으로 만들 수 있는 모든 마커를 트라이에 넣어준 뒤 아호코라식을 통하여 몇개의 마커가 주어진 문자열에서 출연하는지 개수를 세주면 되는 문제이다.


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
struct trie {
    trie *go[4];
    trie *fail;
    int out;
    trie() {
        for (int i = 0; i < 4; i++)
            go[i] = NULL;
        out = 0;
    }
    ~trie() {
        for (int i = 0; i < 4; i++)
            if (go[i])delete go[i];
    }
    void insert(const char *key) {
        if (*key == '\0') {
            out = 1;
            return;
        }
        int next;
        if (*key == 'A')next = 0;
        else if (*key == 'G')next = 1;
        else if (*key == 'T')next = 2;
        else next = 3;
        if (!go[next])
            go[next] = new trie;
        go[next]->insert(key + 1);
    }
};
int solve(char *str, trie *t) {
    int ret = 0;
    trie* here = t;
    for (int i = 0; str[i]; i++) {
        int next;
        if (str[i] == 'A')next = 0;
        else if (str[i] == 'G')next = 1;
        else if (str[i] == 'T')next = 2;
        else next = 3;
        while (here != t&&!here->go[next])
            here = here->fail;
        if (here->go[next])
            here = here->go[next];
        if (here->out)
            ret += here->out;
    }
    return ret;
}
void bfs(trie* root) {
    queue<trie*> qu;
    qu.push(root);
    while (qu.size()) {
        trie *here = qu.front();
        qu.pop();
        for (int i = 0; i < 4; i++) {
            trie* next = here->go[i];
            if (!next)continue;
            if (here == root)next->fail = root;
            else {
                trie *failure = here->fail;
                while (failure != root&&!failure->go[i])
                    failure = failure->fail;
                if (failure->go[i])failure = failure->go[i];
                next->fail = failure;
            }
            next->out += next->fail->out;
            qu.push(next);
        }
    }
}
int t, n, m;
char s[1000010], v[101];
int main() {
    scanf("%d"&t);
    while (t--) {
        trie* tr = new trie;
        scanf("%d%d"&n, &m);
        scanf("%s %s"&s, &v);
        for (int i = 0; i < m; i++) {
            for (int j = i + 1; j <= m; j++) {
                reverse(v + i, v + j);
                tr->insert(v);
                reverse(v + i, v + j);
            }
        }
        bfs(tr);
        printf("%d\n", solve(s, tr));
        delete tr;
    }
    return 0;
}
cs


'알고리즘 관련 > BOJ' 카테고리의 다른 글

BOJ)1938 통나무 옮기기  (0) 2017.08.03
BOJ)1445 일요일 아침의 데이트  (11) 2017.08.03
BOJ)10256 돌연변이  (0) 2017.08.03
BOJ)9250 문자열 집합 판별  (0) 2017.08.02
BOJ)1063 킹  (0) 2017.08.02
BOJ)1514 자물쇠  (0) 2017.08.01
댓글
댓글쓰기 폼