LeedCode刷题系列0006——Z 字形变换

问题：

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

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

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

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

测试用例：

示例 1：

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

示例 2：

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

代码如下：

package com.lmm.LeedCode;

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

/**
 * @author lmm E-mail:violet_mmhh@163.com
 * @version 时间：2019年2月24日
 */
public class Solution0006 {

	public static 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;
		boolean goingDown = false;
		for (char c : s.toCharArray()) {
			rows.get(curRow).append(c);
			if (curRow == 0 || curRow == numRows - 1)
				goingDown = !goingDown;
			curRow += goingDown ? 1 : -1;
		}
		StringBuilder ret = new StringBuilder();
		for (StringBuilder row : rows)
			ret.append(row);
		return ret.toString();
	}

	public static void main(String[] args) {
		// Scanner in = new Scanner(System.in);
		// String s = in.next();
		// int n = in.nextInt();
		String s = "LEETCODEISHIRING";
		System.out.println("字符串为：" + s);
		int n = 3;
		System.out.println("当n=3时,结果是：" + convert(s, n));
		n = 4;
		System.out.println("当n=4时,结果是：" + convert(s, n));
	}

}

运行结果：