#LeetCode:6. ZigZag Conversion

找出每一行的 index 會怎麼跳即可。至於怎麼跳的方式寫在下面:
1. row = 0row = numRows - 1 的情況,第一項是從 index = row 開始,往後 index 的跳法是 2 * (numRows - 1)
2. 其餘的情況,可以從最上面的那排去想,每一行的某一項會是與最上面那排的那項相差往前 row 個和往後 row 個,這樣就可以找到該行的每一項。

C++(20ms)

/*******************************************************/
/* LeetCode 6. ZigZag Conversion                       */
/* Author: Maplewing [at] knightzone.studio            */
/* Version: 2018/09/26                                 */
/*******************************************************/
class Solution {
public:
  string convert(string s, int numRows) { 
    int sLength = s.length();
    if(sLength <= numRows || numRows == 1) return s;

    int distance = 2 * (numRows - 1);
    string zigzagOrderString = "";

    for(int row = 0 ; row < numRows ; ++row){
      if(row == 0 || row == numRows - 1){
        for(int i = row ; i < sLength ; i += distance){
          zigzagOrderString += s[i];
        }
      }
      else{
        for(int i = 0 ; i - row < sLength ; i += distance){
          if( i - row >= 0 ) zigzagOrderString += s[i - row];
          if( i + row < sLength ) zigzagOrderString += s[i + row];
        }
      }
    }

    return zigzagOrderString;
  }
};

Kotlin(228ms)

/*******************************************************/
/* LeetCode 6. ZigZag Conversion                       */
/* Author: Maplewing [at] knightzone.studio            */
/* Version: 2019/05/02                                 */
/*******************************************************/
class Solution {
    fun convert(s: String, numRows: Int): String {
        if(s.length <= numRows || numRows == 1) return s

        val distance = 2 * (numRows - 1)
        var zigzagOrderString = StringBuilder()
        for(row in 0 until numRows){
            if(row == 0 || row == numRows - 1){
                for(i in row until s.length step distance){
                    zigzagOrderString.append(s[i])
                }
            }
            else{
                for(i in 0 until row + s.length step distance){
                    if(i - row >= 0) zigzagOrderString.append(s[i - row])
                    if(i + row < s.length) zigzagOrderString.append(s[i + row])
                }
            }
        }

        return zigzagOrderString.toString()
    }
}

回應

目前無留言,歡迎留言!

發表迴響

這個網站採用 Akismet 服務減少垃圾留言。進一步瞭解 Akismet 如何處理網站訪客的留言資料

%d 位部落客按了讚: