题目
题目描述:
快递业务范围有N个站点,A站点与B站点可以中转快递,则认为A-B站可达,如果A-B可达,B-C可达,则A-C可达。现在给N个站点编号0、1、…n-1,用s[i][j]表示i-j是否可达,s[i][j]=1表示i-j可达,s[i][j]=0表示i-j不可达。
现用二维数组给定N个站点的可达关系,请计算至少选择从几个主站点出发,才能可达所有站点(覆盖所有站点业务)。
说明:s[i][j]与s[j][i]取值相同。
输入描述:
第一行输入为N,N表示站点个数。
之后N行表示站点之间的可达关系,第i行第j个数值表示编号为i和j之间是否可达。
输出描述:
输出站点个数,表示至少需要多少个主站点。
补充说明:
1<N<10000
InsCode
java
搜索
(图像)
示例1
输入:
4
1 1 1 1
1 1 1 0
1 1 1 0
1 0 0 1
输出:
1
说明:
选择0号站点作为主站点,0站点可达其他所有站点,所以至少选择1个站点作为主站才能覆盖所有站点业务。
示例2
输入:
4
1 1 0 0
1 1 0 0
0 0 1 0
0 0 0 1
输出:
3
说明:
选择0号站点可以覆盖0、1站点,选择2号站点可以覆盖2号站点,选择3号站点可以覆盖3号站点,所以至少选择3个站点作为主站才能覆盖所有站点业务。
并查集
function find(root, id) {
while (id !== root[id]) {
id = root[id];
}
return id;
}
function union(roots, a, b) {
let rootA = find(roots, a);
let rootB = find(roots, b);
roots[rootA] = rootB;
}
function deal(arr) {
let roots = {};
let n = Number(arr[0]);
let station = arr.slice(1, arr.length).map((line) => {
return line.split(' ').map((v) => {
return Number(v);
})
});
// console.log(JSON.stringify(station));
for (let i = 0; i < n; i++){
roots[i] = i;
}
for (let i = 0; i < n; i++) {
for (let j = 0; j < n; j++) {
if (station[i][j] === 1) {
union(roots, i, j);
}
}
}
for (let i = 0; i < n; i++){
let r = find(roots,i);
roots[i] = r;
}
console.log(JSON.stringify(roots));
}
let r1 = [
"4",
"1 1 1 1",
"1 1 1 0",
"1 1 1 0",
"1 0 0 1"
];
deal(r1);
let r2 = [
"4",
"1 1 0 0",
"1 1 0 0",
"0 0 1 0",
"0 0 0 1"
];
deal(r2);