【腾讯】技术研究类和数据分析-2019校园招聘 笔试题

腾讯的笔试是我遇过最难的了，选择题，全是数学，真的难，哎。
题目没拍，先贴上代码，后续补上。
第一题70%，第二题30%，第三题30%。

k = int(raw_input())
str1 = raw_input()
str2 = raw_input()

substring_set = set()
len1 = len(str1)
len2 = len(str2)
for i in range(len1-k+1):
    substring_set.add(str1[i:i+k])

coff = 0
for sub in substring_set:
    for i in range(len2-k+1):
        if str2[i:i+k]==sub:
            coff += 1

print coff

line1 = map(int,raw_input().split())
score1 , score2 = line1[0] , line1[1]
score = score1 + score2

total = 0
n = 0
while total<score:
    n += 1
    total = total + n
if total!=score:
    print -1
else:
    if score1<=n:
        print 1
    else:
        k = 1
        target = score1-n
        while target not in [x for x in range(n-1)]:
            k += 1
            n -= 1
            target -= n
        k += 1

        print k