11

我在 C++ 中使用 Levenshtein 距离算法来比较两个字符串以测量它们彼此之间的接近程度。但是,普通的 Levenshtein 距离算法不区分由空格分隔的单词边界。这导致距离计算比我想要的要小。我正在比较标题以查看它们彼此之间的接近程度,并且我希望算法不会将来自多个单词的字符计为匹配。

例如,如果我比较这两个字符串,我会得到以下结果,+指定一个匹配项并-指定一个不匹配项:

Al Chertoff Et
Al Church Department of finance Et
+++++------+--++-----++-+------+++
Al Ch      e  rt     of f       Et

我得到一个与"Chertoff"四个单词匹配的单词的距离为 20,"Church Department of finance"而我真的希望它们被认为彼此之间的距离更远,因为不允许字符与多个单词匹配,并且与单词的距离为 25"Chertoff"最匹配一个单词"Department",三个字符匹配:

Al Chertoff Et
Al Church Department of finance Et
+++--------+--++---------------+++
Al         e  rt                Et
         Ch     off

我如何调整 Levenshtein 距离来完成此任务,或者是否有另一种更适合此的距离算法?也许在每个单词上单独使用 Levenshtein 距离并选择距离最小的单词?但是,如果在字符串深处匹配一个单词会导致后续单词匹配不佳,因为它们的匹配最好在字符串的较早位置?这可以通过将 Levenshtein 距离调整为单词级别来以某种方式完成吗?

例如,对于以下更复杂的示例,此想法的最短距离是 20:

Al Chertoff Deport Et
Al Church Department of finance Et
+++++----++++-++---------------+++
Al Ch     Dep rt                Et
     ertoff  o

而不是最大化"Chertoff"的匹配并获得 24 的更长距离:

Al Chertoff Deport Et
Al Church Department of finance Et
+++--------+--++-----+---------+++
Al         e  rt     o          Et
         Ch     off
                  Dep rt

我目前对 Levenshtein 距离的实现如下:

size_t
levenshtein_distance(const std::string& a_compare1,
                     const std::string& a_compare2) {
  const size_t length1 = a_compare1.size();
  const size_t length2 = a_compare2.size();
  std::vector<size_t> curr_col(length2 + 1);
  std::vector<size_t> prev_col(length2 + 1);

  // Prime the previous column for use in the following loop:
  for (size_t idx2 = 0; idx2 < length2 + 1; ++idx2) {
    prev_col[idx2] = idx2;
  }

  for (size_t idx1 = 0; idx1 < length1; ++idx1) {
    curr_col[0] = idx1 + 1;

    for (size_t idx2 = 0; idx2 < length2; ++idx2) {
      const size_t compare = a_compare1[idx1] == a_compare2[idx2] ? 0 : 1;

      curr_col[idx2 + 1] = std::min(std::min(curr_col[idx2] + 1,
                                             prev_col[idx2 + 1] + 1),
                                    prev_col[idx2] + compare);
    }

    curr_col.swap(prev_col);
  }

  return prev_col[length2];
}
4

2 回答 2

6

levenshtein_distance通过在序列容器上创建一个通用算法并包含一个计算两个元素之间距离的成本函数,我可以非常接近你想要的:

template<typename T, typename C>
size_t
seq_distance(const T& seq1, const T& seq2, const C& cost,
             const typename T::value_type& empty = typename T::value_type()) {
  const size_t size1 = seq1.size();
  const size_t size2 = seq2.size();

  std::vector<size_t> curr_col(size2 + 1);
  std::vector<size_t> prev_col(size2 + 1);

  // Prime the previous column for use in the following loop:
  prev_col[0] = 0;
  for (size_t idx2 = 0; idx2 < size2; ++idx2) {
    prev_col[idx2 + 1] = prev_col[idx2] + cost(empty, seq2[idx2]);
  }

  for (size_t idx1 = 0; idx1 < size1; ++idx1) {
    curr_col[0] = curr_col[0] + cost(seq1[idx1], empty);

    for (size_t idx2 = 0; idx2 < size2; ++idx2) {
      curr_col[idx2 + 1] = std::min(std::min(
        curr_col[idx2] + cost(empty, seq2[idx2]),
        prev_col[idx2 + 1] + cost(seq1[idx1], empty)),
        prev_col[idx2] + cost(seq1[idx1], seq2[idx2]));
    }

    curr_col.swap(prev_col);
    curr_col[0] = prev_col[0];
  }

  return prev_col[size2];
}

