算法题之字符串的Z字变换

算法题之字符串的Z字变换

题目描述

题解

代码实现

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

public class ZWordExchange {
    /*
     * 将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
     * 比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，
     * 排列如下：
     * L   C   I   R
     * E T O E S I I G
     * E   D   H   N
     * */
    public static void main(String[] args) {
        String str = "abcdefghijk";
        ZWordExchange solution = new ZWordExchange();
        String zResult = solution.convert(str, 3);
        System.out.println(zResult);
    }

    public String convert(String s, int numRows) {
        if (numRows < 2) {
            return s;
        }
        //将字符串转换成字符数组
        char[] chars = s.toCharArray();
        //定义一个集合,保存每一行的字符
        List<StringBuilder> list = new ArrayList<>();
        //对集合内的StringBuilder进行初始化:一共保存numRows行元素
        for (int i = 0; i < numRows; i++) {
            list.add(new StringBuilder());
        }
        //遍历字符串,每当遍历到行数为rowNums - 1(即最后一行)或者第0行,进行Z字反转
        //利用row:表示行,flag:表示反转的标志
        int row = 0;
        int flag = -1;
        for (char val : chars) {
            list.get(row).append(val);
            if (row == 0 || row == numRows - 1) {
                //若行数为0,则flag = 1,每次进行行数加1操作,即将字符串的字符依次存入每一行中
                //若行数为rowNums,则flag = -1,每次进行行数减1操作,即将字符串的字符依次按Z字存入对应的行中
                flag = -flag;
            }
            row = row + flag;
        }
        //此时list集合中的每一个StringBuilder对象中存入的即是存入对应下标行的元素
        //将其按照下标取出,保存到一个新的StringBuilder对象进行返回
        StringBuilder res = new StringBuilder();
        for (StringBuilder rowVal : list) {
            //System.out.println(rowVal);
            res.append(rowVal);
        }
        return res.toString();
    }
}