linkedList_helper.go

package leetcode

import (
	"fmt"
	"strconv"
	"strings"
)

type ListNode struct {
	Val  string
	Next *ListNode
}

type IntListNode struct {
	Val  int
	Next *IntListNode
}

func ParseLinkedListFromStr(input string) *ListNode {
	arr := strings.Split(input, `,`)
	if len(arr) == 0 {
		return nil
	}
	node := &ListNode{arr[0], nil}
	p := node
	for i := 1; i < len(arr); i++ {
		p.Next = &ListNode{arr[i], nil}
		p = p.Next
	}
	return node
}

func ParseIntLinkedListFromStr(input string) *IntListNode {
	arr := strings.Split(input, `->`)
	if len(arr) == 0 {
		return nil
	}
	intArr := make([]int, len(arr)-1)
	for i := 0; i < len(arr)-1; i++ {
		t, err := strconv.Atoi(arr[i])
		if err != nil {
			panic(err)
		}
		intArr[i] = t
	}
	node := &IntListNode{intArr[0], nil}
	p := node
	for i := 1; i < len(arr)-1; i++ {
		p.Next = &IntListNode{intArr[i], nil}
		p = p.Next
	}
	return node
}

func PrintIntLinkedListNode(node *IntListNode) {
	for p := node; p != nil; p = p.Next {
		fmt.Printf("%d->", p.Val)
	}
	fmt.Println()
}

func PrintLinkedListNode(node *ListNode) {
	for p := node; p != nil; p = p.Next {
		fmt.Printf("%s->", p.Val)
	}
	fmt.Println()
}