枚举,计算贡献
题目描述
对于一个字符串 SS,我们定义 SS 的分值 f(S)f(S) 为 SS 中出现的不同的字符个数。例如 f(“aba”)=2,f(“abc”)=3,f(“aaa”)=1f(“aba”)=2,f(“abc”)=3,f(“aaa”)=1。
现在给定一个字符串 S[0...n−1]S[0...n−1](长度为 nn),请你计算对于所有 SS 的非空子串 S[i...j](0≤i≤j<n)S[i...j](0≤i≤j<n),f(S[i...j])f(S[i...j]) 的和是多少。
输入描述
输入一行包含一个由小写字母组成的字符串 SS。
其中,1≤n≤1051≤n≤105。
输出描述
输出一个整数表示答案。
输入输出样例
示例 1
输入
ababc
输出
28
思路:首先,我们需要知道包含某个第i个字母子串的个数(该字母为第一次出现的字母子串的个数),
对于任意一个在字符串中的字母,所有包含这个字母子串的个数算法(该字母为第一次出现的字母子串的个数):字母初始时“上一次“”出现的位置设为-1,(本次出现的位置-上次出现的位置)*(将来最近一次出现的位置-本次出现的位置)
例如ababc中
第二个a
ba bab bac
a ab abc
2*3
第2个b
它的子串个数可以是
ab abc
b bc
2*2
其次:为什么要计算字母为第一次出现的字母子串的个数呢?
因为题目中说出现不同子串的个数,一个子串中如果出现两次相同字母,那么它只计算一次,它的贡献就为一。
例如:
对于这样的子串 上面例子中加粗的子串ab,遍历到a的时候a对这个子串的贡献值为1,遍历到b的时候b对这个子串的贡献为1,所以这个子串ab包含的字符串数为2
实现代码:
import java.util.*;
// 1:无需package
// 2: 类名必须Main, 不可修改
public class Main {
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
//在此输入您的代码...
int[] arr=new int[26];
long sum=0;
for(int i=0;i<26;i++){
arr[i]=-1;
}
char[] str=scan.next().toCharArray();
int len=str.length;
for(int i=0;i<len;i++){
sum+=(long)(i-arr[str[i]-'a'])*(len-i);
arr[str[i]-'a']=i;
}
System.out.println(sum);
scan.close();
}
}