牛客网NC26-20.8.4-树

最新推荐文章于 2022-02-12 14:51:46 发布

原创最新推荐文章于 2022-02-12 14:51:46 发布 · 338 阅读

0 ·

CC 4.0 BY-SA版权

每日一题专栏收录该内容

13 篇文章

订阅专栏

本文解析了NC26算法题，介绍了群员等级、问题等级等关键概念，并提供了Java实现代码。通过深度优先搜索（DFS）算法解决群员间问题解答的计数问题。

题目链接：NC26
题意、输入、输出：
在这里插入图片描述

代码：

import java.util.*;


public class Solution {
    /**
     * 算法交流群
     * @param n int整型 群员个数
     * @param a int整型一维数组 群员的等级
     * @param p int整型一维数组 群友寻求帮助的人
     * @param k int整型一维数组 群友产生的问题等级
     * @return int整型一维数组
     */
    int[] cnt;
    int[] a;
    int[] p;
    public int[] solve (int n, int[] a, int[] p, int[] k) {
        // write code here
        cnt=new int[n];
        this.a=a;
        this.p=p;
        for(int i=0;i<n;i++){
           dfs(i,k[i]);
        }
        return cnt;
    }
    void dfs(int n,int q){
        if(q<=a[n]){
            cnt[n]++;
            return;
        }
        if(n==0){
            return;
        }
        dfs(p[n-1]-1,q);
    }
}