力扣——Z字形变换

原创

CMMKK 2022-10-20 10:22:06 博主文章分类：力扣中等题目集 ©著作权

文章标签 java 算法 leetcode 字符串 i++ 文章分类 后端开发

©著作权归作者所有：来自51CTO博客作者CMMKK的原创作品，请联系作者获取转载授权，否则将追究法律责任

Z 字形变换

将一个给定字符串 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<2) return s;
        ArrayList<StringBuilder> list=new ArrayList<StringBuilder>();
        for(int i=0;i<numRows;i++){
            list.add(new StringBuilder());//给每个list添加StringBuilder 方法
        }
        int i=0,flag=-1;
        for(char c:s.toCharArray()){
            list.get(i).append(c);
            if(i==0||i==numRows-1) flag=-flag;
            i+=flag;
        }
        StringBuilder res=new StringBuilder();
        for(StringBuilder stringBuilder:list) res.append(stringBuilder);
        return res.toString();
    }
}

实现调用：

package com.kk;

import java.util.ArrayList;
import java.util.Scanner;

public class ZGlyphTransformation {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String s=scanner.nextLine();
        int numRows=scanner.nextInt();
        System.out.println(convert(s, numRows));


    }


    public static String convert(String s, int numRows) {
        if (numRows<2) return s;
        ArrayList<StringBuilder> list = new ArrayList<StringBuilder>();
        for (int i = 0; i < numRows; i++) {
            list.add(new StringBuilder()); //给每个list添加StringBuilder 方法
        }
        int i=0,flag=-1;
        //将字符串转换为字符数组s.toCharArray()
        for (char c:s.toCharArray()){
            list.get(i).append(c);//使用StringBuilder的功能 append
            if (i==0||i==numRows-1) flag=-flag;
            i+=flag;
        }
        StringBuilder res = new StringBuilder();
        for (StringBuilder stringBuilder: list) res.append(stringBuilder);
        return res.toString();

    }
}