01串的价值

import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息

public class Main {

    public static void main(String[] args) {

        Scanner in = new Scanner(System.in);

        int x = in.nextInt();

        String y = in.next();

        char[]arr = y.toCharArray();

        char head = arr[0], tail = arr[arr.length - 1];//头尾字符

        int sum1 = 0, sum0 = 0;//0 1总数

        int headchar = 0, tailchar = 0;//头部 尾部 字符数量

        int j = 0;//当前位置

        int ans1 = 0, ans0 = 0;//答案

        while (j<arr.length&&arr[j] == head) {

            j++;

            headchar++;

        }

        if(j!=arr.length)j = arr.length - 1;

        else{

            j=-1;

        }

        while (j>=0&&arr[j] == tail) {

            j--;

            tailchar++;

        }

        for (int i = 0; i < arr.length; i++) {

            if (arr[i] == '1')sum1++;

            else sum0++;

        }

        ans0=sum0+sum0*(sum0-1)/2;

        ans1=sum1+sum1*(sum1-1)/2;

        if (head == '0') {

            ans1 += headchar + headchar * (headchar - 1) / 2;

        } else {

            ans0 += headchar + headchar * (headchar - 1) / 2;

        }

        if (tail == '0') {

            ans1 += tailchar + tailchar * (tailchar - 1) / 2;

        } else {

            ans0 += tailchar + tailchar * (tailchar - 1) / 2;

        }

        System.out.print(ans0>ans1?ans0:ans1);

    }

}