考研数据结构之topsort核心代码

代码

// 由于拓扑排序需要考虑到入度，因此需要把邻接表的数据结构做如下更改如下更改
struct vNode {
    int count; // 代表当前节点入度
    ArcNode *firstarc;
};

// 核心代码
int topsort(graph *G)
{
    // 将入度为0的节点放到队列中
    // 书中是放入栈，并无区别
    queue<int> que;
    ArcNode *p;
    int num = 0;

    // 1. 将当前图中所有入度为0的节点放入栈中
    for (int i=0; i<G->n; i++)
    {
        if (G->adjlist[i].count == 0)
        {
            que.push(i);
        }
    }

    while (!que.empty())
    {
        p = G->adjlist[que.front()].firstarc;
        que.pop();
        num ++;  // 计数器加一，统计当前已经移除节点数量

        while (p) // p指向的节点的入度均减去一，并且如果入度为零就放入队列中
        {
            j = p->adjvex;
            G->adjlist[j].count --;
            if (G->adjlist[j] == 0)
                que.push(j);
            p = p->next;
        }
    }

    if (num == n) // 当图中含有环路时，num ！= n，因此也可以用来判断图是否含有回路
        return 1;
    else
        return 0;
}