牛客网编程题-----彩色的砖块（python）

一、题目

题目描述

小易有一些彩色的砖块。每种颜色由一个大写字母表示。各个颜色砖块看起来都完全一样。现在有一个给定的字符串s,s中每个字符代表小易的某个砖块的颜色。小易想把他所有的砖块排成一行。如果最多存在一对不同颜色的相邻砖块,那么这行砖块就很漂亮的。请你帮助小易计算有多少种方式将他所有砖块排成漂亮的一行。(如果两种方式所对应的砖块颜色序列是相同的,那么认为这两种方式是一样的。)
例如: s = "ABAB",那么小易有六种排列的结果:
"AABB","ABAB","ABBA","BAAB","BABA","BBAA"
其中只有"AABB"和"BBAA"满足最多只有一对不同颜色的相邻砖块。

输入描述:

输入包括一个字符串s,字符串s的长度length(1 ≤ length ≤ 50),s中的每一个字符都为一个大写字母(A到Z)。

输出描述:

输出一个整数,表示小易可以有多少种方式。

示例1

输入

ABAB

输出

2

二、分析及代码

    题意：给一个由A-Z组成的字符串，让求解最多只有一组不同字符相邻的组合形式。

    分析：通过理解题意可知，如果给出的字符串由三种或三种以上的字符组成，则没有满足题意的组合形式，即res=0。如果给出的字符串由一种字符组成，则只有一种组合形式，即res=1。如果有两种字符串组成，则只有AAAABBBB或BBBBAAAA这种形式，其他均不满足，经过分析后，代码也就很简单，代码如下：

data='ABCDEFGHIJKLMNOPQRSTUVWXYZ'
s = input()
num = 0
res = 0
for x in data:
    if x in s:
        num += 1
if num >=3:
    res = 0
elif num == 1:
    res = 1
else:
    res = 2
print(res)