泛化布尔函数以获得更优雅的代码

Question

我一直坚信，如果您正在复制和粘贴代码，那么会有更优雅的解决方案。我目前正在用 C++ 实现一个字符串后缀特里树，并且有两个几乎相同的函数，只是它们的 return 语句不同。

第一个函数检查是否存在子字符串，第二个函数检查它是否是后缀（所有字符串都在末尾添加字符 #）。

子字符串（字符串 S）

bool Trie::substring(string S){
    Node* currentNode = &nodes[0];                          //root
    for(unsigned int i = 0; i < S.length(); i++){
        int edgeIndex = currentNode->childLoc(S.at(i));
        if(edgeIndex == -1)
            return false;
        else currentNode = currentNode->getEdge(edgeIndex)->getTo();
    }
    return true;
}

后缀(字符串S)

bool Trie::suffix(string S){
    Node* currentNode = &nodes[0];                          //root
    for(unsigned int i = 0; i < S.length(); i++){
        int edgeIndex = currentNode->childLoc(S.at(i));
        if(edgeIndex == -1)
            return false;
        else currentNode = currentNode->getEdge(edgeIndex)->getTo();
    }
    return (currentNode->childLoc('#') == -1)? false : true;    //this will be the index of the terminating character (if it exists), or -1.
}

如何更优雅地概括逻辑？也许使用模板？

Answer 1

在这种情况下，我个人所做的是将通用代码移动到函数中，我从多个地方调用它，在我需要的地方，说通用功能。考虑一下：

Node* Trie::getCurrentNode (string S){
    Node* currentNode = &nodes[0];  
    for(unsigned int i = 0; i < S.length(); i++){
        int edgeIndex = currentNode->childLoc(S.at(i));
        if(edgeIndex == -1)
            return nullptr;
        else currentNode = currentNode->getEdge(edgeIndex)->getTo();
    }
    return currentNode;
}

然后，在所有需要的地方使用它：

bool Trie::substring(string S){
    return getCurrentNode (S) != nullptr;
}

bool Trie::suffix(string S){
    Node* currentNode = getCurrentNode(S);
    return currentNode != nullptr && currentNode->childLoc('#') != -1;
    // I decided to simplify the return statement in a way François Andrieux suggested, in the comments. It makes your code more readable.
}

泛化布尔函数以获得更优雅的代码

Generalising Boolean Functions for more elegant code

c++

repeat

suffix-tree