JAVA程序设计：统计子串中的唯一字符（LeetCode：828）

我们定义了一个函数 countUniqueChars(s) 来统计字符串 s 中的唯一字符，并返回唯一字符的个数。

例如：s = "LEETCODE" ，则其中 "L", "T","C","O","D" 都是唯一字符，因为它们只出现一次，所以 countUniqueChars(s) = 5 。

本题将会给你一个字符串 s ，我们需要返回 countUniqueChars(t) 的总和，其中 t 是 s 的子字符串。注意，某些子字符串可能是重复的，但你统计时也必须算上这些重复的子字符串（也就是说，你必须统计 s 的所有子字符串中的唯一字符）。

由于答案可能非常大，请将结果 mod 10 ^ 9 + 7 后再返回。

 

示例 1：

输入: "ABC"
输出: 10
解释: 所有可能的子串为："A","B","C","AB","BC" 和 "ABC"。
     其中，每一个子串都由独特字符构成。
     所以其长度总和为：1 + 1 + 1 + 2 + 2 + 3 = 10
示例 2：

输入: "ABA"
输出: 8
解释: 除了 countUniqueChars("ABA") = 1 之外，其余与示例 1 相同。
示例 3：

输入：s = "LEETCODE"
输出：92
 

提示：

0 <= s.length <= 10^4
s 只包含大写英文字符

思路：我们考虑对每个字母分别计数，我采用的方法是用mao进行字母映射，对于每个字母，我们需要预处理出来该字母分别在那些位置出现，该字母能对答案产生贡献的区间数量一定是上一次该字母出现的位置的下一个位置到下一次出现的位置的上一个位置的乘积。

class Solution {
    public int uniqueLetterString(String s) {
    	
    	int ans=0;
    	int len=s.length();
    	Map<Character,List<Integer>> count=new HashMap<>();
    	
    	for(int i=0;i<len;i++) {
    		if(!count.containsKey(s.charAt(i)))
    			count.put(s.charAt(i), new ArrayList<>());
    		count.get(s.charAt(i)).add(i);
    	}
    	
    	for(List<Integer> list : count.values()) {
    		if(list.size()==1) {
    			ans+=(len-list.get(0))*(list.get(0)+1);
    			continue;
    		}
    		for(int i=0;i<list.size();i++) {
    			if(i==0) 
    				ans+=(list.get(i)+1)*(list.get(i+1)-list.get(i));
    			else if(i==list.size()-1)
    				ans+=(len-list.get(i))*(list.get(i)-list.get(i-1));
    			else
    				ans+=(list.get(i)-list.get(i-1))*(list.get(i+1)-list.get(i));
    		}
    	}
    	
    	return ans;
    }
}