2、黑化的牛牛--全国模拟（四）

[编程题] 黑化的牛牛

时间限制：1秒

空间限制：32768K

牛牛变得黑化了,想要摧毁掉地球。但他忘记了开启地球毁灭器的密码。牛牛手里有一个字符串S,牛牛还记得从S中去掉一个字符就恰好是正确的密码,请你帮牛牛求出他最多需要尝试多少次密码。

如样例所示S = "ABA",3个可能的密码是"BA", "AA", "AB".

当S = "A", 牛牛唯一可以尝试的密码是一个空的密码,所以输出1. 

输入描述:

输入包括一个字符串S,字符串长度length(1 ≤ length ≤ 50),其中都是从'A'到'Z'的大写字母。

 

 

输出描述:

输出一个整数,表示牛牛最多需要尝试的密码次数。

 

输入例子:

ABA

 

输出例子:

3

 

解题思路：本题使用set来存储，可以自动过滤相同的字符串，然后针对字符串，求出其删掉第0位-第n-1位的全部结果，存入set中，set的大小即为结果数，其中length==1的情况特殊处理。

#include <iostream>
#include <set>
 
using namespace std;
//去掉字符串的第i位字符
string fun(string str,int n)
{
    int length = str.size();
    string tmp;
    tmp = str;
    for(int i=n;i<length-1;i++)
    {
        tmp[i] = tmp[i+1];
    }
 
    tmp[length-1] = '';
 
    return tmp;
}
int main()
{
    string s;
    while(cin>>s)
    {
        set<string> res;
        int count = 0;
        int length = s.size();
        if(length == 1)
        {
            cout<<1<<endl;
        }
        else
        {
            for(int i=0;i<length;i++)
            {
                string temp;
                temp = fun(s,i);
                res.insert(temp);
            }
            count = res.size();
            cout<<count<<endl;
        }
    }
    return 0;
}