洛谷div2【XR-4】歌唱比赛

本人水平有限，题解不到为处，请多多谅解

本蒟蒻谢谢大家观看

题目：传送门

题目有地方没有说清楚：

1：即小X的各个位数的点赞数之和==小Y的各个位数的点赞数之和（就是这个毒瘤导致我只有11分）

2：出现-1的情况，从左往右出现第一个Z时，若之后只要有一个为X或Y，那么就无法得出，输出-1

模拟即可

code:

#include<bits/stdc++.h>
#pragma GCC optimize(3)
const int N=1e6+100;
using namespace std;
string ch;
int len;
int k[N],h[N],tot,cnt,wz[N];
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    return x*f;
}
inline void write(int x){
     char F[200];
     int tmp=x>0?x:-x ;
     if(x<0)putchar('-') ;
     int cnt=0 ;
        while(tmp>0)
        {
            F[cnt++]=tmp%10+'0';
            tmp/=10;
        }
        while(cnt>0)putchar(F[--cnt]) ;
}
void solve1(){
    if(ch[0]=='X'){
        cout<<3<<endl;
        cout<<2<<endl;
        return ;
    }
    if(ch[0]=='Y'){
        cout<<2<<endl;
        cout<<3<<endl;
        return ;
    }
    if(ch[0]=='Z'){
        cout<<1<<endl;
        cout<<1<<endl;
        return ;
    } 
    return ;
}
void solve2(){
    for(int i=0;i<len;i++){
        if(ch[i]=='X'){
            k[++tot]=9;
            h[++cnt]=0;
        }
        if(ch[i]=='Y'){
            k[++tot]=0;
            h[++cnt]=9;
        }
        if(ch[i]=='Z'){
            k[++tot]=1;
            h[++cnt]=1;
            wz[i]=1;
        }
    }
    for(int i=0;i<len-1;i++){
        if(wz[i]!=0){
        if(wz[i]!=wz[i+1]){
            printf("-1
");
            return ;
        }
        //cout<<"wz= "<<wz[i]<<endl;
    }
    }
    for(int i=1;i<=tot;i++){
        printf("%d",k[i]);
    }
    cout<<endl;
    for(int i=1;i<=cnt;i++){
        printf("%d",h[i]);
    }
    return ;
}
int main()
{
    cin>>ch;
    len=ch.size();
    //cout<<"len= "<<ch[1]<<endl;
    if(len==1)solve1();
    else solve2();
    return 0;
}