字形变换-头歌

 将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行Z字形排列。之后,你的输出需要从左往右逐行读取,产生出一个新的字符串

示例 : 输入: s ="QAZWSXEDCRFVTG",numRows = 4 输出:"QETAXDVGZSCFWR"

 A->X->D对应代码中的注释部分

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

char* convert(char* s, int numRows) {
    if (numRows == 1) {
        return s; // 如果行数为1,直接返回原字符串
    }

    int len = strlen(s);
    char* ans = (char*)malloc((len+1)*sizeof(char));
    ans[len] = '\0'; // 新建字符串数组并且初始化为'\0'
    int k = 0;
    int step = (numRows-1)*2; // 每个Z字形的步长
    for (int i = 0; i < numRows; i++) {
        int index = i;
        int diff = (numRows-i-1)*2; 
        /*******Begin********/
       int diff2 = i*2; 

        while(index < len) {
            if(i != 0 && i != numRows-1) { // 非首尾行需要考虑两种跨度
                if(index < len) {
                    ans[k++] = s[index];    //A
                }
                index += diff;
                if(index < len) {
                    ans[k++] = s[index];      //X
                }
                index += diff2;                //D
            } else {                    // 首尾行只需考虑一个跨度
                if(index < len) {
                    ans[k++] = s[index];
                }
                index += step;
            }
        }
    }

    return ans;
}

int main() {
    char s[20];
    scanf("%s", &s);
    int numRows;
    scanf("%d", &numRows);
    printf("%s", convert(s,numRows)); 
    return 0;
}

结果

猜你喜欢

转载自blog.csdn.net/m0_70732442/article/details/134339829