[牛客网]彩色砖块


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

example

s = “ABAB”,那么小易有六种排列的结果:
“AABB”,”ABAB”,”ABBA”,”BAAB”,”BABA”,”BBAA”
其中只有”AABB”和”BBAA”满足最多只有一对不同颜色的相邻砖块

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#include <iostream>
using namespace std;
class Solution
{
public:
void Colour()
{
string s;
cin>>s;
string a,b,c;
int n=0;
int len=s.length();
b=s[0];
for(int i=0;i<len;i++)
{
a=s[i];

if(a!=b)
n=n+1;
if (n==2)
{
if (s[i]==s[i-2])
n=1;
else
cout<<0<<endl;
return ;
}
b=a;
}
cout<<n+1<<endl;
}
};
int main()
{

Solution ss;
ss.Colour();
}

改进方法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#include <iostream>
using namespace std;
int main()
{
int a[27] = { 0 }, c, sum = 0;
while((c=getchar())>32)
if (!a[c - 65])
{
sum++;
a[c - 65] = 1;
}
if (sum > 2)
cout << 0;
else
cout << sum;
}

算法

26个大写英文字母A~Z的ASCII码值是65~90
getchar()函数

该函数以无符号 char 强制转换为 int 的形式返回读取的字符,如果到达文件末尾或发生读错误,则返回 EOF。