鉴于上述seq_distance情况,两个句子之间的编辑距离使得无法在单词边界之间进行编辑,可以用以下方式定义:

size_t
letter_distance(char letter1, char letter2) {
  return letter1 != letter2 ? 1 : 0;
}

size_t
word_distance(const std::string& word1, const std::string& word2) {
  return seq_distance(word1, word2, &letter_distance);
}

size_t
sentence_distance(const std::string& sentence1, const std::string& sentence2) {
  std::vector<std::string> words1;
  std::vector<std::string> words2;
  std::istringstream iss1(sentence1);
  std::istringstream iss2(sentence2);
  std::copy(std::istream_iterator<std::string>(iss1),
            std::istream_iterator<std::string>(),
            std::back_inserter(words1));
  std::copy(std::istream_iterator<std::string>(iss2),
            std::istream_iterator<std::string>(),
            std::back_inserter(words2));
  return seq_distance(words1, words2, &word_distance);
}

这是在ideone上工作的代码。我已经测试了几个案例,我很确定它做了正确的事情,但你应该多尝试一下,以确保结果是合理的。

请注意,这并不是您所要求的,因为它忽略了编辑距离测量中的所有空格:我认为修改它不这样做应该不会太难,但我还没有完全考虑过。无论如何,这可能一样好(甚至更好),具体取决于您的需要,所以我会让您决定是否要尝试调整它。

只是一个小提示,您的原始代码在以下两行中略有错误:

curr_col.reserve(length2 + 1);
prev_col.reserve(length2 + 1);

在向量中保留容量,但实际上并没有更改它们的大小,因此在此之后访问数组是未定义的行为。resize如果您要访问某个范围内的元素,您实际上应该使用向量:reserve通常用于您将要push_back一个接一个地访问一定数量的元素的情况(这会随着您的移动而增加大小,而不是一次全部增加)并且您希望避免多次内部重新分配的成本(因为每次超出容量时,内部容量只会增加一定的系数)。

编辑:

此版本将单词之间的空格作为编辑距离的一部分考虑在内,但由于在某些情况下需要添加多个空格,因此结果仍与您的示例不完全相同。

于 2013-03-14T22:01:42.150 回答
-2

如果单个单词的长度不同,则将跨越单词边界。如果您想保持在各个单词中比较的索引,那么您需要使单词的长度相同。例如,这里有一个 Javascript(是的,我知道你问的是 C++,但这是为了说明 - 代码取自维基百科)距离计算例程:

var memo = {};

function d(str1, i, len1, str2, j, len2){
    var key = [i,len1,j,len2].join(',');
    if(memo[key] != undefined) return memo[key];

    if(len1 == 0) return len2;
    if(len2 == 0) return len1;
    var cost = 0;
    if(str1[i] != str2[j]) cost = 1;

    var dist = Math.min(
        d(str1, i+1,len1-1, str2,j,len2)+1, 
        d(str1,i,len1,str2,j+1,len2-1)+1,
        d(str1,i+1,len1-1,str2,j+1,len2-1)+cost);
    memo[key] = dist;
    return dist;
}

var str1 = "Al Chertoff Deport$$$$ $$ $$$$$$$ Et";
var str2 = "Al Church$$ Department of finance Et";

console.log(d(str1, 0, str1.length, str2, 0, str2.length));

请注意我是如何修改两个输入字符串以匹配单个单词级别的。运行这个我得到了 19 的距离。同样,如果我将字符串更改为:

var str1 = "Al Chertoff $$$$$$$$$$ $$ $$$$$$$ Et";
var str2 = "Al Church$$ Department of finance Et";

我得到24的距离。

于 2013-03-14T19:06:37.343 回答