【Leetcode】[6] ZigZag Conversion

最新推荐文章于 2022-07-13 23:27:43 发布

原创最新推荐文章于 2022-07-13 23:27:43 发布 · 195 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种将字符串以Z字形方式写入指定行数并按行读取的算法实现。通过创建二维数组存储输入字符串，实现了从上到下再到上的Z字形排列，并最终输出转换后的字符串。

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P   A   H   N
A P L S I I G
Y   I   R

And then read line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string s, int numRows);

Example 1:

Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"

Example 2:

Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"
Explanation:

P     I    N
A   L S  I G
Y A   H R
P     I

对于这个题目，我做法比较复杂。建立了一个二维数组，将输入的字符串首先以Z字形存储起来，最后再拼接为字符串输出。

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

int main()
{
    string s;
    int numRows;
    cin>>s>>numRows;
    int len = s.length();
    //加1，考虑输入s为空的情况
    char str[numRows][len+1];
    int strFlag[numRows][len+1];
    memset(strFlag,0,sizeof(strFlag));
    //行
    int row=0;
    //列
    int col = 0;
    //flag为true表示row++，false表示row--
    bool flag = true;
    //flag1为true表示col不加，false表示自加
    bool flag1 = true;
    //j通过自加，控制flag1的变化
    int j=0;
    for(int i=0;i<s.length();i++){
        str[row][col] = s[i];
        strFlag[row][col] = 1;
        //输入1行情况下，row为0，col一直自加
        if(numRows==1){
            row = 0;
            col++;
        }else{
            if(row==0){
                row++;
                flag = true;
            }else if(row==numRows-1){
                row--;
                flag = false;
            }else if(flag){
                row++;
            }else{
                row--;
            }
            j++;
            if(j>numRows-1){
                j=1;
                if(flag1){flag1 = false;
                }else{flag1 = true;}
            }
            if(!flag1){col++;}
        }
    }
    string sstr;
    for(int t=0;t<numRows;t++){
        for(int h=0;h<=col;h++){
            if(strFlag[t][h]==1){
               cout<<str[t][h]<<" ";
               sstr = sstr+str[t][h];
            }else{
                cout<<" "<<" ";
            }
        }
        cout<<endl;
    }
    cout <<endl<< sstr << endl;
    return 0;
}