【蓝桥杯 子串分值和】

子串分值和

题目描述

对于一个字符串 S，我们定义 S 的分值 f(S ) 为 S 中出现的不同的字符个 数。例如 f(”aba”) = 2，f(”abc”) = 3, f(”aaa”) = 1。 现在给定一个字符串 S [0…n − 1]（长度为 n），请你计算对于所有 S 的非空 子串 S [i… j](0 ≤ i ≤ j < n)，f(S [i… j]) 的和是多少。
【输入格式】
输入一行包含一个由小写字母组成的字符串 S。
【输出格式】
输出一个整数表示答案。
【样例输入】
ababc
【样例输出】
28

解题思路

考虑每个节点的字母对答案的贡献。对于第i个节点，它能被算作贡献为1的区间有多少个？记nex[i]为与s[i]相同的下一个字母的下标(没有就是n+1)，pre[i]为与s[i]相同的上一个字母的下标(没有就是0)。
统计一下答案：ans+=(i-pre[i])*(nex[i]-i);

上代码

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;

#define N 100050 
#define