彩色的砖块

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/iov3Rain/article/details/89005029

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

例如: s = "ABAB",那么小易有六种排列的结果:

"AABB","ABAB","ABBA","BAAB","BABA","BBAA"

其中只有"AABB"和"BBAA"满足最多只有一对不同颜色的相邻砖块。

输入描述:

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

输出描述:

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

示例1

输入

ABAB

输出

2

题目不仔细阅读，有点难懂。整体来说就是变形的排列题

#include<iostream>
#include<unordered_map>
#include<string>
using namespace std;

int main()
{
    string str;
    cin >> str;
    unordered_map<char, int> um;
    //重复板块的类型总数
    int cnt1 = 0;
    //不同板块类型的总数
    int cnt2 = 0;
    for(int i = 0; i < str.size(); i++)
    {
        if(um[str[i]] == 0)
            cnt2++;
        else if(um[str[i]] == 1)
            cnt1++;
    }
    long long ans = (cnt1 <= 2 && cnt1 > 0)? 1 : 0;
    for(int i = 1; i <= cnt2; i++)
    {
        ans *= i;
    }
    cout << ans << endl;
    return 0;
}