Leetcode-06-Z字形变换

发布于:2024-03-21 ⋅ 阅读:(63) ⋅ 点赞:(0)

将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:

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

之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"

请你实现这个将字符串进行指定行数变换的函数:

string convert(string s, int numRows);

示例 1:

输入:s = "PAYPALISHIRING", numRows = 3
输出:"PAHNAPLSIIGYIR"

示例 2:

输入:s = "PAYPALISHIRING", numRows = 4
输出:"PINALSIGYAHRPI"
解释:
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3:

输入:s = "A", numRows = 1
输出:"A"

提示:

  • 1 <= s.length <= 1000
  • s 由英文字母(小写和大写)、',' 和 '.' 组成
  • 1 <= numRows <= 1000

解题思路:创建二维数组

class Solution {
    public String convert(String s, int numRows) {
        if(numRows ==1) return s;
		int strLen = s.length();
		char[][] charArray = new char[numRows][strLen];
		int row=0,col=0,idx=numRows-2;
		for(int i=0;i<strLen;i++) {
            int num = (col+numRows-1)%(numRows-1);
			if( num==0 ) {
				charArray[row][col] = s.charAt(i);
				if(row<numRows-1) {
					row++;
				}else {
					row = 0;
					col++;
				}
			}else {
				if(idx<=0||idx>numRows-2) {
					idx = numRows -2;
				}
				charArray[idx--][col] = s.charAt(i);
				col++;
			}
			
		}
		String rstStr = "";
		for(int i=0;i<charArray.length;i++) {
			for(int j=0;j<charArray[i].length;j++) {
				if(charArray[i][j]!='\0') {
					rstStr+=charArray[i][j];
				}
			}
		}

		return rstStr;
    }
}


网站公告

今日签到

点亮在社区的每一天
去签到