-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathsolve.js
56 lines (46 loc) · 1.17 KB
/
solve.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
const readline = require('readline');
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout
});
let T = -1;
let n = -1, m = -1, cnt;
let edges;
const bfs = (src, dst) => {
let queue = [[src, 0]];
let visited = new Array(n).fill(false);
while (queue.length !== 0) {
let [u, d] = queue.shift();
if (u === dst && d <= 6) {
return 'YES';
}
visited[u] = true;
for (let v of edges[u]) {
if (!visited[v]) {
queue.push([v, d + 1]);
}
}
}
return 'NO';
}
rl.on('line', line => {
if (T === -1) {
T = parseInt(line);
} else if (n === -1) {
[n, m] = line.trim().split(' ').map(x => parseInt(x));
edges = new Array(n).fill(null).map(_ => []);
cnt = 0;
} else if (cnt === m) {
let [src, dst] = line.trim().split(' ').map(x => parseInt(x));
console.log(bfs(src, dst));
n = -1;
T -= 1;
if (T === 0) {
rl.close();
}
} else {
let [u, v] = line.trim().split(' ').map(x => parseInt(x));
edges[u].push(v);
cnt += 1;
}
});