У меня есть дерево суффиксов, каждый узел этого дерева является структурой
struct state {
int len, link;
map<char,int> next; };
state[100000] st;
Мне нужно сделать DFS для каждого узла и получить все строки, которые я могу достичь, но я не знаю, как сделать.
Это моя функция DFS
void getNext(int node){
for(map<char,int>::iterator it = st[node].next.begin();it != st[node].next.end();it++){
getNext(it->second);
}
}
Будет прекрасно, если я смогу сделать что-то вроде
map<int,vector<string> >
где int — это узел моего дерева и векторных строк, к которым я могу добраться
сейчас работает
void createSuffices(int node){//, map<int, vector<string> > &suffices) {
if (suffices[sz - 1].size() == 0 && (node == sz - 1)) {
// node is a leaf
// add a vector for this node containing just
// one element: the empty string
//suffices[node] = new vector<string>
//suffices.add(node, new vector<string>({""}));
vector<string> r;
r.push_back(string());
suffices[node] = r;
} else {
// node is not a leaf
// create the vector that will be built up
vector<string> v;
// loop over each child
for(map<char,int>::iterator it = st[node].next.begin();it != st[node].next.end();it++){
createSuffices(it->second);
vector<string> t = suffices[it->second];
for(int i = 0; i < t.size(); i ++){
v.push_back(string(1,it->first) + t[i]);
}
}
suffices[node] = v;
}
}
Вы можете передать map<int, vector<string>>
вместе с вашей глубиной первого поиска. Когда рекурсивный вызов возвращается из определенного узла n
, вы знаете, что все, что достаточно от этого узла готовы. Мои навыки C ++ слишком ограничены, поэтому я напишу это в псевдокоде:
void createSuffices(int node, map<int, vector<string>> suffices) {
if (st[node].next.empty()) {
// node is a leaf
// add a vector for this node containing just
// one element: the empty string
suffices.add(node, new vector<string>({""}));
} else {
// node is not a leaf
// create the vector that will be built up
vector<string> v;
// loop over each child
foreach pair<char, int> p in st[node].next {
// handle the child
createSuffices(p.second, suffices);
// prepend the character to all suffices of the child
foreach string suffix in suffices(p.second) {
v.add(concatenate(p.first, suffix));
}
}
// add the created vector to the suffix map
suffices.add(node, v);
}
}
Других решений пока нет …