【华为OD机试B卷】服务器广播、需要广播的服务器数量(C++/Java/Python)
【华为OD机试B卷】服务器广播、需要广播的服务器数量(C++/Java/Python)
题目
实现代码
c++
#include <iostream>
#include <vector>
using namespace std;
int count = 0;
void dfs(vector<vector<int>>& arr, vector<bool>& visited, int index) {
visited[index] = true;
bool flag = true;
for (int i = index + 1; i < arr.size(); i++) {
if (arr[index][i] == 1) {
flag = false;
dfs(arr, visited, i);
}
}
if (flag) {
count++;
}
}
int main() {
string input;
getline(cin, input);
vector<string> str;
size_t pos = 0;
while ((pos = input.find(" ")) != string::npos) {
str.push_back(input.substr(0, pos));
input.erase(0, pos + 1);
}
str.push_back(input);
int n = str.size();
vector<vector<int>> arr(n, vector<int>(n, 0));
for (int i = 0; i < n; i++) {
arr[0][i] = stoi(str[i]);
}
for (int i = 1; i < n; i++) {
getline(cin, input);
pos = 0;
vector<string> s;
while ((pos = input.find(" ")) != string::npos) {
s.push_back(input.substr(0, pos));
input.erase(0, pos + 1);
}
s.push_back(input);
for (int j = 0; j < n; j++) {
arr[i][j] = stoi(s[j]);
}
}
vector<bool> visited(n, false);
for (int i = 0; i < n; i++) {
if (!visited[i]) {
dfs(arr, visited, i);
}
}
cout << count << endl;
return 0;
}
java
import java.util.*;
public class main {
public static void main(string[] args) {
scanner in = new scanner(system.in);
string[] str = in.nextline().split(" ");
int n = str.length;
int[][] arr = new int[n][n];
for(int i = 0; i < n; i++) {
arr[0][i] = integer.parseint(str[i]);
}
for(int i = 1; i < n; i++) {
string[] s = in.nextline().split(" ");
for(int j = 0; j < n; j++) {
arr[i][j] = integer.parseint(s[j]);
}
}
int count = 0;
queue<integer> queue = new linkedlist<>();
for(int i = 0; i < n; i++) {
if(!queue.contains(i)) {
dfs(arr, queue, i);
count++;
}
}
system.out.println(count);
}
public static void dfs(int[][] arr, queue<integer> queue, int index) {
queue.offer(index);
for (int i = index + 1; i < arr.length; i++) {
if (arr[index][i] == 1 && !queue.contains(i)) {
dfs(arr, queue, i);
}
}
}
}
python
import sys
def dfs(arr, visited, index):
visited[index] = true
flag = true
for i in range(index + 1, len(arr)):
if arr[index][i] == 1:
flag = false
dfs(arr, visited, i)
if flag:
global count
count += 1
count = 0
str = input().split(" ")
n = len(str)
arr = [[0]*n for _ in range(n)]
for i in range(n):
arr[0][i] = int(str[i])
for i in range(1, n):
s = input().split(" ")
for j in range(n):
arr[i][j] = int(s[j])
visited = [false]*n
for i in range(n):
if not visited[i]:
dfs(arr, visited, i)
print(count)
相关文章:
-
-
一、treeset比较常见形式treeset可以对元素按照某种规则进行排序:public class test { public static void main(strin…
-
-
-
-
版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。
发表评论