Merge pull request #36 from RichardWeiYang/master
implement ring queue in c
This commit is contained in:
commit
b1244bb9de
130
c-cpp/09_queue/ring_queue.c
Normal file
130
c-cpp/09_queue/ring_queue.c
Normal file
@ -0,0 +1,130 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
|
||||||
|
struct ring_queue {
|
||||||
|
int cap;
|
||||||
|
int head, tail;
|
||||||
|
int *_q;
|
||||||
|
};
|
||||||
|
|
||||||
|
int alloc_queue(struct ring_queue* queue, int cap)
|
||||||
|
{
|
||||||
|
if (!queue || cap < 0)
|
||||||
|
return -1;
|
||||||
|
if (queue->_q)
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
queue->_q = (int *)malloc(cap * sizeof(int));
|
||||||
|
if (!queue->_q)
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
queue->head = queue->tail = 0;
|
||||||
|
queue->cap = cap;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void free_queue(struct ring_queue *queue)
|
||||||
|
{
|
||||||
|
queue->cap = 0;
|
||||||
|
queue->head = queue->tail = 0;
|
||||||
|
free(queue->_q);
|
||||||
|
}
|
||||||
|
|
||||||
|
int _valid_index(int curr, int step, int cap)
|
||||||
|
{
|
||||||
|
return (curr + step) % cap;
|
||||||
|
}
|
||||||
|
|
||||||
|
int _next(int curr, int cap)
|
||||||
|
{
|
||||||
|
return _valid_index(curr, 1, cap);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool is_empty(struct ring_queue *queue)
|
||||||
|
{
|
||||||
|
return (queue->head == queue->tail);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool is_full(struct ring_queue *queue)
|
||||||
|
{
|
||||||
|
int next_tail = _next(queue->tail, queue->cap);
|
||||||
|
return (next_tail == queue->head);
|
||||||
|
}
|
||||||
|
|
||||||
|
int enqueue(struct ring_queue* queue, int elem)
|
||||||
|
{
|
||||||
|
if (is_full(queue))
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
queue->_q[queue->tail] = elem;
|
||||||
|
queue->tail = _next(queue->tail, queue->cap);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int dequeue(struct ring_queue* queue, int *elem)
|
||||||
|
{
|
||||||
|
if (is_empty(queue))
|
||||||
|
return -1;
|
||||||
|
|
||||||
|
if (elem)
|
||||||
|
*elem = queue->_q[queue->head];
|
||||||
|
queue->head = _next(queue->head, queue->cap);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int size(struct ring_queue* queue)
|
||||||
|
{
|
||||||
|
int size = queue->tail - queue->head;
|
||||||
|
|
||||||
|
if (size < 0)
|
||||||
|
size += queue->cap;
|
||||||
|
return size;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dump(struct ring_queue* queue)
|
||||||
|
{
|
||||||
|
int i, idx;
|
||||||
|
|
||||||
|
printf("Queue has %d elements with %d capacity\n",
|
||||||
|
size(queue), queue->cap);
|
||||||
|
for (i = 0; i < size(queue); i++) {
|
||||||
|
idx = _valid_index(queue->head, i, queue->cap);
|
||||||
|
printf("[%02d]: %08d\n", idx, queue->_q[idx]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int main()
|
||||||
|
{
|
||||||
|
struct ring_queue queue = {0, 0, 0, NULL};
|
||||||
|
int i;
|
||||||
|
|
||||||
|
if (alloc_queue(&queue, 8)) {
|
||||||
|
printf("Failed to allocate a queue\n");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
printf("A new queue is %s\n", is_empty(&queue)?"empty":"not empty");
|
||||||
|
|
||||||
|
enqueue(&queue, 1);
|
||||||
|
printf("After enqueue 1 element, queue is %s\n", is_empty(&queue)?"empty":"not empty");
|
||||||
|
dequeue(&queue, NULL);
|
||||||
|
printf("After dequeue 1 element, queue is %s\n", is_empty(&queue)?"empty":"not empty");
|
||||||
|
|
||||||
|
for (i = 0; i < 7; i++)
|
||||||
|
enqueue(&queue, i);
|
||||||
|
printf("After enqueue 7 element, queue is %s\n", is_full(&queue)?"full":"not full");
|
||||||
|
|
||||||
|
for (i = 0; i < 4; i++) {
|
||||||
|
dequeue(&queue, NULL);
|
||||||
|
enqueue(&queue, i);
|
||||||
|
}
|
||||||
|
printf("After enqueue/dequeue 4 element, queue is %s\n",
|
||||||
|
is_full(&queue)?"full":"not full");
|
||||||
|
printf("Head is %d, Tail is %d\n", queue.head, queue.tail);
|
||||||
|
|
||||||
|
dump(&queue);
|
||||||
|
free_queue(&queue);
|
||||||
|
return 0;
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user