图算法-拓扑排序(参考GTA)

拓扑排序用途

拓扑排序用于解决图中是否存在环的问题

拓扑排序思想

1. 初始化图和图中所有节点的入度
2. 找出所有入度为0的节点加入队列
3. 列首出队，将该节点指向的所有节点入度-1
   如果-1后节点入度为0，则加入队列
4. 重复3，遍历图中所有节点
5. 如果图中还存在入度不为0的节点，则图中存在环。

例题

• 例题：LeetCode 457 环形数组是否存在循环
  存在一个不含 0 的 环形 数组 nums ，每个 nums[i] 都表示位于下标 i 的角色应该向前或向后移动的下标个数：

  如果 nums[i] 是正数，向前 移动 nums[i] 步
  如果 nums[i] 是负数，向后 移动 nums[i] 步
  因为数组是 环形 的，所以可以假设从最后一个元素向前移动一步会到达第一个元素，而第一个元素向后移动一步会到达最后一个元素。

  数组中的 循环 由长度为 k 的下标序列 seq ：

  遵循上述移动规则将导致重复下标序列 seq[0] -> seq[1] -> … -> seq[k - 1] -> seq[0] -> …
  所有 nums[seq[j]] 应当不是 全正 就是 全负
  k > 1
  如果 nums 中存在循环，返回 true ；否则，返回 false 。

  示例 1：
  输入：nums = [2,-1,1,2,2]
  输出：true
  解释：存在循环，按下标 0 -> 2 -> 3 -> 0 。循环长度为 3 。

  示例 2：
  输入：nums = [-1,2]
  输出：false
  解释：按下标 1 -> 1 -> 1 … 的运动无法构成循环，因为循环的长度为 1 。根据定义，循环的长度必须大于 1 。
  示例 3:
  输入：nums = [-2,1,-1,-2,-2]
  输出：false
  解释：按下标 1 -> 2 -> 1 -> … 的运动无法构成循环，因为 nums[1] 是正数，而 nums[2] 是负数。
  所有 nums[seq[j]] 应当不是全正就是全负。

拓扑排序用于此题的缺陷

拓扑排序不能解决自环问题，本题需要忽略自环，因此要进行特殊处理

基于拓扑排序的解决方案

1. 建图：点 i 可以移动到 i + nums[i]的位置，由于给定的是循环数组，所以我们以 i 为有向边起点，利用公式 end = ((i + nums[i]) % n + n) % n得到终点建图。
2. 忽略自环：若某一有向边起点终点相等（i == end），则说明该点有自环，直接忽略，不存入图中。
3. 同向环路：题目要求环路同向，所以我们可以分别处理 nums[i] > 0 和 nums[i] < 0 的情况，分别建图，并进行拓扑排序。
4. 代码中，将拓扑排序的过程提取到了函数 toposort() 中。

代码

class Solution {
public:
    vector<vector<int>> graph;
    vector<int> deg;
    
    bool topoSort(int n){
        queue<int> q;
        for(int i=0;i<n;i++){
            if(!deg[i]) q.push(i);
        }

        while(!q.empty()){
            int cur=q.front();
            q.pop();

            for(int i=0;i<graph[cur].size();i++){
                int end=graph[cur][i];
                deg[end]--;

                if(!deg[end]) q.push(end);
            }
        }

        for(int i=0;i<deg.size();i++){
            //如果仍有入度不为0的点，则存在环
            if(deg[i]!=0) return true;
        }
        return false;
    }

    bool circularArrayLoop(vector<int>& nums) {
        int n=nums.size();
        graph.resize(n);
        deg.resize(n);

        //处理正向边
        for(int i=0;i<n;i++){
            int end=((i+nums[i])%n+n)%n;
            if(nums[i]<0 || end==i) continue;
            graph[i].push_back(end);
            deg[end]++;
        }

        if(topoSort(n)) return true;

        graph.clear();
        graph.resize(n);
        deg.clear();
        deg.resize(n);

        //处理负向边
        for(int i=0;i<n;i++){
            int end=((i+nums[i])%n+n)%n;
            if(nums[i]>0 || end==i) continue;
            graph[i].push_back(end);
            deg[end]++;
        }

        if(topoSort(n)) return true;
        return false;
    }
};