go 언어 구현stack

2073 단어
golang
package stack

import (
    "errors"
)

type Stack []interface{}

func(stack *Stack) Push(v interface{}) {
    *stack = append(*stack, v)
}

func(stack *Stack) Pop() (interface{}, error){
    if len(*stack) == 0 { 
        return nil, errors.New("stack empty")
    }   
    v := (*stack)[len(*stack) - 1]
    *stack = (*stack)[:len(*stack) - 1]
    return v, nil 
}

func(stack *Stack) Top() (interface{}, error) {
    if len(*stack) == 0 { 
        return nil, errors.New("stack empty")
    }   

    return (*stack)[len(*stack) - 1], nil 
}

func(stack *Stack) Len() int {
    return len(*stack)
}

같은 사고방식은 c로 실현된다.
typedef struct my_stack_t my_stack_t;
struct my_stack_t {
    void **s;  /*  void *  */
    int use;   /*   */
    int size;  /*   */
    void (*do_free)(void *);
};

#define DEFAULT_N 100
int stack_init(my_stack_t *s, void (*do_free)()) {
    assert(s != NULL);
    s->s = (void **)malloc(sizeof(void *) * DEFAULT_N);
    s->use = 0;
    s->size = DEFAULT_N;
    s->do_free = do_free;
    return 0;
}

int stack_push(my_stack_t *s, void *v) {
    assert(s != NULL);
    void **p;
    if (s->use == s->size) {
        p = (void **)realloc(s->s, s->size * 2);
        if (p == NULL)
            return -1;
        s->s = p;
        s->size *= 2;
    }
    s->s[s->use++] = v;
    return 0;
}

void *stack_pop(my_stack_t *s) {
    assert(s != NULL);
    if (s->use == 0) {
        return NULL;
    }   

    return s->s[--s->use];
}

void *stack_top(my_stack_t *s) {
    assert(s != NULL);
    if (s->use == 0) {
        return NULL;
    }   
    return s->s[s->use - 1]; 
}

int stack_len(my_stack_t *s) {
    assert(s != NULL);
    return s->use;
}

int stack_free(my_stack_t *s) {
    void *v; 
    if (s->do_free == NULL) {
        goto done;
    }   

    while ((v = stack_pop(s)) != NULL) {
        s->do_free(s);
    }   
done:
    free(s->s);
    return 0;
}

*golang의 interface{}는 c의void와 같다*

좋은 웹페이지 즐겨찾기