LeetCode第155场周赛：5200. 项目管理（拓扑排序）

5200. 项目管理

 显示英文描述 

我的提交返回竞赛

• 用户通过次数12
• 用户尝试次数27
• 通过次数12
• 提交次数45
• 题目难度Hard

公司共有 n 个项目和  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]

思路：我的大致思路是拓扑排序的一般思路，首先令入度为0的点先进入队列，这里需要注意只能进一个点，因为题目要求所有在同一个项目组的人需要紧挨着输出，之后我们便进行遍历即可。但是我的方法最后一个样例就是超时，很蛋疼呀啊啊啊啊。。。

坑后期再埋好了~~~

import java.awt.Paint;
import java.awt.Point;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Deque;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.Scanner;
import java.util.Set;
import java.util.Stack;
import java.util.TreeSet;
import java.util.Vector;



class Solution {
    public int[] sortItems(int n, int m, int[] group, List<List<Integer>> beforeItems) {
    	int[] ans=new int[n];
    	if(n+m==0) return ans;
    	
        Queue<Integer> q=new LinkedList<>();
        Vector<Integer>[] arr=new Vector[n];
        for(int i=0;i<n;i++) arr[i]=new Vector();
        
        int len=0;
        int[] inagree=new int[n];
        int[] groupnum=new int[n+n];
        int[] groupagree=new int[n+n];
        boolean[] flag=new boolean[n];
        
        for(int i=0;i<n;i++)
        	if(group[i]==-1)
        		group[i]=n+i;
        
        for(int i=0;i<n;i++)
        	groupnum[group[i]]++;
        
        for(List<Integer> var : beforeItems)
        {
        	for(int index=0;index<var.size();index++)
        	{
        		inagree[len]++;
        		arr[var.get(index)].add(len);
        		
        		if(group[len]!=group[var.get(index)])
        			groupagree[group[len]]++;
        	}
        	len++;
        }
        
        len=0;
        int nowGroup=-1;
        for(int i=0;i<n;i++)
        	if(inagree[i]==0 && groupagree[group[i]]==0)
        	{
        		q.add(i);
        		nowGroup=group[i];
        		flag[i]=true;
        		break;
        	}
        
        if(nowGroup==-1) return new int[0];
        
        while(!q.isEmpty())
        {
        	int now=q.poll();
        	nowGroup=group[now];
        	for(int i=0;i<arr[now].size();i++)
            {
        		inagree[arr[now].get(i)]--;
                if(nowGroup!=group[arr[now].get(i)])
                    groupagree[group[arr[now].get(i)]]--;
            }

        	groupnum[nowGroup]--;
	        if(groupnum[nowGroup]==0)
	        	nowGroup=-2;
        	ans[len++]=now;
        	if(!q.isEmpty())
        		continue;
        	for(int i=0;i<n;i++)
        	{
        		if(flag[i]) continue;
        		if(inagree[i]==0 && groupagree[group[i]]==0 &&  (group[i]==nowGroup || nowGroup==-2))
        		{
        			q.add(i);
        			flag[i]=true;
        			nowGroup=group[i];
        		}
        	}
        }
        
        for(int i=0;i<n;i++)
        	if(!flag[i])
        		return new int[0];
        return ans;
    }
}