题目
给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环
所有边的长度都是 1,点的编号为 1∼n
请你求出 1 号点到 n 号点的最短距离,如果从 1 号点无法走到 n 号点,输出 −1
涉及数组模拟单链表和队列
可以看我之前的文章:
java 数组模拟队列
java 数组模拟单链表
输入:
4 5
1 2
2 3
3 4
1 3
1 4
输出:
1
public class 树与图的BFS广度优先遍历_图中点的层次 {
//初始化,n和m记得在外面开,要不然调不到
//h存的是所有链表的头,e存的是节点的元素,ne存的是节点的指向
//q是模拟queue队列,d是存储初始节点到当前节点的距离
public static int N = 1000010, M = N*2, index = 0, n, m;
public static int[] h = new int[N], e = new int[M], ne = new int[M];
public static int[] q = new int[N], d = new int[M];
public static void main(String[] args) {
//初始化,将所有的链表头都设为-1, n是要查询的节点,m是进行几次操作
Scanner in = new Scanner(System.in);
n = in.nextInt(); m = in.nextInt();
Arrays.fill(h, -1);
for (int i = 0; i<m; i++){
int x = in.nextInt(), y = in.nextInt();
insert(x, y); //将x指向y
}
System.out.println(bfs());
}
public static int bfs() {
//hh和tt是模拟队列的头和尾,将1号节点放入队列中
//d数组也就是距离初始设置为-1,1号节点设置为0,自己到自己就没有距离嘛~
int hh = 0, tt = 0;
q[0] = 1;
Arrays.fill(d, -1);
d[1] = 0;
//循环队列不为空,每次都将队头节点拿出来看看,遍历这个节点的链表,也就是这个节点都和哪些节点有边可以走
//如果发现有可以走的,那就过去然后将那个点标记为走过了,顺便给那个点赋值距离,也就是上个点的距离+1
while (hh <= tt) {
int t = q[hh++];
for (int i = h[t]; i!=-1; i = ne[i]) {
int j = e[i];
if (d[j]==-1) {
d[j] = d[t] + 1;
q[++tt] = j;
}
}
}
//最后返回n节点到初始节点的距离
return d[n];
}
//单链表模拟中的头节点插入
public static void insert(int x, int y) {
e[index] = y;
ne[index] = h[x];
h[x] = index++;
}
}
dfs感觉还是比bfs难啊,昨天那个题真的花了不少时间,今天轻轻松松
声明:算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流