小明的实验室有N台电脑,编号1~N。原本这N台电脑之间有N-1条数据链接相连,恰好构成一个树形网络。在树形网络上,任意两台电脑之间有唯一的路径相连。
不过在最近一次维护网络时,管理员误操作使得某两台电脑之间增加了一条数据链接,于是网络中出现了环路。环路上的电脑由于两两之间不再是只有一条路径,使得这些电脑上的数据传输出现了BUG。
为了恢复正常传输。小明需要找到所有在环路上的电脑,你能帮助他吗?
样例输入
5
1 2
3 1
2 4
2 5
5 3
样例输出
1 2 3 5
思路:这题就是用拓扑排序,因为这是一个有向图,拓扑排序只能用在有向图,通过拓扑排序去判断环,注意,有的节点虽然在环上,但是它不参与闭合,那么就是它有入度,但是出度为0,那么只需要再记录一下每个结点的出度,然后在最后不把没有出度的节点加入环汇总就能拿到正确答案了。
代码如下:注解详细
static int MAX = Integer.MAX_VALUE;
public static void main(String[] args)
{
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[][] e = new int[n][n];
for (int i = 0; i < n; i++)
{
int a = in.nextInt();
int b = in.nextInt();
e[a-1][b-1] = 1;
}
for (int i = 0; i < e.length; i++)
{
for (int j = 0; j < e[i].length; j++)
{
if(i!=j && e[i][j]!=1)
e[i][j]=MAX;
}
}
System.out.println(topSort(n,e));
}
public static List<Integer> topSort(int n, int[][] e) {
LinkedList<Integer> path = new LinkedList<>();
LinkedList<Integer> ring = new LinkedList<>();
int[] inDegree = new int[n];
int[] outDegree = new int[n];
//获取出度
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
if(e[i][j]==1)
outDegree[i] += e[i][j];
}
}
//获取入度
for(int j=0;j<n;j++)
{
for(int i=0;i<n;i++)
{
if(e[i][j]==1)
inDegree[j] += e[i][j];
}
}
Queue<Integer> q = new LinkedList<>();
//将所有入度为0的节点加入队列
for (int i = 0; i < n; i++) {
if (inDegree[i] == 0)
q.offer(i);
}
while (!q.isEmpty()) {
int cur = q.poll();
path.add(cur);
for(int j=0;j<n;j++)
{
if(e[cur][j]!=MAX)
e[cur][j] = 0;
if(e[cur][j]!=MAX && inDegree[j]!=0)
{
inDegree[j]--;
if(inDegree[j]==0)
q.offer(j);
}
}
}
//这里去掉出度为0的值
if(path.size()!=n)
{
for (int i = 0; i < n; i++)
{
if(!path.contains(i) && outDegree[i]!=0)
ring.add(i+1);//输出的值加1
}
}
return path.size() == n ? path : ring;
}