Найти количество подстрок в строке

Question

надо найти количество подстрок в строке. Т. е., сколько раз встретится первая переменная во втором

input:

aba
ababaa

output:

мой код:

s = input()
n = input()

if s in n:
    print(n.count(s))

пробовал и регулярным выражением:

import re

s = input()
t = input()

pattern=re.compile(s)
result = re.findall(pattern,t)

print(len(result))

Answer 1

Накидал алгоритм, попробуйте:

s = 'aba'
t = 'ababaa'

indxs = set()
for i in range(len(t)):
    idx = t.find(s, i)
    if idx != -1:
        indxs.add(idx)

print(len(indxs), indxs)
# 2 {0, 2}

Answer 2

Доработал Ваше второе решение:

import re

s = 'aba'
t = 'ababaa'

pattern = re.compile('(?=(' + re.escape(s) + '))')

result = pattern.findall(t)
print(len(result))

re.escape - экранировать спец-символы регулярных выражений в строке (чтобы корректно обработались всякие ., () и тд);
(?=(aba)) - небольшой трюк с positive lookahead, чтобы можно было искать пересекающиеся совпадения. Подробнее можно посмотреть на regular-expressions.info.

Answer 3

Или просто проверять, с чего начинается очередной срез:

s = 'aba'
t = 'ababaa'

counter = 0
for i in range(len(t)):
   if t[i:].startswith(s):
      counter += 1

>>> print(counter)
2

Answer 4

Можно воспользоваться алгоритмом Кнута — Морриса — Пратта. Он позволяет за линейное время найти все подстроки.

# Knuth-Morris-Pratt string matching
# David Eppstein, UC Irvine, 1 Mar 2002

#from http://code.activestate.com/recipes/117214/
def KnuthMorrisPratt(text, pattern):

    '''Yields all starting positions of copies of the pattern in the text.
Calling conventions are similar to string.find, but its arguments can be
lists or iterators, not just strings, it returns all matches, not just
the first one, and it does not need the whole text in memory at once.
Whenever it yields, it will have read the text exactly up to and including
the match that caused the yield.'''

    # allow indexing into pattern and protect against change during yield
    pattern = list(pattern)

    # build table of shift amounts
    shifts = [1] * (len(pattern) + 1)
    shift = 1
    for pos in range(len(pattern)):
        while shift <= pos and pattern[pos] != pattern[pos-shift]:
            shift += shifts[pos-shift]
        shifts[pos+1] = shift

    # do the actual search
    startPos = 0
    matchLen = 0
    for c in text:
        while matchLen == len(pattern) or \
              matchLen >= 0 and pattern[matchLen] != c:
            startPos += shifts[matchLen]
            matchLen -= shifts[matchLen]
        matchLen += 1
        if matchLen == len(pattern):
            yield startPos

Воспользуемся генератором:

result = sum(1 for x in KnuthMorrisPratt('ababaa', 'aba'))

БЛОГ НА HUSL

Найти количество подстрок в строке

Ответы (4 шт):