文章目录
- 1. 题目
- 2. 解题
1. 题目
公司共有 n 个项目和 m 个小组,每个项目要不无人接手,要不就由 m 个小组之一负责。
group[i] 表示第 i 个项目所属的小组,如果这个项目目前无人接手,那么 group[i] 就等于 -1。(项目和小组都是从零开始编号的)小组可能存在没有接手任何项目的情况。
请你帮忙按要求安排这些项目的进度,并返回排序后的项目列表:
- 同一小组的项目,排序后在列表中彼此相邻。
- 项目之间存在一定的依赖关系,我们用一个列表 beforeItems 来表示,其中 beforeItems[i] 表示在进行第 i 个项目前(位于第 i 个项目左侧)应该完成的所有项目。
如果存在多个解决方案,只需要返回其中任意一个即可。
如果没有合适的解决方案,就请返回一个 空列表 。
示例 1:
输入:n = 8, m = 2,
group = [-1,-1,1,0,0,1,0,-1],
beforeItems = [[],[6],[5],[6],[3,6],[],[],[]]
输出:[6,3,4,1,5,2,0,7]
示例 2:输入:n = 8, m = 2,
group = [-1,-1,1,0,0,1,0,-1],
beforeItems = [[],[6],[5],[6],[3],[],[4],[]]
输出:[]
解释:与示例 1 大致相同,但是在排序后的列表中,4 必须放在 6 的前面。提示:
1 <= m <= n <= 3 * 10^4
group.length == beforeItems.length == n
-1 <= group[i] <= m - 1
0 <= beforeItems[i].length <= n - 1
0 <= beforeItems[i][j] <= n - 1
i != beforeItems[i][j]
beforeItems[i] 不含重复元素
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sort-items-by-groups-respecting-dependencies
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
拓扑排序学习、及相关题目
- 两次拓扑排序即可
class Solution {
public:vector<int> sortItems(int n, int m, vector<int>& group, vector<vector<int>>& beforeItems) {for(int i = 0; i < n; i++){if(group[i] == -1)group[i] = m++;//无人接管的分配一个虚拟团队号}vector<vector<int>> itemgraph(n);vector<vector<int>> groupgraph(m);vector<int> itemIndegree(n, 0);vector<int> groupIndegree(m, 0);for(int i = 0; i < n; i++){for(auto j : beforeItems[i]){itemgraph[j].push_back(i);//建图itemIndegree[i]++;//记录出入度if(group[i] != group[j]) // 注意条件{ // 团队也建图,记录出入度groupgraph[group[j]].push_back(group[i]);groupIndegree[group[i]]++;}}}vector<vector<int>> g_items(m);// item 拓扑排序queue<int> q;for(int i = 0; i < n; i++)if(itemIndegree[i] == 0)q.push(i);int countItem = 0;while(!q.empty()){int i = q.front();q.pop();countItem++;g_items[group[i]].push_back(i);//每个item顺序存入自己的团队for(auto j : itemgraph[i]){if(--itemIndegree[j]==0)q.push(j);}}if(countItem != n)return {};// 团队拓扑排序vector<int> g_order;for(int i = 0; i < m; i++)if(groupIndegree[i] == 0)q.push(i);int countgroup = 0;while(!q.empty()){int g = q.front();q.pop();countgroup++;g_order.push_back(g);for(auto j : groupgraph[g]){if(--groupIndegree[j]==0)q.push(j);}}if(countgroup != m)return {};// 写入答案vector<int> ans(n);int idx = 0;for(auto g : g_order){for(auto i : g_items[g])ans[idx++] = i;}return ans;}
};
200 ms 43 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!