Форум программистов, компьютерный форум CyberForum.ru

суффиксный бор.Алгоритм Ахо-Корасик. - C++

Войти
Регистрация
Восстановить пароль
 
Рейтинг: Рейтинг темы: голосов - 18, средняя оценка - 4.61
Gabberr
 Аватар для Gabberr
101 / 101 / 17
Регистрация: 13.10.2009
Сообщений: 402
08.05.2011, 14:22     суффиксный бор.Алгоритм Ахо-Корасик. #1
Приветствую господа!
Решаю вот такую задачку :

Формат входного файла :
В первой строке входного файла записаны 2 целых числа n и k (1<=n,k<=10000).
Во второй и третьей строке записаны n и k целых чисел соответственно.
каждое число не превосходит по модулю 10^9.

Формат выходного файла :

Для каждого числа в третьей строке вывести "Yes" ,если такое число есть во второй строке и "No" в противном случае.

можно без труда решить эту задачку ,используя множество <set> ,но т.к время работы ограничено,решил воспользоваться алгоритмом Ахо-Корасик.

и вот собственно проблема :
входной файл :
5 5
123 213 331 456 198
1 2 3 4 5
нужно чтоб выдало "No" для каждого числа ,а выдаст что каждое число встречается и не по разу! Потому что видимо ищет совпадение чисел из третьей строки с подпоследовательностями из второй.
так как же сделать чтоб искало только по полным совпадениям?

вот код
C++
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <map>
#include <vector>
#include <string>
#include <queue>
//#include <iostream>
using namespace std;
 
class AhoCorasick
{
public:
    typedef void (*Callback) (const char* substr, int begin, int end);
 
    ~AhoCorasick()
    {
        queue<BorNode*> q;
        for(map<char, BorNode*>::const_iterator iter = root.links.begin(); iter != root.links.end(); ++iter)
            q.push(iter->second);
        while(!q.empty())
        {
            BorNode* current_node = q.front();
            q.pop();
            for(map<char, BorNode*>::const_iterator iter = current_node->links.begin();
                iter != current_node->links.end(); ++iter)
                q.push(iter->second);
            delete current_node;
        }
    }
 
    // Метод добавляет строку в бор
    void AddString(const char* str)
    {
        BorNode* node = &root;
        for(const char* s = str; *s; ++s)
        {
            map<char, BorNode*>::iterator iter = node->links.find(*s);
            if(iter != node->links.end())
                node = iter->second;
            else
            {
                BorNode* new_node = new BorNode;
                node->links[*s] = new_node;
                node = new_node;
            }
        }
        node->out = words.size();
        words.push_back(str);
    }
 
    // Метод вычисляет функции неудачи
    void Init()
    {
        root.fail = &root;
        queue<BorNode*> q;
        q.push(&root);
        while(!q.empty())
        {
            BorNode* current_node = q.front();
            q.pop();
            for(map<char, BorNode*>::const_iterator iter = current_node->links.begin();
                iter != current_node->links.end(); ++iter)
            {
                BorNode* child = iter->second;
                char symb = iter->first;
                q.push(child);
 
                BorNode* parent_fail = current_node->fail;
                while(true)
                {
                    map<char, BorNode*>::const_iterator it = parent_fail->links.find(symb);
                    if(it != parent_fail->links.end())
                    {
                        child->fail = it->second != child ? it->second : &root;
                        if(child->out < 0)
                            child->out = child->fail->out;
                        break;
                    }
                    if(parent_fail == &root)
                    {
                        child->fail = &root;
                        break;
                    }
                    else
                        parent_fail = parent_fail->fail;
                }
            }
        }
    }
 
    void Search(const char* str, Callback callback)
    {
        BorNode* current_node = &root;
        for(int pos = 1; *str; ++str, ++pos)
        {
            map<char, BorNode*>::const_iterator iter = current_node->links.find(*str);
            while(iter == current_node->links.end())
            {
                current_node = current_node->fail;
                iter = current_node->links.find(*str);
                if(current_node == current_node->fail)
                    break;
            }
            if(iter != current_node->links.end())
            {
                current_node = iter->second;
 
                if(current_node->out >= 0)
                
                    callback(words[current_node->out].c_str(), pos - words[current_node->out].length(), pos - 1);
                return
            }
        }
        printf("No\n");
 
    }
 
private:
    struct BorNode
    {
        BorNode() : fail(NULL), out(-1) {}
 
        map<char, BorNode*> links;
        BorNode* fail;
        int out;
    };
 
    BorNode root;
    vector<string> words;
};
 
void print(const char* str, int start, int end)
{
    printf("Yes\n");
 //   cout << "Найдена подстрока " << str << " (начало " << start << ", конец " << end << ")\n";
}
 
int main()
{
    int n,k;
    char line[110000];
    char *first_str;
    char second_str[10];
    AhoCorasick ak;
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    scanf("%d ",&n);
    scanf("%d\n",&k);
    first_str=gets(line);
    
    for (int i=0;i<k;++i)
    {
        scanf("%s ",&second_str);
        ak.AddString(second_str);       
    }
    ak.Init();
    ak.Search(first_str,print);
}
Similar
Эксперт
41792 / 34177 / 6122
Регистрация: 12.04.2006
Сообщений: 57,940
08.05.2011, 14:22     суффиксный бор.Алгоритм Ахо-Корасик.
Посмотрите здесь:

Волновой алгоритм (алгоритм Ли) C++
C++ c++/алгоритм
алгоритм C++
C++ Алгоритм
Помогите алгоритм для char переделать в алгоритм для float C++
Алгоритм C++
C++ QR алгоритм
C++ Алгоритм
C++ Суффиксный массив
C++ Реализовать алгоритм Ахо-Корасика
C++ Алгоритм Ахо-Корасика
Нужен алгоритм поиска пути в этом лабиринте (будь то волновой алгоритм или алгоритм правой/левой руки ) C++

Искать еще темы с ответами

Или воспользуйтесь поиском по форуму:
После регистрации реклама в сообщениях будет скрыта и будут доступны все возможности форума.
Ответ Создать тему
Опции темы

Текущее время: 03:56. Часовой пояс GMT +3.
КиберФорум - форум программистов, компьютерный форум, программирование
Powered by vBulletin® Version 3.8.9
Copyright ©2000 - 2017, vBulletin Solutions, Inc.
Рейтинг@Mail.ru