UVa 12545 - Bits Equalizer

链接：

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3990

 

题意：

输入两个等长（长度不超过100）的串S和T，其中S包含字符0, 1, ?，但T只包含0和1。
你的任务是用尽量少的步数把S变成T。每步有3种操作：把S中的0变成1；把S中的“?”变成0或者1；
交换S中任意两个字符。如果S不能变为T则输出-1。

 

分析：

既然S中的0可以变为1，'?'可以变为0或1，那么S不能变为T的唯一情况是S中1的个数比T多。
可以把解题过程分为几个阶段：
阶段一：分别统计S和T中1的个数，判断是否有解。
阶段二：如果S中某个位置是'?'且T中对应的位置是1，S中1的个数又比T少，则先把该位置变为1，以减少交换次数。
阶段三：将S中剩余的'?'变为0或1，只要能让S和T中1的个数相等，可以随意变换。
阶段四：判断是否需要把S中的某些0变为1，以及需要交换的次数。

 

代码：

#include <cstdio>
#include <cstring>

const int UP = 100 + 5;
char s[UP], t[UP];

int judge(){
    int sum = 0; //1的个数
    for(int i = 0; s[i]; i++) sum += (t[i]=='1'), sum -= (s[i]=='1');
    if(sum < 0) return -1; //阶段一：判断是否有解

    int change = 0, remain = strlen(s); //变换的次数，对应位置不相等的总个数
    for(int i = 0; s[i]; i++) if(s[i] == '?' && sum && t[i] == '1'){
        s[i] = '1';
        change++;
        sum--;
    } //阶段二：优先把S中的某些位置变成1

    for(int i = 0; s[i]; i++){ //阶段三：将S中剩余的'?'变为0或1
        if(s[i] == '?'){
            change++;
            if(sum) s[i] = '1', sum--;
            else s[i] = '0';
            if(s[i] == t[i]) remain--;
        }
        else if(s[i] == t[i]) remain--;
    }
    change += sum; //阶段四：把S中的某些0变为1，以及累加交换的次数
    remain -= sum;
    return change + remain / 2;
}

int main(){
    int T;
    scanf("%d", &T);
    for(int cases = 1; cases <= T; cases++){
        scanf("%s%s", s, t);
        printf("Case %d: %d\n", cases, judge());
    }
    return 0;
}