Leet Code 6 ZigZag Conversion - 字符串之字形转换 - Java

问题原始链接 https://leetcode.com/problems/zigzag-conversion

字符串"PAYPALISHIRING"按给定行数，之字形输出，即先从上往下，再从下往右上，再从上往下...

P   A   H   N
A P L S I I G
Y   I   R

然后按行从上往下输出："PAHNAPLSIIGYIR"。

第一种方法：

按行从输入字符串中查找字符，拼接到结果字符串。

Leet Code 测试耗时 44ms。大量的字符串拼接导致性能低。

public class Solution {
  public String convert(String s, int numRows) {
    if (s == null || s.length() == 0 || numRows == 1) {
      return s;
    }
    String result = "";
    int m = 2 * numRows - 2;
    for (int i = 0; i < numRows; i++) {
      for (int j = 0;; j++) {
        int start = j * m;
        if (start >= s.length()) {
          break;
        }
        int k1 = start + i;
        if (k1 < s.length()) {
          result += s.charAt(k1);
        }

        if (i != 0 && i != numRows - 1) {
          int k2 = start + m - i;
          if (k2 != k1 && k2 < s.length()) {
            result += s.charAt(k2);
          }
        }
      }
    }
    return result;
  }
}

 

第二种方法：

使用StringBuilder，按行从输入字符串中查找字符，添加到StringBuilder中，最后一次性构造结果字符串输出。

Leet Code 测试耗时 10ms。

public class Solution {
  public String convert(String s, int numRows) {
    if (s == null || s.length() == 0 || numRows == 1) {
      return s;
    }
    StringBuilder sb = new StringBuilder();
    int m = 2 * numRows - 2;
    for (int i = 0; i < numRows; i++) {
      for (int j = 0;; j++) {
        int start = j * m;
        if (start >= s.length()) {
          break;
        }
        int k1 = start + i;
        if (k1 < s.length()) {
          sb.append(s.charAt(k1));
        }
        if (i != 0 && i != numRows - 1) {
          int k2 = start + m - i;
          if (k2 != k1 && k2 < s.length()) {
            sb.append(s.charAt(k2));
          }
        }
      }
    }
    return sb.toString();
  }
}

 

第三种方法：

构造指定行数个StringBuffer，扫描输入字符串，把每个字符添加到对应行的StringBuffer，最后把所有行的StringBuffer拼接成一个字符串。

Leet Code 测试耗时 14ms。

public class Solution {
  public String convert(String s, int numRows) {
    if (s == null || s.length() == 0 || numRows == 1) {
      return s;
    }
    StringBuilder[] sbs = new StringBuilder[numRows];
    for (int i = 0; i < numRows; i++) {
      sbs[i] = new StringBuilder();
    }
    int m = 2 * numRows - 2;
    for (int i = 0; i < s.length(); i++) {
      int j = i % m;
      sbs[j < numRows ? j : m - j].append(s.charAt(i));
    }
    for (int i = 1; i < numRows; i++) {
      sbs[0].append(sbs[i]);
    }
    return sbs[0].toString();
  }
}

 

第四种方法：

构造一个长度等于输入字符串长度的结果字符数组，构造一个索引数组记录每行下个字符对应的结果字符数组的索引，扫描输入字符串，把字符放入结果字符数组中相应位置。

Leet Code 测试耗时 7ms。

public class Solution {
  public String convert(String s, int numRows) {
    if (s == null || s.length() == 0 || numRows == 1) {
      return s;
    }
    char[] result = new char[s.length()];
    int[] index = new int[numRows];
    int m = 2 * numRows - 2;
    int times = (s.length() - 1) / m;
    int firstRowLastIndexInS = times * m;
    int left = s.length() - firstRowLastIndexInS - 1;
    for (int i = 1; i < numRows; i++) {
      int preRowCount = 0;
      if (i - 1 == 0) {
        preRowCount = times + 1;
      } else {
        preRowCount = times == 0 ? 0 : 2 * times;
        if (left >= i - 1) {
          preRowCount++;
        }
        if (left >= numRows && i - 1 >= m - left) {
          preRowCount++;
        }
      }
      index[i] = index[i - 1] + preRowCount;
    }
    for (int i = 0; i < s.length(); i++) {
      int j = i % m;
      int row = j < numRows ? j : m - j;
      result[index[row]] = s.charAt(i);
      index[row] = index[row] + 1;
    }
    return new String(result);
  }
}

 

转载于:https://my.oschina.net/mistymarch/blog/687566