包含每个子字符串的词典单词数

时间:2016-08-13 03:26:07

标签: algorithm search substring

我遇到了一个问题,要求用户计算K个字符串列表中N个子串的匹配字数(每个字符串的长度为M),以满足以下约束条件:

0 < length of substring <= 100
0 < M <= 100
0 < N <= 10,000
0 < K <= 10,000

例如: {serpent,lose,last}的子串se将产生2。

鉴于此输入的巨大范围,检查每个子字符串的所有K个字符串将太昂贵。由于这个原因,KMP无法运作。使用后缀树预处理字符串将是下一个更好的选择。但我无法为每个单词创建后缀树,因为它会再次导致上述问题。即使我尝试将所有单词连接在一起,问题是我无法检测同一个单词中的子串(例如,{stools,sat}的子串s将产生3)。

有没有更有效的方法来解决这个问题?

1 个答案:

答案 0 :(得分:0)

方法可以是索引要搜索的字符串的每个字符。

“蛇”将给出:

  • s&gt; {0}
  • e&gt; {1,4}
  • r&gt; {2}
  • p&gt; {3}
  • n&gt; {5}
  • t&gt; {6}

由此,对于您在此字符串中搜索的每个单词,请在字典中查找其第一个和最后一个字符。

如果找到它们,您需要查看相应的索引以找到一个可以匹配字符串长度的对。

找到后,您可以进行完整的字符串比较,但仅限于此情况。

代码可以是这样的(c#):

static void Main( string[] args )
{
    List<string> words = new List<string>() { "se", "s", "pen", "oo", "st" };
    List<int> scores = new List<int>( words.Select( w => 0 ) );

    List<string> strings = new List<string>() { "serpent", "lose", "last", "stools", "sat" };

    foreach ( var s in strings )
    {
        var indexes = MakeIndexes( s );

        for ( int i = 0 ; i < words.Count ; i++ )
        {
            scores[i] += Score( words[i], s, indexes );
        }
    }
}

static int Score( string word, string s, Dictionary<char, List<int>> indexes )
{
    int firstPos = 0, lastPos = word.Length - 1;
    char first = word[firstPos];
    char last = word[lastPos];

    List<int> firstIndexes;
    if ( indexes.TryGetValue( first, out firstIndexes ) )
    {
        if ( firstPos == lastPos )
            return 1;
        else
        {
            List<int> lastIndexes;
            if ( indexes.TryGetValue( last, out lastIndexes ) )
            {
                int fiPos = 0, liPos = 0;

                while ( fiPos < firstIndexes.Count && liPos < lastIndexes.Count )
                {
                    int fi = firstIndexes[fiPos], li = lastIndexes[liPos];
                    int len = li - fi;

                    if ( len < lastPos )
                        liPos++;
                    else if ( len == lastPos )
                    {
                        if ( FullCompare( word, s, fi ) )
                            return 1;
                        fiPos++;
                        liPos++;
                    }
                    else
                        fiPos++;
                }
            }
        }
    }

    return 0;
}

static bool FullCompare( string word, string s, int from )
{
    for ( int i = 0 ; i < word.Length ; i++ )
        if ( word[i] != s[i + from] )
            return false;
    return true;
}

static Dictionary<char, List<int>> MakeIndexes( string s )
{
    Dictionary<char, List<int>> result = new Dictionary<char, List<int>>();

    for ( int i = 0 ; i < s.Length ; i++ )
    {
        char c = s[i];

        List<int> indexes;
        if ( result.TryGetValue( c, out indexes ) == false )
        {
            indexes = new List<int>();
            result.Add( c, indexes );
        }

        indexes.Add( i );
    }

    return result;
}