Acwing.4261 孤独的照片(贡献法）

题目

Farmer John 最近购入了 N
头新的奶牛，每头奶牛的品种是更赛牛（Guernsey）或荷斯坦牛（Holstein）之一。

奶牛目前排成一排，Farmer John 想要为每个连续不少于三头奶牛的序列拍摄一张照片。

然而，他不想拍摄这样的照片，其中只有一头牛的品种是更赛牛，或者只有一头牛的品种是荷斯坦牛——他认为这头奇特的牛会感到孤立和不自然。

在为每个连续不少于三头奶牛的序列拍摄了一张照片后，他把所有「孤独的」照片，即其中只有一头更赛牛或荷斯坦奶牛的照片，都扔掉了。

给定奶牛的排列方式，请帮助 Farmer John 求出他会扔掉多少张孤独的照片。

如果两张照片以不同位置的奶牛开始或结束，则认为它们是不同的。

输入格式

输入的第一行包含 N。

输入的第二行包含一个长为 N的字符串。如果队伍中的第 i头奶牛是更赛牛，则字符串的第 i个字符为 G。否则，第 i头奶牛是荷斯坦牛，该字符为 H。

输出格式

输出 Farmer John 会扔掉的孤独的照片数量。

数据范围

3≤N≤5×10⁵

输入样例：

5
GHGHG

输出样例：

样例解释

这个例子中的每一个长为 3的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片，并会被 Farmer John 扔掉。

所有更长的子串（GHGH、HGHG 和 GHGHG）都可以被接受。

题解

import java.util.Scanner;

/**
 * @author akuya
 * @create 2024-03-13-19:13
 */
public class Main {
    static int N=500010;
    static int n;
    static int l[]=new int[N];
    static int r[]=new int[N];
    static long res=0;
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        n=scanner.nextInt();
        String s=scanner.next();
        char c[]=s.toCharArray();

        for(int i=0,sg=0,sh=0;i<n;i++){
            if(c[i]=='G'){
                l[i]=sh;
                sh=0;
                sg++;
            }else{
                l[i]=sg;
                sh++;
                sg=0;
            }
        }

        for(int i=n-1,sg=0,sh=0;i>=0;i--){
            if(c[i]=='G'){
                r[i]=sh;
                sh=0;
                sg++;
            }else{
                r[i]=sg;
                sh++;
                sg=0;
            }
        }

        for(int i=0;i<n;i++){
            res+=(long)l[i]*r[i]+Math.max(l[i]-1,0)+Math.max(r[i]-1,0);
        }


        System.out.println(res);
    }
}