牛客网[编程题]--彩色的砖块

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

ABAB

输出

2

思路：求字符串的种类数。

1、使用set函数

#include<iostream>
#include<string>
#include<set>
using namespace std;
int main()
{
    string s;
    set<char> c;
    cin>>s;
    int i, ans=0;
    for(i=0;i<s.size();i++)
    {
        c.insert(s[i]);
        if(c.size()>2)break;
    }
    if(c.size()<=2)
        ans=c.size();
    cout<<ans;
    return 0;
}

2、使用sort，unique和erase函数

unique()函数：去掉相邻且重复的元素。若两个相同的元素连续出现就只留下一个，其余的删除。

此处是伪删除，被删除的元素放在未删除元素序列的后面。

返回：未删除的最后一个元素后面的元素的迭代器。

例如:

 string s("hello,worlddd");

 string::iterator iterEnd=unique(s.begin(),s.end());

得到新的字符串：helo,worldldd。ldd被放在了字符串的最后，同时返回的s.end()指向ldd的第一个字符l。

erase()函数：

string.erase(pos,n)          //删除从pos开始的n个字符    string.erase(0,1);   删除第一个字符

string.erase(pos)           //删除pos处的一个字符（pos是string类型的迭代器）s.erase(s.begin()+3)

string.erase(first,last)    //删除从first到last中间的字符（first和last都是string类型的迭代器）s.erase(s.begin()+3,s.end()-2)

 s.erase(iterEnd,s.end())  //删除从iterEnd到s.end()中间的字符,执行后s的值为的helo,world

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
int main()
{
    string s,s1;
    cin>>s;
    int i, ans=0;
    sort(s.begin(),s.end());
    s.erase(unique(s.begin(),s.end()),s.end());
    if(s.size()<=2)
        ans=s.size();
    cout<<ans;
    return 0;
}