AtCoder Beginner Contest 166 F.Three Variables Game(思维题/分类讨论)

题目

开始给定三个数a,b,c(0<=a,b,c<=1e9)，

以下输入n(1<=n<=1e5)个串，第i个串只可能是“AB”、“AC”、“BC”中的一种，

对于串AB，代表你可以对a加1对b减1，或者对a减1对b加1，其余串同理

在处理串的过程中，a,b,c的值应始终保持非负，

如果不能保证输出No，否则输出Yes并且以下n行输出第i次是对哪个值加1的

思路来源

官方题解

题解

面向题解编程，

如果a+b+c=0，由于至少一次操作，显然No

如果a+b+c=1，0 0 1的形式，相当于两数互换，所有操作都唯一，模拟出Yes或No即可

如果a+b+c>=2，如果第一步是两个0的值，显然No，否则一定是Yes

 

如果有两个非0的值，则其最坏情况是1 1 0，

1 1 0可以转移到1 1 0或2 0 0，但是2 0 0后续如果询问两个0，可能没解

但是我们可以考虑相邻的两个询问，期间必有一个字母相交，

如1 1 0时，先询问AB，再询问BC，可以先令B加1，再令B减1，就又回到了1 0 1的情形

其余情况，如果唯一就唯一模拟，如果非上述情况就可以随便选了

 

还见到了一种dp[N][3][3][3]的dp[i][j][k][l]的做法，代表第i轮a为j，b为k，c为l是否有解，

dp[N][2][2][2]一定有解，所以初始情况abc和222取个min，然后开始dp即可，把2视作>=2的状态

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,a,b,c,op[N],ans[N],cnt;
char s[5];
int g(){
    if(strcmp(s,"AB")==0)return 1;
    if(strcmp(s,"AC")==0)return 2;
    if(strcmp(s,"BC")==0)return 3;
}
bool solve(){
    if(a+b+c==0)return 0;
    if(a+b+c==1){
        for(int i=1;i<=n;++i){
            if(op[i]==1){
                if(a==0 && b==0)return 0;
                if(a==0)ans[++cnt]=0,a++,b--;
                else ans[++cnt]=1,b++,a--;
            }
            else if(op[i]==2){
                if(a==0 && c==0)return 0;
                if(a==0)ans[++cnt]=0,a++,c--;
                else ans[++cnt]=2,c++,a--;
            }
            else if(op[i]==3){
                if(b==0 && c==0)return 0;
                if(b==0)ans[++cnt]=1,b++,c--;
                else ans[++cnt]=2,c++,b--;
            }
        }
        return 1;
    }
    if(a+b+c>=2){
        for(int i=1;i<=n;++i){
            if(op[i]==1){
                if(a==0 && b==0)return 0;
                if(a==1 && b==1 && i+1<=n && op[i+1]!=op[i]){
                    if(op[i+1]==2){
                        ans[++cnt]=0,a++,b--;
                    }
                    else if(op[i+1]==3){
                        ans[++cnt]=1,b++,a--;
                    }
                }
                else{
                    if(a<b)ans[++cnt]=0,a++,b--;
                    else ans[++cnt]=1,b++,a--;
                }
            }
            else if(op[i]==2){
                if(a==0 && c==0)return 0;
                if(a==1 && c==1 && i+1<=n && op[i+1]!=op[i]){
                    if(op[i+1]==1){
                        ans[++cnt]=0,a++,c--;
                    }
                    else if(op[i+1]==3){
                        ans[++cnt]=2,c++,a--;
                    }
                }
                else{
                    if(a<c)ans[++cnt]=0,a++,c--;
                    else ans[++cnt]=2,c++,a--;
                }
            }
            else if(op[i]==3){
                if(b==0 && c==0)return 0;
                if(b==1 && c==1 && i+1<=n && op[i+1]!=op[i]){
                    if(op[i+1]==1){
                        ans[++cnt]=1,b++,c--;
                    }
                    else if(op[i+1]==2){
                        ans[++cnt]=2,c++,b--;
                    }
                }
                else{
                    if(b<c)ans[++cnt]=1,b++,c--;
                    else ans[++cnt]=2,c++,b--;
                }
            }
        }
        return 1;
    }
}
int main(){
    scanf("%d%d%d%d",&n,&a,&b,&c);
    for(int i=1;i<=n;++i){
        scanf("%s",s);
        op[i]=g();
    }
    if(!solve())puts("No");
    else{
        puts("Yes");
        for(int i=1;i<=n;++i){
            printf("%c\n",ans[i]+'A');
        }
    }
    return 0;
}