每天都在嘟嘟嘟 level
获赞
3
粉丝
2
关注
7
看过 TA
19
华中师范大学
2025
Java
IP属地:湖北
暂未填写个人简介
私信
关注
0 点赞 评论 收藏
分享
2024-09-07 22:34
华中师范大学 Java
做了半天愣是0%,发帖记录(Java)节点类:class Node {int val;List<Node> neighbor = new ArrayList<>();boolean visited;}思路:1. 先获取到所有边,存到节点数组中;2. 已知根节点为“1”,使用广度优先或者层式遍历方式遍历;3. 遍历时注意判断当前节点的neighbor是否已被访问(即为父节点),排除父节点后则可知子节点数量;代码实践:import java.util.ArrayList;import java.util.HashSet;import java.util.List;import java.util.Set;import java.util.stream.Collectors;class Node {int val;List<Node> neighbor = new ArrayList<>();public Node(int val) {this.val = val;}}public class T2 {public static void main(String[] args) {//        Scanner in = new Scanner(System.in);//        int T = in.nextInt();int T = 1;int[] ns = new int[]{7};int[][] bs = new int[][]{{1, 2},{1, 3},{3, 5},{3, 7},{2, 4},{2, 6}};for (int i = 0; i < T; i++) {long res = 0;//            int n = in.nextInt();int n = ns[i];Node[] nodeArr = new Node[n];for (int j = 0; j < n - 1; j++) {int u = bs[j][0];int v = bs[j][1];if (nodeArr[u - 1] == null) {nodeArr[u - 1] = new Node(u - 1);}if (nodeArr[v - 1] == null) {nodeArr[v - 1] = new Node(v - 1);}nodeArr[u - 1].neighbor.add(nodeArr[v - 1]);nodeArr[v - 1].neighbor.add(nodeArr[u - 1]);}int[] count = new int[3];count[nodeArr[0].neighbor.size()]++;Set<Integer> visited = new HashSet<>();visited.add(nodeArr[0].val);List<Node> curNodes = nodeArr[0].neighbor;List<Node> tmpNodes = new ArrayList<>();while (curNodes.size() > 0) {for (Node node : curNodes) {visited.add(node.val);List<Node> nextNodes = node.neighbor.stream().filter(nodeTmp -> !visited.contains(nodeTmp.val)).collect(Collectors.toList());int childNum = nextNodes.size();count[childNum]++;tmpNodes.addAll(nextNodes);}curNodes = tmpNodes;tmpNodes = new ArrayList<>();}for (int c: count) {res += getPairCount(c);}System.out.println(res);}}static long getPairCount(int n) {if (n <= 1) {return 0;}return (long) n * (n - 1) / 2;}}
投递美团等公司9个岗位
0 点赞 评论 收藏
分享
2021-06-27 11:56
华中师范大学 Java
0 点赞 评论 收藏
分享
2021-01-06 16:17
华中师范大学 Java
0 点赞 评论 收藏
分享

创作者周榜

更多
关注他的用户也关注了:
牛客网
牛客网在线编程
牛客网题解
牛客企业服务