【Leetcode】【ZigZag conversion】【Z字形变换】【C++】

最新推荐文章于 2025-09-13 19:30:09 发布

转载最新推荐文章于 2025-09-13 19:30:09 发布 · 87 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/dreamer123/p/9163063.html

文章标签：

#c/c++ #数据结构与算法

本文介绍了如何将字符串以Z字形方式排列成指定行数的方法，并提供了两种实现思路及对应的C++代码示例。一种是通过构建字符数组再逐行读取，另一种则是根据Z字形规律直接计算。

题目：将给定字符串以Z字形排列成给定的行数：如"PAYPALISHIRING" ，numRows=3

P A H N

A P L S I I G

Y I R 之后从左到右排列，逐行读取字符：得到"PAHNAPLSIIGYIR"并返回

思路1：将Z字形字符数组存储到一个字符数组中，然后按行读取字符。
思路2：直接根据规律计算。
代码1：

class Solution {
public:
    string convert(string s, int numRows) {
        int len_s=s.size();
        if(numRows==1)
            return s;
        int numCols=(numRows-1)*len_s/(numRows*2-2)+numRows-1;
        
        
        char res[numRows][numCols];
        for(int i=0;i<numRows;i++)
        {
            for(int j=0;j<numCols;j++)
            {
                res[i][j]=' ';
            }
        }
            
        int cur_row=0;
        int cur_col=0;
        bool down=true;
        for(int i=0;i<len_s;i++)
        {
            if(down)
            {
                res[cur_row][cur_col]=s[i];
                if(cur_row && cur_row==(numRows-1))
                {
                    down=false;
                    cur_row--;
                    cur_col++;
                }
                else
                {
                    cur_row++;
                }
            }
            else
            {
                res[cur_row][cur_col]=s[i];
                if(cur_row==0)
                {
                    down=true;
                    cur_row++;
                }
                else
                {
                    cur_row--;
                    cur_col++;
                }
            }
        }
        numCols=cur_col+1;
        string str="";
        for(int i=0;i<numRows;i++)
        {
            for(int j=0;j<numCols;j++)
            {
                if(res[i][j]!=' ')
                    str+=(res[i][j]);
            }
        }
        return str;
    }
};

代码2：

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows==1)
            return s;
        int n=numRows*2-2;
        int len_s=s.size();
        
        string res="";
        
        for(int i=0;i<numRows;i++)
        {
            int index=i;
            
            while(index<len_s)
            {
                res+=s[index];
                
                if(i!=0 && i!=(numRows-1) && index+n-2*(i%n)<len_s)
                {
                    res+=s[index+n-2*(i%n)];
                }
                
                index=index+n;
            }  
        }
        return res;
        
    }
};