CF 1465D Grime Zoo

https://codeforces.ml/contest/1465/problem/D

参考：https://www.cnblogs.com/JDFZ-ZZ/p/14171488.html  &  https://blog.csdn.net/weixin_43918473/article/details/111879784

分别从前往后预处理前面的1的数量，从后往前预处理后面0的数量。

通过减法可以获得相应0和1的数量，对于‘？’正序预处理将其视为0，逆序预处理将其视为1。

对于某一个位置，我们先求出外部结果，也即pre[i] + suf[i + 1]

然后是二者组间数量，有两部分，也即prec[i] * sufc[i + 1] * y和(i - prec[i]) * (len - i - sufc[i + 1]) * x的和

扫一遍求最小值。

看了半天才看懂...我好笨555

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
char s[N];
ll x, y; 
ll prec[N], sufc[N], pre[N], suf[N];
//前1的个数，后0的个数 
int main(){
    scanf("%s", s + 1);
    scanf("%lld %lld",&x, &y);
    int len = strlen(s + 1);

    if(x > y){
        swap(x, y);
        for(int i = 1 ; i <= len ; i++){
            if(s[i] == '0')    s[i] = '1';
            else if(s[i] == '1')    s[i] = '0';
        }
        //reverse(s.begin(), s.end());
    }
    
    ll res = (1ll << 61);
    for(int i = 1 ; i <= len ; i++){
        if(s[i] == '1'){
            prec[i] = prec[i - 1] + 1;
            pre[i] = pre[i - 1] + 1ll * (i - prec[i]) * x;
        }else{
            prec[i] = prec[i - 1];
            pre[i] = pre[i - 1] + prec[i] * y; 
        }
    }
    
    for(int i = len ; i >= 1 ; i--){
        if(s[i] == '0'){
            sufc[i] = sufc[i + 1] + 1;
            suf[i] = suf[i + 1] + (len - i + 1 - sufc[i]) * x;
        }else{
            sufc[i] = sufc[i + 1];
            suf[i] = suf[i + 1] + (sufc[i] * y);
        }
    }
    for(int i = 0 ; i <= len ; i++){
        res = min(res, pre[i] + suf[i + 1] + prec[i] * sufc[i + 1] * y + (ll)(i - prec[i]) * (len - i - sufc[i + 1]) * x);
        //             前预处理+后预处理组内数量 + 前面1的数量*后面0的数量*y + 前0数*后1数*x 
    }
    printf("%lld
", res);
    
    return 0;
}