Z字型の変換(Go、LeetCode)

目次

タイトル説明

解決

コード

コードウォークスルー

ポータル


 

タイトル説明

指定された行数に従って、指定された文字列を上から下、左から右にジグザグパターンで配置します。たとえば、入力文字列がLEETCODEISHIRINGで、行数が3の場合、配置は次のようになります。

L   C   I   R
E T O E S I I G
E   D   H   N

その後、出力を左から右に1行ずつ読み取る必要があり、新しい文字列になります:LCIRETOESIIGEDHN

 

解決

シミュレーション手法を使用して、鮮明な画像を解決します。

行数がnであると仮定して、n行を表すn個のリストを生成し、ジグザグ傾向をシミュレートするポインター変数を定義します。最初の行から順番に下に移動し、最後の行に到達した後に上に移動して、ループを繰り返します。すべての要素がトラバースされるまで。

ポインタ変数が要素に到達するたびに、その要素は対応する行数のリストに挿入されます。最後に、文字の各行が接続されて出力され、目的の結果が得られます。

複雑さの分析:文字列を1回トラバースするだけでよいため、時間計算量は オン)です。一定数の変数のみを定義する必要があるため、スペースの複雑さは O(1) です。

 

コード

package main

import (
	"fmt"
	"strings"
)

func convert(s string, numRows int) string {
	if numRows == 1 {
		return s
	}

	pool := make([][]string, 0)
	for i := 0; i < numRows; i++ {
		pool = append(pool, make([]string, 0))
	}
	down := false
	currentLine := 0

	for _, c := range s {
		if currentLine == 0 || currentLine == numRows-1 {
			down = !down
		}

		pool[currentLine] = append(pool[currentLine], string(c))
		if down {
			currentLine++
		} else {
			currentLine--
		}
	}

	result := ""
	for i := range pool {
		line := strings.Join(pool[i], "")
		result += line
	}
	return result
}

 

コードウォークスルー

package main

import (
   "fmt"
   "strings"
)

func convert(s string, numRows int) string {
   // 如果给定的行数只有一行,那么只需要将原字符串返回
   if numRows == 1 {
      return s
   }

   // 初始化存储每一行字符的二维切片
   pool := make([][]string, 0)
   for i := 0; i < numRows; i++ {
      pool = append(pool, make([]string, 0))
   }

   // down变量用来模拟Z字型游标的走向(flase向上,true向下)
   down := false

   // 游标指针变量,表示当前行数
   currentLine := 0

   for _, c := range s {
      // 如果游标抵达边界,需要改变方向
      if currentLine == 0 || currentLine == numRows-1 {
         down = !down
      }

      // 对应行添加字符
      pool[currentLine] = append(pool[currentLine], string(c))
      if down {
         currentLine++
      } else {
         currentLine--
      }
   }

   // 按照每一行字符依次拼接的方式输出结果
   result := ""
   for i := range pool {
      line := strings.Join(pool[i], "")
      result += line
   }
   return result
}

// 自测用例
func main() {
   fmt.Println(convert("LEETCODEISHIRING", 3))
}

 

ポータル

LeetCodeテストの質問リンク

おすすめ

転載: blog.csdn.net/TCatTime/article/details/107074091