牛客网在线编程（7）：彩色宝石项链-python_python用turtle画蓝宝石项链的代码-CSDN博客

本文链接：https://blog.csdn.net/weixin_41481113/article/details/83717746

题目描述

有一条彩色宝石项链，是由很多种不同的宝石组成的，包括红宝石，蓝宝石，钻石，翡翠，珍珠等。有一天国王把项链赏赐给了一个学者，并跟他说，你可以带走这条项链，但是王后很喜欢红宝石，蓝宝石，紫水晶，翡翠和钻石这五种，我要你从项链中截取连续的一小段还给我，这一段中必须包含所有的这五种宝石，剩下的部分你可以带走。如果无法找到则一个也无法带走。请帮助学者找出如何切分项链才能够拿到最多的宝石。

输入描述:

我们用每种字符代表一种宝石，A表示红宝石，B表示蓝宝石，C代表紫水晶，D代表翡翠，E代表钻石，F代表玉石，G代表玻璃等等，我们用一个全部为大写字母的字符序列表示项链的宝石序列，注意项链是首尾相接的。每行代表一种情况。

输出描述:

输出学者能够拿到的最多的宝石数量。每行一个

示例1

输入

ABCYDYE
ATTMBQECPD

输出

1
3

代码实现

import sys

lines=sys.stdin.readlines()

alpha_basic=['A','B','C','D','E']
for line in lines:
    line=line.strip()
    poly=''
    alpha=[]
    Index=[]
    for i,item in enumerate(line):
        poly+=item
        if item in alpha_basic:
            alpha.append(item)
            Index.append(i)
            if sorted(list(set(alpha))) == alpha_basic:
                restart=i+1
                break
    Length=len(line)
    line+=poly
    result=Length+Index[0]-Index[-1]-1
    for i in range(restart,len(line)):
        if line[i] in alpha_basic:
            alpha.append(line[i])
            Index.append(i)
            alpha_=[]
            for j in reversed(alpha):
                alpha_.append(j)
                if sorted(list(set(alpha_))) == alpha_basic:
                    break
            alpha=alpha[-len(alpha_):]
            Index=Index[-len(alpha_):]
            result=max(result,Length+Index[0]-Index[-1]-1)
    print(result)