笔试刷题——度度熊找子串（百度2017秋招真题）

题目

度度熊收到了一个只有小写字母的字符串S，他对S的子串产生了兴趣，S的子串为S中任意连续的一段。他发现，一些子串只由一种字母构成，他想知道在S中一共有多少种这样的子串。

例如在串”aaabbaa”中，度度熊想找的子串有”a”,”aa”,”aaa”,”b”,”bb”五种。

（本题只考虑子串的种数，相同的子串在多个位置出现只算一次）
输入
输入只有一行，一个字符串，长度不超过100000，只由小写字母组成

样例输入
aaabbaa

输出
输出一行，符合要求的子串种数

样例输出
5

思路

set集合存储各子串 最后求size的大小
set存不用担心重复

代码

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

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String string = in.nextLine().trim();
        System.out.println(numOfSubString(string));
    }

    private static int numOfSubString(String string) {
        int index = 0;
        Set<String> set = new HashSet<>();
        for(int i = 0;i<string.length();i++) {
            index = string.charAt(index)==string.charAt(i)?index:i;
            set.add(string.substring(index, i+1));
        }
        return set.size();
    }
}