【中等】【LeetCode刷题笔记(九)】之 6. Z字形变换

本文章由公号【开发小鸽】发布!欢迎关注!!!


老规矩–妹妹镇楼:

一. 题目

(一)概述

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

       比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:

L   C   I   R
E T O E S I I G
E   D   H   N

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

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

string convert(string s, int numRows);

(二)示例

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:

L     D     R
E   O E   I I
E C   I H   N
T     S     G

二.题解

(一)思路

       1. 通过题目得知,将字符串按照Z字形排列后,从左到右,输出每一行的字符。因此,想到每一行的字符串都是分开存储的。将每一行的字符串存储在一个字符串数组rows中。

       s是字符串,numRows是行数,如果行数为1,则直接返回原字符串s。

       2. 若字符串的长度小于numRows,说明字符串连第一列都没排完,则rows数组只需存储s.size()个字符串即可;若字符串的长度大于或等于numRows,说明第一列排完了,可以继续排列,rows数组需存储numRows个字符串。

       3. 模拟排列的场景,可知连续字符的行数都是递增或者递减的,当向下排列时,则行数递增;当向上排列时,则行数递减。我们可以设置一个flag,表示排列的方向。向下时为+1,行数+1;向上时为-1,行数-1。

       4. 注意,有边界的问题。当字符到了最后一行时,方向需要变化,flag需要改变。

       5. 我们要做的是以行数为索引,为每一行的字符串添加当前的字符。最后再遍历一次rows,将所有的字符串都连接起来,返回字符串。

(二)代码实现


C++

#include
#include
#include

using namespace std;

class Solution {
public:
	string convert(string s, int numRows) {

		if (numRows == 1)
			return s;
		//创建字符串数组
		vector<string> rows(min(numRows, int(s.size())));
		int curRow = 0;
		int flag = 1;
		for (char c : s) {
			rows[curRow] += c;
			if (curRow == 0 || curRow == numRows - 1)
				flag = -flag;
			curRow += flag;
		}

		string ret;
		for (string row : rows)
			ret += row;
		return ret;
	}
};

Java:

import java.util.ArrayList;
import java.util.List;

public class Solution {
    public String convert(String s, int numRows){
        if(numRows == 1)
            return s;
        List<StringBuilder> rows = new ArrayList<>();
        for(int i = 0; i < Math.min(numRows, s.length()); ++i)
            rows.add(new StringBuilder());
        
        int curRow = 0;
        int flag = -1;
        for(char c : s.toCharArray()){
            rows.get(curRow).append(c);
            if(curRow == 0 || curRow == numRows-1)
                flag = -flag;
            curRow += flag;
        }
        
        StringBuilder ret = new StringBuilder();
        for(StringBuilder row : rows)
            ret.append(row);
        return ret.toString();
    }
}

你可能感兴趣的:(#,LeetCode,字符串,算法,leetcode)