P1098 字符串的展开 （模拟）

题目描述

在初赛普及组的“阅读程序写结果”的问题中，我们曾给出一个字符串展开的例子：如果在输入的字符串中，含有类似于“d-h”或者“4-8”的字串，我们就把它当作一种简写，输出时，用连续递增的字母或数字串替代其中的减号，即，将上面两个子串分别输出为“defgh”和“45678"。在本题中，我们通过增加一些参数的设置，使字符串的展开更为灵活。具体约定如下：

(1) 遇到下面的情况需要做字符串的展开：在输入的字符串中，出现了减号“-”，减号两侧同为小写字母或同为数字，且按照ASCII码的顺序，减号右边的字符严格大于左边的字符。

(2) 参数p_1p1​：展开方式。p_1=1p1​=1时，对于字母子串，填充小写字母；p_1=2p1​=2时，对于字母子串，填充大写字母。这两种情况下数字子串的填充方式相同。p_1=3p1​=3时，不论是字母子串还是数字字串，都用与要填充的字母个数相同的星号“*”来填充。

(3) 参数p_2p2​：填充字符的重复个数。p_2=kp2​=k表示同一个字符要连续填充k个。例如，当p_2=3p2​=3时，子串“d-h”应扩展为“deeefffgggh”。减号两边的字符不变。

(4) 参数p_3p3​：是否改为逆序：p3=1p3=1表示维持原来顺序，p_3=2p3​=2表示采用逆序输出，注意这时候仍然不包括减号两端的字符。例如当p_1=1p1​=1、p_2=2p2​=2、p_3=2p3​=2时，子串“d-h”应扩展为“dggffeeh”。

(5) 如果减号右边的字符恰好是左边字符的后继，只删除中间的减号，例如：“d-e”应输出为“de”，“3-4”应输出为“34”。如果减号右边的字符按照ASCII码的顺序小于或等于左边字符，输出时，要保留中间的减号，例如：“d-d”应输出为“d-d”，“3-1”应输出为“3-1”。

输入格式

共两行。

第11行为用空格隔开的33个正整数，依次表示参数p_1,p_2,p_3p1​,p2​,p3​。

第22行为一行字符串，仅由数字、小写字母和减号“-−”组成。行首和行末均无空格。

输出格式

共一行，为展开后的字符串。

输入输出样例

输入 #1复制

1 2 1
abcs-w1234-9s-4zz

输出 #1复制

abcsttuuvvw1234556677889s-4zz

输入 #2复制

2 3 2
a-d-d

输出 #2复制

aCCCBBBd-d

题解：

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

int p1, p2, p3;

//flag = 1代表数字
string solve(char a, char b, int flag) {
    string ans = "";
    int p, q;
    if(flag == 1) {
        p = a - '0';
        q = b - '0';
    } else if (flag == 2) {
        p = a - 'a';
        q = b - 'a';
    }
    for(int i = p + 1; i <= q - 1; i++) {
        for(int j = 0; j < p2; j++) {
            if(p1 == 3) ans += '*';
            else if(flag == 1) ans += (i + '0');
            else if(flag == 2) {
                if(p1 == 1) ans += char(i + 'a');
                else if(p1 == 2) ans += char(i + 'A');
            }
        }
    }
    if(p3 == 2) reverse(ans.begin(), ans.end());
    return ans;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif // ONLINE_JUDGE
    cin >> p1 >> p2 >> p3;
    string s;
    cin >> s;
    for(int i = 0; i < s.length(); i++) {
        if(s[i] == '-' && i != 0 && i != s.length() - 1) {
            int x = i - 1, y = i + 1;
            if((isdigit(s[x]) && isdigit(s[y])) || (isalpha(s[x]) && isalpha(s[y]))) {
                if(s[x] >= s[y]) cout << "-";
                else {
                    if(s[x] + 1 < s[y]) {
                        string str;
                        if(isdigit(s[x])) str = solve(s[x], s[y], 1);
                        if(isalpha(s[x])) str = solve(s[x], s[y], 2);
                        cout << str;
                    }
                }
            } else {
                cout << '-';
            }
        } else {
            cout << s[i];
        }
    }
    return 0;
}