2022GPLT团体程序设计天梯赛L1-086 斯德哥尔摩火车上的题

上图是新浪微博上的一则趣闻，是瑞典斯德哥尔摩火车上的一道题，看上去是段伪代码：

s = ''
a = '1112031584'
for (i = 1; i < length(a); i++) {
  if (a[i] % 2 == a[i-1] % 2) {
    s += max(a[i], a[i-1])
  }
}
goto_url('www.multisoft.se/' + s)

其中字符串的 + 操作是连接两个字符串的意思。所以这道题其实是让大家访问网站 www.multisoft.se/112358（注意：比赛中千万不要访问这个网址！！！）。

当然，能通过上述算法得到 112358 的原始字符串 a 是不唯一的。本题就请你判断，两个给定的原始字符串，能否通过上述算法得到相同的输出？

输入格式：

输入为两行仅由数字组成的非空字符串，长度均不超过 104，以回车结束。

输出格式：

对两个字符串分别采用上述斯德哥尔摩火车上的算法进行处理。如果两个结果是一样的，则在一行中输出那个结果；否则分别输出各自对应的处理结果，每个占一行。题目保证输出结果不为空。

输入样例 1：

1112031584
011102315849

输出样例 1：

112358

输入样例 2：

111203158412334
12341112031584

输出样例 2：

1123583
112358

#include <iostream>
#include <string>
using namespace std;

string s1, s2;

string solve(string a) {
    string s = "";
    int len = a.length();
    for (int i = 1; i < len; i++) {
        if (a[i]%2 == a[i-1]%2) s += max(a[i], a[i-1]);
    }
    return s;
}

int main()
{
    cin >> s1 >> s2;
    s1 = solve(s1);
    s2 = solve(s2);
    if (s1 == s2) cout << s1;
    else cout << s1 << endl << s2;
    return 0;
}