蓝桥杯 ALGO-48 关联矩阵 java

问题描述

问题描述
　　有一个n个结点m条边的有向图，请输出他的关联矩阵。
输入格式
　　第一行两个整数n、m，表示图中结点和边的数目。n<=100,m<=1000。
　　接下来m行，每行两个整数a、b，表示图中有(a,b)边。
　　注意图中可能含有重边，但不会有自环。
输出格式
　　输出该图的关联矩阵，注意请勿改变边和结点的顺序。
样例输入
5 9
1 2
3 1
1 5
2 5
2 3
2 3
3 2
4 3
5 4
样例输出
1 -1 1 0 0 0 0 0 0
-1 0 0 1 1 1 -1 0 0
0 1 0 0 -1 -1 1 -1 0
0 0 0 0 0 0 0 1 -1
0 0 -1 -1 0 0 0 0 1

参考代码

package 关联矩阵;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

public class Exam {
public static void main(String[] args) {
	Scanner sr = new Scanner(System.in);
	int n = sr.nextInt();	
	int m = sr.nextInt();
	//收集所有的顶点
	ArrayList<Integer> nodes = new ArrayList<Integer>();
	int[][] map = new int[m][2];//接收输入,a-b
	int[][] regs = new int[n][m];//关联矩阵
	//生成顶点集合
	for (int j = 0; j < m; j++) {
		int a = sr.nextInt();
		int b = sr.nextInt();
		map[j][0] = a;
		map[j][1] = b;
		if(!nodes.contains(a))nodes.add(a);
		if(!nodes.contains(b))nodes.add(b);
	}
	
	Collections.sort(nodes);
	//生成关联矩阵
	for (int i = 0; i < map.length; i++) {
		int a = nodes.indexOf(map[i][0]);
		int b = nodes.indexOf(map[i][1]);
		regs[a][i] = 1;//a-b连着i这个线,ai就是1
		regs[b][i] = -1;//a-b连着i这个线,bi就是-1
	}
	//输出
	for (int i = 0; i < regs.length; i++) {
		for (int j = 0; j < regs[0].length; j++) {
			System.out.print(regs[i][j]+" ");
		}System.out.println();
	}
}
}