彩色的砖块

来源：

牛客网2018校招真题

题目描述：

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

输入描述：

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

输出描述：

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

示例：

输入

ABAB

输出

2

思路：

这道题的思路很重要，只要看懂题，做出来就简单了

我们考虑输入字符串的字符种类，如果是一种，它肯定是漂亮的，有一种排列方式，如果是两种字符，就只有两种排列方式是漂亮的，如果是三种或三种以上类型的字符，你会发现无论怎么排列都不可能是漂亮的，所以只需要计算出输入字符串的字符种类即可，可以使用桶排序来计算

AC代码：

#include<bits/stdc++.h>
using namespace std;
int a[27];
int main(){
    string s;
    int b;
    cin>>s;
    for(int i=0;i<s.size();i++){
        a[s[i]-'A']++;
    }
    for(int i=0;i<27;i++){
        if(a[i]!=0){
            b++;
        }
    }
    if(b==1){
        cout<<1<<endl;
    }else if(b==2){
        cout<<2<<endl;
    }else{
        cout<<0<<endl;
    }
}