helper.go 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. package main
  2. import (
  3. "strconv"
  4. )
  5. // ListNode ...
  6. type ListNode struct {
  7. Val int
  8. Next *ListNode
  9. }
  10. // TreeNode ...
  11. type TreeNode struct {
  12. Val int
  13. Left *TreeNode
  14. Right *TreeNode
  15. }
  16. // Interval ...
  17. type Interval struct {
  18. Start int
  19. End int
  20. }
  21. func abs(x int) int {
  22. if x < 0 {
  23. return -x
  24. }
  25. return x
  26. }
  27. func maxInt(x, y int) int {
  28. if x > y {
  29. return x
  30. }
  31. return y
  32. }
  33. func minInt(x, y int) int {
  34. if x < y {
  35. return x
  36. }
  37. return y
  38. }
  39. func minInts(vals ...int) (min int) {
  40. min = vals[0]
  41. for i := 1; i < len(vals); i++ {
  42. if vals[i] < min {
  43. min = vals[i]
  44. }
  45. }
  46. return
  47. }
  48. func toLinkedList(num []int) *ListNode {
  49. length := len(num)
  50. if length == 0 {
  51. return nil
  52. }
  53. head := ListNode{num[0], nil}
  54. curr := &head
  55. for i := 1; i < length; i++ {
  56. curr.Next = &ListNode{num[i], nil}
  57. curr = curr.Next
  58. }
  59. return &head
  60. }
  61. func printList(head *ListNode) {
  62. curr := head
  63. for curr != nil {
  64. print(strconv.FormatInt(int64(curr.Val), 10), " ")
  65. curr = curr.Next
  66. }
  67. println()
  68. }
  69. func printTree(root *TreeNode) { // Level order traversal
  70. if root == nil {
  71. println("nil")
  72. return
  73. }
  74. queue := make([]*TreeNode, 0) // Important!
  75. queue = append(queue, root)
  76. for len(queue) != 0 {
  77. curr := queue[0]
  78. queue = queue[1:] // Dequeue
  79. if curr == nil {
  80. print("nil ")
  81. continue
  82. }
  83. print(curr.Val, " ")
  84. if curr.Left != nil {
  85. queue = append(queue, curr.Left)
  86. } else if curr.Right != nil {
  87. queue = append(queue, nil)
  88. }
  89. if curr.Right != nil {
  90. queue = append(queue, curr.Right)
  91. } else if curr.Left != nil {
  92. queue = append(queue, nil)
  93. }
  94. }
  95. println()
  96. }
  97. func printBoard(board [][]byte) {
  98. for _, row := range board {
  99. for _, grid := range row {
  100. print(string(grid), " ")
  101. }
  102. println()
  103. }
  104. }