Question

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

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

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

推广布尔函数以获得更优雅的代码

1 个答案: