src/02/05/stack.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

#include "stack.h"
#include <stdlib.h>

Node *node_init(int data) {
  Node *node = malloc(sizeof(Node));
  node->next = NULL;
  node->data = data;
  return node;
}

Node *node_tail(Node *self) {
  Node *current = self;
  while (current) {
    if (current->next == NULL)
      return current;
    current = current->next;
  }
  return NULL;
}

Stack *stack_init(int data) {
  Stack *stack = malloc(sizeof(Stack));
  stack->head = node_init(data);
  return stack;
}

int stack_size(Stack *self) {
  if (!self || !self->head)
    return 0;

  int count;
  Node *current = self->head;
  while (current) {
    ++count;
    current = current->next;
  }

  return count;
}

int stack_peek(Stack *self) {
  Node *tail = node_tail(self->head);
  if (tail)
    return tail->data;
  return -1;
}