【华为OD机试B卷】服务器广播、需要广播的服务器数量(C++/Java/Python)

avatar
作者
筋斗云
阅读量:2

题目

题目描述

服务器连接方式包括直接相连,间接连接。

AB直接连接,BC直接连接,则AC间接连接。

直接连接和间接连接都可以发送广播。

给出一个N*N数组,代表N个服务器,

matrix[i][j] == 1
则代表ij直接连接;不等于 1 时,代表ij不直接连接。

matrix[i][i] == 1

即自己和自己直接连接。matrix[i][j] == matrix[j][i]

计算初始需要给几台服务器广播, 才可以使每个服务器都收到广播。

输入

输入为N行,每行有N个数字,为01,由空格分隔,

构成N*N的数组,N的范围为 1 <= N <= 40

输出

输出一个数字,为需要广播的服务器的数量

用例一

输入

1 0 0 0 1 0 0 0 1 

输出

3 

说明

3 台服务器互不连接,所以需要分别广播这 3 台服务器

用例二

输入

1 1 1 1 

输出

1 

说明

2 台服务器相互连接,所以只需要广播其中一台服务器

实现代码

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) 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!