diff --git a/codes/c/chapter_array_and_linkedlist/my_list.c b/codes/c/chapter_array_and_linkedlist/my_list.c index f80802448..a1ca7717a 100644 --- a/codes/c/chapter_array_and_linkedlist/my_list.c +++ b/codes/c/chapter_array_and_linkedlist/my_list.c @@ -20,7 +20,7 @@ typedef struct mylist MyList; void extendCapacity(MyList* l); /* 构造函数 */ -void constructor(MyList* l) { +void newMyList(MyList* l) { l->capacity = 10; l->nums = malloc(sizeof(int) * l->capacity); l->size = 0; @@ -31,60 +31,56 @@ void constructor(MyList* l) { int size(MyList* l) { return l->size; } + /* 获取列表容量 */ int capacity(MyList* l) { return l->capacity; } + /* 访问元素 */ -int get(MyList* l, int idx) { - if (idx < l->size) { - return l->nums[idx]; - } else { - - } +int get(MyList* l, int index) { + assert(index < l->size); + return l->nums[index]; } + /* 更新元素 */ -void set(MyList* l, int idx, int elem) { - if (idx < l->size) { - l->nums[idx] = elem; - } +void set(MyList* l, int index, int num) { + assert(index < l->size); + l->nums[index] = num; } + /* 尾部添加元素 */ -void add(MyList* l, int elem) { +void add(MyList* l, int num) { if (size(l) == capacity(l)) { extendCapacity(l); // 扩容 } - l->nums[size(l)] = elem; - l->size ++; + l->nums[size(l)] = num; + l->size++; } + /* 中间插入元素 */ -void insert(MyList* l, int idx, int elem) { - if (idx < size(l)) { - for (int i=size(l); i>idx; --i) { - l->nums[i] = l->nums[i-1]; - } - l->nums[idx] = elem; - l->size++; - } else { - // 越界:异常处理 +void insert(MyList* l, int index, int num) { + assert(index < size(l)); + for (int i=size(l); i>index; --i) { + l->nums[i] = l->nums[i-1]; } + l->nums[index] = num; + l->size++; } + /* 删除元素 */ // 由于引入了 stdio.h ,此处无法使用 remove 关键词 // 详见 https://github.com/krahets/hello-algo/pull/244#discussion_r1067863888 -int removeNum(MyList* l, int idx) { - if (idx < size(l)) { - size_t i = idx; - if ( i != size(l)-1) { - for (; inums[i] = l->nums[i+1]; - } - } - l->size--; - } else { - // 数组越界 +int removeNum(MyList* l, int index) { + assert(index < size(l)); + int num = l->nums[index]; + for (int i=index; inums[i] = l->nums[i+1]; } + l->size--; + return num; } + /* 列表扩容 */ void extendCapacity(MyList* l) { // 先分配空间 @@ -112,7 +108,7 @@ int* toArray(MyList* l) { int main() { /* 初始化列表 */ MyList list; - constructor(&list); + newMyList(&list); /* 尾部添加元素 */ add(&list, 1); add(&list, 3); diff --git a/codes/c/chapter_stack_and_queue/linkedlist_stack.c b/codes/c/chapter_stack_and_queue/linkedlist_stack.c index c0d02c968..5f09b8cf8 100644 --- a/codes/c/chapter_stack_and_queue/linkedlist_stack.c +++ b/codes/c/chapter_stack_and_queue/linkedlist_stack.c @@ -9,31 +9,36 @@ /* 基于链表实现的栈 */ struct LinkedListStack { ListNode* stackTop; // 将头结点作为栈顶 - int stkSize; // 栈的长度 + int size; // 栈的长度 }; typedef struct LinkedListStack LinkedListStack; + /* 构造函数 */ -void constructor(LinkedListStack* stk) { +void newLinkedListStack(LinkedListStack* stk) { stk->stackTop = NULL; - stk->stkSize = 0; + stk->size = 0; } + /* 获取栈的长度 */ int size(LinkedListStack* stk) { assert(stk); - return stk->stkSize; + return stk->size; } + /* 判断栈是否为空 */ bool empty(LinkedListStack* stk) { assert(stk); return size(stk) == 0; } + /* 访问栈顶元素 */ int top(LinkedListStack* stk) { assert(stk); assert(size(stk) != 0); return stk->stackTop->val; } + /* 入栈 */ void push(LinkedListStack* stk, int num) { assert(stk); @@ -41,8 +46,9 @@ void push(LinkedListStack* stk, int num) { node->next = stk->stackTop; // 更新新加结点指针域 node->val = num; // 更新新加结点数据域 stk->stackTop = node; // 更新栈顶 - stk->stkSize++; // 更新栈大小 + stk->size++; // 更新栈大小 } + /* 出栈 */ void pop(LinkedListStack* stk) { assert(stk); @@ -51,15 +57,14 @@ void pop(LinkedListStack* stk) { stk->stackTop = stk->stackTop->next; // 释放内存 free(tmp); - stk->stkSize--; + stk->size--; } - /* Driver Code */ int main() { /* 初始化栈 */ LinkedListStack stack; - constructor(&stack); + newLinkedListStack(&stack); /* 元素入栈 */ push(&stack, 1);