6. Z 字形变换

/**
 * 6. Z 字形变换
 * @author wsq
 * @date 2020/09/24
	将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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
	
	链接：https://leetcode-cn.com/problems/zigzag-conversion
 */
package notsubmit;

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

public class ZigzagTransformation {
	/**
	 *	通过寻找每一行元素之间的位置规律实现按行访问。
	 *	官方给的解法一，看完之后觉得自己很菜了，这里将解法一也写在下面
	 * @param s
	 * @param numRows
	 * @return
	 */
	public String convert(String s, int numRows) {
		if(s.length() < numRows) {
			return s;
		}
		StringBuilder sb = new StringBuilder();
		for(int i = 0; i < numRows; i++) {
			int pos = i;
			while(pos < s.length()) {
				sb.append(s.charAt(pos));
				pos += 2 * numRows - 2;
				if((i != 0 && i != numRows - 1) && (pos - 2 * i) < s.length()) {
					sb.append(s.charAt(pos - 2 * i));
				}
			}
		}
		return sb.toString();
	}
	/**
	 *	官方的解法一，想法很独特，唯一不足就是空间的消耗
	 *	用四个列表存储对应排序后的元素
	 * @param s
	 * @param numRows
	 * @return
	 */
	public String convert2(String s, int numRows) {
		if(numRows == 1) {
			return s;
		}
		List<StringBuilder> rows = new ArrayList();
		for(int i = 0; i < Math.min(s.length(), numRows); i++) {
			rows.add(new StringBuilder());
		}
		
		boolean goDown = false;
		int curRow = 0;
		for(char c: s.toCharArray()) {
			rows.get(curRow).append(c);
			if(curRow == 0 || curRow == numRows - 1) {
				goDown = !goDown;
			}
			curRow += goDown ? 1 : -1;
		}
		StringBuilder ans = new StringBuilder();
		for(StringBuilder sb: rows) {
			ans.append(sb);
		}
		return ans.toString();
	}
	
	
	public static void main(String[] args) {
		String s = "LEETCODEISHIRING";
		int numRows = 3;
		ZigzagTransformation zt = new ZigzagTransformation();
		String ans = zt.convert2(s, numRows);
		System.out.println(ans);
	}
}