Z 字形变换 JAVA

日期:2020-10-9

作者:19届 LZ

标签:JAVA

题目描述

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

解题思路：

把2numRows-2作为一个循环cyc,
按公式直接输出就行了,
第一行和最后一行,每个字符之间差一个cyc;
中间部分每个字符分别差cyc-2i (i表示是第几行),以及2i;

代码

class Solution {
    public String convert(String s, int numRows) {
      
      if(numRows==1)
         return s.toString();

      StringBuilder result= new StringBuilder();

      int all=s.length();
      int cyc=2*numRows-2;
        
      for(int j=0;j<all;j+=cyc){                      //第一行
          result.append(s.charAt(j));
      }

      for(int i=1;i<numRows-1;i++){                  //中间部分
          for(int j=i;j<all;){
            
            result.append(s.charAt(j));
            j+=cyc-2*i;

            if(j<all){
            result.append(s.charAt(j));
            j+=2*i;
            }
          }
      }

      for(int j=numRows-1;j<all;j+=cyc){             //最后一行
          result.append(s.charAt(j));
      }
     return result.toString();
    }
}