题解 | #01串的价值#

/** 解题思路：
 * 计算1和0各自数目，假设1更多有sum次，找出1的最左和最右，求和公式1+到sum，然后左边右边都是0，按规则加上。
 */
#include <bits/stdc++.h>

using namespace std;

int val_of_len(int len)
{
    return (len * (1 + len)) / 2;
}
int main()
{
    int t, max_val;
    string str;
    int count[2] = {0, 0};

    cin >> t >> str;
    // cout << str << endl;

    for (int i = 0; i < str.length(); ++i)
    {
        if (str[i] == '0')
        {
            count[0]++;
        }
        else if (str[i] == '1')
        {
            count[1]++;
        }
    }
    // if (count[0] > count[1])
    // {
    //     max_val = val_of_len(count[0]) + val_of_len(str.find_first_of('0')) + val_of_len(str.length() - 1 - str.find_last_of('0'));
    // }
    // else if (count[1] > count[0])
    //     max_val = val_of_len(count[1]) + val_of_len(str.find_first_of('1')) + val_of_len(str.length() - 1 - str.find_last_of('1'));
    // else
    // {
    //     // cout << str.find_first_of('0') << endl;
    //     // cout << str.find_last_of('0') << endl;
    //     // cout << str.find_first_of('1') << endl;
    //     // cout << str.find_last_of('1') << endl;
    //     max_val = max(val_of_len(count[0]) + val_of_len(str.find_first_of('0')) + val_of_len(str.length() - 1 - str.find_last_of('0')),
    //                   val_of_len(count[1]) + val_of_len(str.find_first_of('1')) + val_of_len(str.length() - 1 - str.find_last_of('1')));
    // }
    max_val = max(val_of_len(count[0]) + val_of_len(str.find_first_of('0')) + val_of_len(str.length() - 1 - str.find_last_of('0')), val_of_len(count[1]) + val_of_len(str.find_first_of('1')) + val_of_len(str.length() - 1 - str.find_last_of('1')));
    cout << max_val << endl;
    return 0;
}

/* 根据牛友指出的错误进行了修正的版本 */