编程初学者入门训练-小乐乐与字符串

题目描述

在庆祝祖国母亲70华诞之际，老师给小乐乐出了一个问题。大家都知道China的英文缩写是CHN，那么给你一个字符串s，你需要做的是统计s中子串“CHN”的个数。 子串的定义：存在任意下标a < b < c，那么“s[a]s[b]s[c]”就构成s的一个子串。如“ABC”的子串有“A”、“B”、“C”、“AB”、“AC”、“BC”、“ABC”。

输入描述

输入只包含大写字母的字符串s。(1 ≤ length ≤ 8000)

输出描述

输出一个整数，为字符串s中字串“CHN”的数量。

示例1

输入

CCHNCHN

输出

7

示例2

输入

CCHNCHNCHNCHN

输出

30

代码实现

import java.io.*;

public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        String strs = bf.readLine();
        char[] arr = strs.toCharArray();
        int n=0;
        int c=0;
        long ans = 0;//注意：int范围不够用，将ans类型改为long
        for(int i=0;i<arr.length;i++){
            if(arr[i]=='N'){
                n++;
            }
        }
        for(int i=0;i<arr.length;i++){
            if(arr[i]=='C'){
                c++;
            }else if(arr[i]=='H'){
                ans = ans+c*n;
            }else if(arr[i]=='N'){
                n--;
            }
        }
        System.out.println(ans);
    }
}