【NC16644】字符串的展开

题目

字符串的展开

模拟题，比较繁琐，考验耐心和细心

思路

这里将字符串处理作为一个函数提取出来（虽然这个函数特别长），具体的思路就是题目所给的思路，没有什么特别的，就是动手模拟而已。
代码里有注释，配合着看更容易理清思路。

代码

#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define N 105
// 题目字符串长度不超过100，最多包含33个“a-z”
// 取 p2 最大值 8，则展开后的字符串最大长度大概为
// 33 x 24 x 8 = 6336，开6500的空间足矣
#define MAX_LEN 6500

char* op(int p1, int p2, int p3, char* s) {
    char* res = (char*)malloc(MAX_LEN * sizeof(char));
    memset(res, 0, MAX_LEN);
    int n = strlen(s), len = 0, i = 0, j = 0, k = 0;
    char a = 0, b = 0;
    for (i = 0; s[i]; i++) {
        if (s[i] != '-') {
            // 如果不是减号，则按原样输出
            res[len++] = s[i];
            continue;
        }
        // 如果减号是字符串的开头和结尾，则不处理，原样输出
        if (i == 0 || i == n - 1) {
            res[len++] = s[i];
            continue;
        }
        // 否则加以处理
        // 首先得到减号两边的字符
        a = s[i - 1], b = s[i + 1];
        // 再判断是同为数字还是同为字母
        if (isalpha(a) && isalpha(b)) {
            // 如果同为字母
            // 则判断左边和右边的字母哪个更大
            if (a >= b) {
                // 左边字母更大或等于，原样输出
                res[len++] = s[i];
            } else {
                // 根据参数展开
                if (p1 == 1) {
                    // 填充小写字母
                    if (p3 == 1) {
                        // 正序
                        for (j = a + 1; j < b; j++) {
                            for (k = 0; k < p2; k++) {
                                res[len++] = j;
                            }
                        }
                    } else if (p3 == 2) {
                        // 逆序
                        for (j = b - 1; j > a; j--) {
                            for (k = 0; k < p2; k++) {
                                res[len++] = j;
                            }
                        }
                    }
                } else if (p1 == 2) {
                    // 填充大写字母
                    if (p3 == 1) {
                        // 正序
                        for (j = a - 'a' + 'A' + 1; j < b - 'a' + 'A'; j++) {
                            for (k = 0; k < p2; k++) {
                                res[len++] = j;
                            }
                        }
                    } else if (p3 == 2) {
                        // 逆序
                        for (j = b - 'a' + 'A' - 1; j > a - 'a' + 'A'; j--) {
                            for (k = 0; k < p2; k++) {
                                res[len++] = j;
                            }
                        }
                    }
                } else if (p1 == 3) {
                    // 填充星号，此时不管顺序逆序
                    for (j = a + 1; j < b; j++) {
                        for (k = 0; k < p2; k++) {
                            res[len++] = '*';
                        }
                    }
                }
            }
        } else if (isdigit(a) && isdigit(b)) {
            // 如果同为数字
            // 则判断左边和右边的数字哪个更大
            if (a >= b) {
                // 左边数字更大或等于，原样输出
                res[len++] = s[i];
            } else {
                // 根据参数展开
                if (p1 == 1 || p1 == 2) {
                    if (p3 == 1) {
                        // 正序
                        for (j = a + 1; j < b; j++) {
                            for (k = 0; k < p2; k++) {
                                res[len++] = j;
                            }
                        }
                    } else if (p3 == 2) {
                        // 逆序
                        for (j = b - 1; j > a; j--) {
                            for (k = 0; k < p2; k++) {
                                res[len++] = j;
                            }
                        }
                    }
                } else if (p1 == 3) {
                    // 填充星号，此时不管顺序逆序
                    for (j = a + 1; j < b; j++) {
                        for (k = 0; k < p2; k++) {
                            res[len++] = '*';
                        }
                    }
                }
            }
        } else {
            // 不同为数字或者字母，则不管，原样输出
            res[len++] = s[i];
        }
    }
    return res;
}

int main(void) {
    int p1 = 0, p2 = 0, p3 = 0;
    char s[N];
    scanf("%d%d%d%s", &p1, &p2, &p3, s);
    char* ans = op(p1, p2, p3, s);
    printf("%s\n", ans);
    return 0;
}