hello-algo/codes/c/chapter_graph/graph_adjacency_list.c

/**
 * File: graph_adjacency_list.c
 * Created Time: 2023-07-07
 * Author: NI-SW (947743645@qq.com)
 */

#include "../utils/common.h"

typedef struct Vertex Vertex;
typedef struct Node Node;
typedef struct linkList linkList;

void freeVertex(Vertex *);
void freeLinklist(linkList *);
linkList *newLinklist(Vertex *);

/* 链表节点 */
struct Node {
    // 链表节点内包含顶点类和下一个节点地址
    Vertex *val;
    Node *next;
};

/* 链表节点构造函数 */
Node *newNode() {
    Node *n = (Node *)malloc(sizeof(Node));
    n->next = 0;
    n->val = 0;
    return n;
}

/* 顶点节点类 */
struct Vertex {
    // 节点值
    int val;
    // 与其它节点相连接的边的链表
    linkList *linked;
    // 索引位，标记该顶点在顶点列表中的索引
    unsigned int pos;
};

/* 顶点节点构造函数 */
Vertex *newVertex(int val) {
    Vertex *v = (Vertex *)malloc(sizeof(Vertex));
    // 为新节点赋值并建立该节点的链表
    v->val = val;
    v->linked = newLinklist(v);
    return v;
}

/* 顶点内存释放函数 */
void freeVertex(Vertex *val) {
    // 释放该顶点和该顶点的链表的内存
    freeLinklist(val->linked);
    free(val);
}

/* 链表 */
struct linkList {
    Node *head;
    Node *tail;
};

/* 链表头插法 */
void pushFront(linkList *l, Vertex *val) {
    Node *temp = newNode();
    temp->val = val;
    temp->next = l->head->next;
    l->head->next = temp;
    if (l->tail == l->head) {
        l->tail = temp;
    }
}

/* 链表尾插法 */
void pushBack(linkList *l, Vertex *val) {
    Node *temp = newNode();
    temp->val = val;
    temp->next = 0;
    l->tail->next = temp;
    l->tail = temp;
}

/* 根据顶点地址与该顶点连接的删除边 */
void removeLink(linkList *l, Vertex *val) {
    Node *temp = l->head->next;
    Node *front = l->head;
    while (temp != 0) {
        if (temp->val == val) {
            front->next = temp->next;
            if (l->tail == temp) {
                l->tail = front;
            }
            free(temp);
            return;
        }
        front = temp;
        temp = temp->next;
    }

    if (temp->next == 0) {
        printf("vertex not found!\n");
    }
}

/* 根据索引查找链表中节点 */
Node *findByindex(linkList *l, unsigned int index) {
    unsigned int i = 0;
    Node *temp = l->head->next;
    while (temp != 0) {
        if (i == index) {
            return temp;
        }
        temp = temp->next;
        i++;
    }

    if (temp->next == 0) {
        printf("vertex not found!\n");
        return 0;
    }
    return 0;
}

/* 根据顶点地址删除顶点 */
void removeNode(linkList *l, Vertex *val) {
    Node *temp = l->head->next;
    Node *front = l->head;
    while (temp != 0) {
        if (temp->val == val) {
            front->next = temp->next;
            if (l->tail == temp) {
                l->tail = front;
            }
            freeVertex(val);
            free(temp);
            return;
        }
        front = temp;
        temp = temp->next;
    }

    if (temp->next == 0) {
        printf("vertex not found!\n");
    }
}

/* 释放链表内存 */
void freeLinklist(linkList *l) {
    Node *temp = l->head->next;
    while (temp != 0) {
        free(l->head);
        l->head = temp;
        temp = temp->next;
    }
    free(l->head);
    l->head = 0;
    free(l);
}

/* 链表构造函数 */
linkList *newLinklist(Vertex *val) {
    linkList *newLinklist = (linkList *)malloc(sizeof(linkList));

    newLinklist->head = newNode();
    newLinklist->head->val = val;
    newLinklist->tail = newLinklist->head;
    newLinklist->head->next = 0;

    return newLinklist;
}

