LeetCode（6）：Z字形转换

最新推荐文章于 2022-11-20 20:54:49 发布

Rage_

最新推荐文章于 2022-11-20 20:54:49 发布

阅读量210

点赞数

分类专栏： LeetCode

LeetCode 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种将字符串以Z字形排列并按行读取的算法实现。通过分析不同行数下字符串的排列规律，给出了C++代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文转自：https://www.cnblogs.com/ariel-dreamland/p/8690523.html

题目描述：

将字符串 "PAYPALISHIRING" 以Z字形排列成给定的行数：（下面这样的形状）

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

之后按逐行顺序依次排列："PAHNAPLSIIGYIR"

实现一个将字符串进行指定行数的转换的函数:

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3) 应当返回 "PAHNAPLSIIGYIR" 。

题意解释：

比如有一个字符串 “0123456789ABCDEF”，转为zigzag

当 n = 2 时：

0 2 4 6 8 A C E

1 3 5 7 9 B D F

当 n = 3 时：

0 4 8 C

1 3 5 7 9 B D F

2 6 A E

当 n = 4 时：

0 6 C

1 5 7 B D

2 4 8 A E

3 9 F

我们发现，除了第一行和最后一行没有中间形成之字型的数字外，其他都有，而首位两行中相邻两个元素的index之差跟行数是相关的，为 2*nRows - 2, 根据这个特点，我们可以按顺序找到所有的黑色元素在元字符串的位置，将他们按顺序加到新字符串里面。对于红色元素出现的位置也是有规律的，每个红色元素的位置为 j + 2*nRows-2 - 2*i, 其中，j为前一个黑色元素的列数，i为当前行数。比如当n = 4中的那个红色5，它的位置为 1 + 2*4-2 - 2*1 = 5，为原字符串的正确位置。当我们知道所有黑色元素和红色元素位置的正确算法，我们就可以一次性的把它们按顺序都加到新的字符串里面。代码如下：

class Solution {
public:
    string convert(string s, int numRows) {
        int len = s.length();
        int d= 2*numRows-2; //两整列之间的差 也就是等差数列中的d  
        string result = "";
        
        if(len == 0 || numRows == 0 || numRows == 1)
        	return s;
        	
        for(int i=0;i<numRows;i++){
        	for(int j=i;j<len;j+=d){
        		result += s[j];
        	
        		if(i != 0 && i!=numRows-1 && j-2*i+d<len){
        			result += s[j-2*i+d];
				}
        		
			}
		}
        return result;
    }
};