【蓝桥杯VIP】试题 历届试题 子串分值（满分 Python解法+Java解法）

【蓝桥杯VIP】试题 历届试题 子串分值（满分 Python解法+Java解法）

资源限制
时间限制：1.0s 内存限制：256.0MB

问题描述
对于一个字符串 S ，我们定义 S 的分值 f(S) 为 S 中恰好出现一次的字符个数。例如 f(“aba”)=1，f(“abc”)=3, f(“aaa”)=0。
现在给定一个字符串 S[0…n-1]（长度为 n），请你计算对于所有 S 的非空子串 Si…j，f(S[i…j])的和是多少。

输入格式
输入一行包含一个由小写字母组成的字符串 。

输出格式
输出一个整数表示答案。

样例输入
ababc

样例输出
21

样例说明
子串 f值
a 1
ab 2
aba 1
abab 0
ababc 1
b 1
ba 2
bab 1
babc 2
a 1
ab 2
abc 3
b 1
bc 2
c 1

评测用例规模与约定
对于 20% 的评测用例，1<=n<=10 ；
对于 40% 的评测用例，1<=n<=100 ；
对于 50% 的评测用例，1<=n<=1000 ；
对于 60% 的评测用例，1<=n<=10000 ；
对于所有评测用例，1<=n<=100000 。

思路：

用 pre[i] 记录第 i 位置上的字母上一次出现的位置，用 next[i] 记录第 i 位置上的字母下一次出现的位置；
那么往左最多能延伸到 pre[i] + 1，其到第 i 个字母一共有 i - pre[i] 个字母；
同理往右最多能延伸到 next[i] - 1，其到第 i 个字母一共有 next[i] - i 个字母；
二者相乘，就是该 i 位置上的字母被不同子串所包含的总次数。
具体可以结合上述输入输出例子思考。

Python解法：

# 100分
list1=list(input())
pre=[-1 for i in range(len(list1))]               # i位置元素的前一个相同元素位置
next=[len(list1) for i in range(len(list1))]      # i位置元素的后一个相同元素位置
count=0

for i in range(1,len(list1)):
    for j in range(i-1,-1,-1):
        if list1[j]==list1[i]:
            pre[i]=j
            break

for i in range(0,len(list1)-1):
    for j in range(i+1,len(list1)):
        if list1[j]==list1[i]:
            next[i]=j
            break

for i in range(len(list1)):
    count+=(i-pre[i])*(next[i]-i)

print(count)

Java解法：

// 100分
import java.util.Arrays;
import java.util.Scanner;

public class Main{
    public static void main(String[] args){

        Scanner scan=new Scanner(System.in);
        String[] arr1=scan.nextLine().split("");
        int count=0;

        int[] pre=new int[arr1.length];
        int[] next=new int[arr1.length];

        Arrays.fill(pre,-1);
        Arrays.fill(next,arr1.length);

        for(int i=1;i<arr1.length;i++){
            for(int j=i-1;j>=0;j--){
                if(arr1[i].equals(arr1[j])){
                    pre[i]=j;
                    break;
                }
            }
        }

        for(int i=0;i<arr1.length-1;i++){
            for(int j=i+1;j<arr1.length;j++){
                if(arr1[i].equals(arr1[j])){
                    next[i]=j;
                    break;
                }
            }
        }

        for(int i=0;i<arr1.length;i++){
            count+=(i-pre[i])*(next[i]-i);
        }

        System.out.println(count);


    }
}