首页 > 试题广场 >

添加最少的字符让字符串变成回文串(2)

[编程题]添加最少的字符让字符串变成回文串(2)
  • 热度指数:879 时间限制:C/C++ 2秒,其他语言4秒 空间限制:C/C++ 256M,其他语言512M
  • 算法知识视频讲解
给定一个字符串str,再给定str的最长回文子序列字符串strlps, 请返回在添加字符最少的情况下,让str整体都是回文字符串的一种结果。进阶问题比原问题多了一个参数,请做到时间复杂度比原问题的实现低。

输入描述:
输出包含两行,第一行包含一个字符串代表str,第二行包含一个字符串,代表strips。


输出描述:
输出一行,代表返回的值。
示例1

输入

A1B21C
121

输出

AC1B2B1CA

说明

str=“A1B21C",strlps="121",返回“AC1B2B1CA”或者“CA1B2B1AC”,总之,只要是添加的字符数最少,只返回其中一种结果即可。 
# 首先找到最长回文串元素所在位置
def find_indices(string, strlps):
    indices = []
    j = 0
    for i, c in enumerate(string):
        if j < len(strlps) and c == strlps[j]:
            indices.append(i)
            j += 1
    return indices

# 按照最长回文串元素分割字符串
# 如 A1B21C 会被分割成 ['A', '1', 'B', '2', '', '1', 'C']
def split_palindrom(string, strlps):
    substrs = []
    indices = find_indices(string, strlps)
    for i in range(len(indices) - 1):
        substrs.extend([string[indices[i] + 1: indices[i + 1]], string[indices[i + 1]]])
    substrs = [string[:indices[0]], string[indices[0]]] + substrs + [string[indices[-1] + 1:]]
    return substrs

# 找两个回文串的最短公共回文串(类似于最小公倍数)
def get_palindrom_string(s1, s2):
    if s1 == s2[::-1]:
        return s1, s2
    i = j = 0
    while i < len(s1) and len(s2) - 1 - i > -1 and s1[i] == s2[len(s2) - 1 - i]:
        i += 1
    while j < len(s2) and len(s1) - 1 - i > -1 and s2[j] == s1[len(s1) - 1 - j]:
        j += 1
    unmatch_part_s1 = s1[i: len(s1) - j]
    unmatch_part_s2 = s2[j: len(s2) - i]
    new_part_s1 = unmatch_part_s1 + unmatch_part_s2[::-1]
    new_part_s2 = unmatch_part_s2 + unmatch_part_s1[::-1]
    return s1[:i] + new_part_s1 + s1[len(s1) - j:], s2[:j] + new_part_s2 + s2[len(s2) - i:]

def solve(string, strlps):
    strings = split_palindrom(string, strlps)
    i = 0
    j = len(strings) - 1
    while i < j:
        strings[i], strings[j] = get_palindrom_string(strings[i], strings[j])
        i += 1
        j -= 1
    return "".join(strings)

if __name__ == "__main__":
    string = input()
    strlps = input()
    print(solve(string, strlps))

发表于 2020-09-07 21:18:52 回复(0)