Leetcode 6 Z字形变换

2019-10-19  本文已影响0人  hekirakuno

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

思路:flag方向标记,list实现二维结构

class Solution {
    public String convert(String s, int numRows) {
        //很愚蠢的办法
        //设置标记
        int index = 0;
        //设置方向
        boolean flag = true;
        //如果numRows>s.length()
        int len = s.length();
        if(numRows>=len||numRows==1){
           return s;
        }
        //结果集
        List<StringBuffer> list = new ArrayList<>();
        //填入创建结果行
        for(int i = 0;i<numRows;i++){
            StringBuffer sb = new StringBuffer();
            list.add(sb);
        }
        //遍历字符串
        for(int i=0;i<len;i++){
            //当前存储方向(flag==true,向下,flag==false向上)
            if(index>=numRows){
                flag = false;
                index=index-2;
            }
            if(index<0){
                flag = true;
                index=index+2;
            }
            //存入当前字符
            list.get(index).append(s.charAt(i));
            //调整下标
            if(flag){
                index++;
            }else{
                index--;
            }
        }
        StringBuffer result = new StringBuffer();
        //组合结果
        for(StringBuffer sb : list){
            result.append(sb.toString());
        }
        return result.toString();
    
    }
}
上一篇 下一篇

猜你喜欢

热点阅读