/* 基于邻接链表实现的无向图类结构 */
struct graphAdjList {
    // 顶点列表
    Vertex **verticesList;
    // 顶点数量
    unsigned int size;
    // 当前容量
    unsigned int capacity;
};

typedef struct graphAdjList graphAdjList;

/* 添加边 */
void addEdge(graphAdjList *t, int i, int j) {
    // 越界检查
    if (i < 0 || j < 0 || i == j || i >= t->size || j >= t->size) {
        printf("Out of range in %s:%d\n", __FILE__, __LINE__);
        return;
    }
    // 查找待连接的节点
    Vertex *v1 = t->verticesList[i];
    Vertex *v2 = t->verticesList[j];

    // 连接节点
    pushBack(v1->linked, v2);
    pushBack(v2->linked, v1);
}

/* 删除边 */
void removeEdge(graphAdjList *t, int i, int j) {
    // 越界检查
    if (i < 0 || j < 0 || i == j || i >= t->size || j >= t->size) {
        printf("Out of range in %s:%d\n", __FILE__, __LINE__);
        return;
    }

    // 查找待删除边的相关节点
    Vertex *v1 = t->verticesList[i];
    Vertex *v2 = t->verticesList[j];

    // 移除待删除边
    removeLink(v1->linked, v2);
    removeLink(v2->linked, v1);
}

/* 添加顶点 */
void addVertex(graphAdjList *t, int val) {
    // 若大小超过容量，则扩容
    if (t->size >= t->capacity) {
        Vertex **tempList = (Vertex **)malloc(sizeof(Vertex *) * 2 * t->capacity);
        memcpy(tempList, t->verticesList, sizeof(Vertex *) * t->size);
        free(t->verticesList);
        // 指向新顶点表
        t->verticesList = tempList;    
        t->capacity = t->capacity * 2;
    }
    // 申请新顶点内存并将新顶点地址存入顶点列表
    Vertex *newV = newVertex(val);
    newV->pos = t->size;
    newV->linked = newLinklist(newV);
    t->verticesList[t->size] = newV;
    t->size++;
}

/* 删除顶点 */
void removeVertex(graphAdjList *t, unsigned int index) {
    // 越界检查
    if (index < 0 || index >= t->size) {
        printf("Out of range in %s:%d\n", __FILE__, __LINE__);
        exit(1);
    }

    // 查找待删节点
    Vertex *v = t->verticesList[index];
    // 若不存在该节点，则返回
    if (v == 0) {
        printf("index is:%d\n", index);
        printf("Out of range in %s:%d\n", __FILE__, __LINE__);
        return;
    }

    // 遍历待删除节点链表，将所有与待删除结点有关的边删除
    Node *temp = v->linked->head->next;
    while (temp != 0) {
        removeLink(temp->val->linked, v);
        temp = temp->next;
    }

    // 定点列表前移
    for (int i = index; i < t->size - 1; i++) {
        t->verticesList[i] = t->verticesList[i + 1];
    }
    t->verticesList[t->size - 1] = 0;
    t->size--;

    //释放被删除顶点的内存
    freeVertex(v);
}

/* 打印顶点与邻接矩阵 */
void printGraph(graphAdjList *t) {
    printf("邻接表  =\n");
    for (int i = 0; i < t->size; i++) {
        Node *n = t->verticesList[i]->linked->head->next;
        printf("%d: [", t->verticesList[i]->val);
        while (n != 0) {
            if (n->next != 0) {
                printf("%d, ", n->val->val);
            } else {
                printf("%d", n->val->val);
            }
            n = n->next;
        }
        printf("]\n");
    }
}

/* 构造函数 */
graphAdjList *newGraphic(unsigned int verticesNumber) {
    // 申请内存
    graphAdjList *newGraph = (graphAdjList *)malloc(sizeof(graphAdjList));
    // 建立顶点表并分配内存
    newGraph->verticesList = (Vertex **)malloc(sizeof(Vertex *) * verticesNumber);
    memset(newGraph->verticesList, 0, sizeof(Vertex *) * verticesNumber);
    // 初始化大小和容量
    newGraph->size = 0;
    newGraph->capacity = verticesNumber;
    return newGraph;
}