mirror of
https://github.com/krahets/hello-algo.git
synced 2026-06-15 22:57:48 +08:00
55 lines
1.7 KiB
JavaScript
55 lines
1.7 KiB
JavaScript
/**
|
||
* File: graph_dfs.js
|
||
* Created Time: 2023-02-21
|
||
* Author: Zhuo Qinyue (1403450829@qq.com)
|
||
*/
|
||
|
||
const { Vertex } = require('../modules/Vertex');
|
||
const { GraphAdjList } = require('./graph_adjacency_list');
|
||
|
||
/* 深さ優先探索 */
|
||
// グラフを隣接リストで表し、指定した頂点の隣接頂点をすべて取得できるようにする
|
||
function dfs(graph, visited, res, vet) {
|
||
res.push(vet); // 訪問した頂点を記録
|
||
visited.add(vet); // この頂点を訪問済みにする
|
||
// この頂点のすべての隣接頂点を走査
|
||
for (const adjVet of graph.adjList.get(vet)) {
|
||
if (visited.has(adjVet)) {
|
||
continue; // 訪問済みの頂点をスキップ
|
||
}
|
||
// 隣接頂点を再帰的に訪問
|
||
dfs(graph, visited, res, adjVet);
|
||
}
|
||
}
|
||
|
||
/* 深さ優先探索 */
|
||
// グラフを隣接リストで表し、指定した頂点の隣接頂点をすべて取得できるようにする
|
||
function graphDFS(graph, startVet) {
|
||
// 頂点の走査順序
|
||
const res = [];
|
||
// 訪問済み頂点を記録するためのハッシュ集合
|
||
const visited = new Set();
|
||
dfs(graph, visited, res, startVet);
|
||
return res;
|
||
}
|
||
|
||
/* Driver Code */
|
||
/* 無向グラフを初期化 */
|
||
const v = Vertex.valsToVets([0, 1, 2, 3, 4, 5, 6]);
|
||
const edges = [
|
||
[v[0], v[1]],
|
||
[v[0], v[3]],
|
||
[v[1], v[2]],
|
||
[v[2], v[5]],
|
||
[v[4], v[5]],
|
||
[v[5], v[6]],
|
||
];
|
||
const graph = new GraphAdjList(edges);
|
||
console.log('\n初期化後のグラフは');
|
||
graph.print();
|
||
|
||
/* 深さ優先探索 */
|
||
const res = graphDFS(graph, v[0]);
|
||
console.log('\n深さ優先探索（DFS）の頂点列は');
|
||
console.log(Vertex.vetsToVals(res));
|