网易 彩色砖

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

#include<iostream>

#include<string>
#include<vector>
#include<map>
using namespace std;
class solution
{
public:
int bricks_style(string& bricks)
{
vector<int> sum;
map<char, int> mmap;
for (size_t i = 0; i < bricks.length(); i++)
{
char str;
str = bricks[i];
if (mmap.find(str) == mmap.end())//map用法
{
mmap[str] = 1;//map用法
}
else
{
mmap[str]++;//map用法
}
}
return mmap.size();
}
};




int main()
{
string bricks;

solution s;
while (true)
{
bool flag=0;
cout << "please input the bricks letters:" << endl;
cin >> bricks;
for (size_t i = 0; i < bricks.length(); i++)
{
if (bricks[i]<'A' || bricks[i]>'Z')
{
cout << "wrong input" << endl;
flag = 1;
break;
}
}
if (flag==0)
break;
}
    string bricks
    cin>>bricks;
int len=s.bricks_style(bricks);
if (len == 1)
cout << 1<<endl;
else if (len == 2)
cout << 2 << endl;
else
{
cout <<0 << endl;
}
return 0;
}