黑化的牛牛

牛牛变得黑化了,想要摧毁掉地球。但他忘记了开启地球毁灭器的密码。牛牛手里有一个字符串S,牛牛还记得从S中去掉一个字符就恰好是正确的密码,请你帮牛牛求出他最多需要尝试多少次密码。
如样例所示S = “ABA”,3个可能的密码是”BA”, “AA”, “AB”.
当S = “A”, 牛牛唯一可以尝试的密码是一个空的密码,所以输出1.
输入描述:
输入包括一个字符串S,字符串长度length(1 ≤ length ≤ 50),其中都是从’A’到’Z’的大写字母。

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

输入例子:
ABA

输出例子:
3

import java.util.HashSet;
import java.util.Scanner;

/*
 * 利用hashset的特性，集合中不能含有重复元素
 * 当输入字符串长度为1时，直接输出0
 * 大于1时，遍历字符串，一次把每位上的元素去除的字符串放入hashset中，
 * 要输出的就是hashset的长度
 */

public class Main {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner in = new Scanner(System.in);
        HashSet<String> set = new HashSet<>();
        String str = in.nextLine();
        char[] ch = str.toCharArray();
        if(ch.length == 1){
            System.out.println(1);
        }else{
            for(int i = 0 ; i < ch.length ; i++){
                String s = "";
                if(i == 0){
                    s += str.substring(i+1);
                }else if(i == ch.length-1){
                    s += str.substring(0, ch.length-1);
                }else{
                    s += str.substring(0, i);
                    s += str.substring(i+1,ch.length);
                }
                set.add(s);
            }
            System.out.println(set.size());
        }
        in.close();
    }

}