当前位置: 代码网 > it编程>编程语言>Java > 【华为OD机试B卷】服务器广播、需要广播的服务器数量(C++/Java/Python)

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

2024年08月01日 Java 我要评论
【华为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)
(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com