leetcode——444. 序列重建

题目

1. 给定一个1~n的序列，判断其在seqs是否能被唯一重构
2. seqs为org的子序列，即部分路径

思路

1. seqs需要包含所有orgs中的结点 —— 数组记录
2. seqs不能包含orgs以外的结点 —— 结点索引范围判断
3. 需要唯一建立org —— 队列弹出每次的size为1
4. 需要与org相同 —— 一边队头弹出 一边扫描对比

代码

步骤：
1. 建图
2. 入队
3. 广度搜索

class Solution {
public:
    bool sequenceReconstruction(vector<int>& org, vector<vector<int>>& seqs) {
        //建图
        int n = org.size();
        vector<vector<int>> g(n + 1);//图
        vector<int> indegree(n + 1, 0);//入度
        vector<bool> exit(n + 1, false);//判断seqs是否包含所有org
        for (auto& s : seqs){
            for (int i = 0; i < s.size();++i) {
                if (s[i] <= 0 || s[i] > n) return false;//seqs包含orgs以外结点
                exit[s[i]] = true;
                if (i < s.size() - 1) {
                    g[s[i]].push_back(s[i + 1]);
                    ++indegree[s[i + 1]];
                }
            }
        }

        //入度为0结点 入队
        queue<int> q;
        for (int i = 1; i <= n; ++i){
            if (!exit[i]) return false;//seqs不包含所有orgs的结点
            if (indegree[i] == 0)
                q.push(i);
        }

        //广度优先遍历
        int i = 0;
        while (!q.empty()){
            //序列不能被唯一重建 或 重建与序列不匹配
            if (q.size() != 1 || q.front() != org[i++])
                return false;
            //队头弹出  新入度为0结点 入队
            int cur = q.front(); q.pop();
            for (int p:g[cur]){
                --indegree[p];
                if (indegree[p] == 0)
                    q.push(p);
            }
        }
        return i == n? true:false;
    }
};