面试题之“度度熊” - 计算字符串中，子字符串的种类

题目描述 
      度度熊收到了一个只有小写字母的字符串S，他对S的子串产生了兴趣，S的子串为S中任意连续的一段。他发现，一些子串只由一种字母构成，他想知道在S中一共有多少种这样的子串。 
例如在串”aaabbaa”中，度度熊想找的子串有”a”,”aa”,”aaa”,”b”,”bb”五种。 

（本题只考虑子串的种数，相同的子串在多个位置出现只算一次）

      看到这个题目的时候，第一反应就是用HashMap，为什么？因为key不允许重复，另外不论是key还是value都有方法直接提供其集合。所以只需要将所有的子串作为key保存到HashMap中即可，但是由于受到文章解题思路的影响，所以先采用文章解题思路的：（文章地址：http://blog.csdn.net/summer_pug/article/details/53262629）

思路见代码方法注释：

import java.util.HashMap;
import java.util.Map;

/**
 * 计算字符串中，子字符串的种类：
 * 例如在串"aaabbaa"中，要找的子串有"a","aa","aaa","b","bb"五种。 
 * @author xiaowei 2017年11月16日 上午8:54:45
 */
public class SubOfStringNum {

    public static void main(String[] args) {
        SubOfStringNum sosn = new SubOfStringNum();
        System.out.println(sosn.numOfSubString("1222bbt65ss22tt222222"));
        System.out.println(sosn.numOfSubString1("1222bbt65ss22tt222222"));
    }

    /**
     * 最长子串包括了所有非最长子串的情况，如aaa包含了a，aa，aaa三种情况，
     * 即子串的种数等于最长子串的长度，因此对于每个小写字母，找出最长连续子串，相加即得结果。
     */
    public int numOfSubString(String str) {

        if (str == null || str.length() == 0) {
            return 0;
        }
        Map<String, Integer> map = new HashMap<>();
        char[] ch = str.toCharArray();
        int count = 1;
        for (int i = 0; i < ch.length; i++) {
            String s = String.valueOf(ch[i]);
            if (i == 0) {
                map.put(s, 1);
                continue;
            }
            if (ch[i] == ch[i - 1]) {
                ++count;
                // 避免不连续的重复字符串覆盖前面
                if (count > map.get(s)) {
                    map.put(s, count);
                }
            } else if (map.get(s) != null) {
                count = 1;
            } else {
                count = 1;
                map.put(s, 1);
            }
        }
        int result = 0;
        for (int i : map.values()) {
            result += i;
        }
        return result;
    }

    /**
     * 最初的想法实现功能，将所有连续且相等的子串作为key添加至HashMap中，最后计算key的数量即可。
     */
    public int numOfSubString1(String str) {

        if (str == null || str.length() == 0) {
            return 0;
        }
        Map<String, String> map = new HashMap<>();
        char[] ch = str.toCharArray();
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < ch.length; i++) {
            if (i == 0) {
                sb.append(ch[0]);
                map.put(sb.toString(), null);
                continue;
            }
            if (ch[i] != ch[i - 1]) {
                sb = new StringBuffer();
            }
            sb.append(ch[i]);
            map.put(sb.toString(), null);
        }
        return map.keySet().size();
    }
}

输出结果：

15
15