1. ホーム
  2. go

[解決済み] golangでリーズナブルなスタック実装をお探しですか?

2022-03-11 22:26:10

質問

今のところ、私の素朴なアプローチは

type stack []int

func (s *stack) Push(v int) {
    *s = append(*s, v)
}

func (s *stack) Pop() int {
    res:=(*s)[len(*s)-1]
    *s=(*s)[:len(*s)-1]
    return res
}

動作する プレイグラウンド しかし、見た目が悪く、参照外しが多すぎる。もっといい方法はないでしょうか?

どのように解決するのですか?

スタイルと個人の好みの問題です。 あなたのコードで大丈夫です (スレッドセーフでないことと、空のスタックからポップするとパニックになることは別として)。これを少し単純化すると、バリューメソッドで作業してスタック自体を返すことができ、よりエレガントになります。 好みによりますが ... すなわち

type stack []int

func (s stack) Push(v int) stack {
    return append(s, v)
}

func (s stack) Pop() (stack, int) {
    // FIXME: What do we do if the stack is empty, though?

    l := len(s)
    return  s[:l-1], s[l-1]
}


func main(){
    s := make(stack,0)
    s = s.Push(1)
    s = s.Push(2)
    s = s.Push(3)

    s, p := s.Pop()
    fmt.Println(p)

}

もう一つの方法は、構造体でラップすることで、レースコンディションを避けるためにミューテックスを簡単に追加することなどもできます。

type stack struct {
     lock sync.Mutex // you don't have to do this if you don't want thread safety
     s []int
}

func NewStack() *stack {
    return &stack {sync.Mutex{}, make([]int,0), }
}

func (s *stack) Push(v int) {
    s.lock.Lock()
    defer s.lock.Unlock()

    s.s = append(s.s, v)
}

func (s *stack) Pop() (int, error) {
    s.lock.Lock()
    defer s.lock.Unlock()


    l := len(s.s)
    if l == 0 {
        return 0, errors.New("Empty Stack")
    }

    res := s.s[l-1]
    s.s = s.s[:l-1]
    return res, nil
}


func main(){
    s := NewStack()
    s.Push(1)
    s.Push(2)
    s.Push(3)
    fmt.Println(s.Pop())
    fmt.Println(s.Pop())
    fmt.Println(s.Pop())
}