c++ - 泛化 bool 函数以获得更优雅的代码

Question

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

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

子串(字符串 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.
}