【华为OD题库-055】金字塔/微商-java

题目

微商模式比较典型,下级每赚100元就要上交15元,给出每个级别的收入,求出金字塔尖上的人收入。
输入描述
第一行输入N,表示有N个代理商上下级关系
接下来输入N行,每行三个数:代理商代号 上级代理商代号 代理商赚的钱
输出描述
输出一行,两个以空格分隔的整数,含义如下: 金字塔顶代理商 最终的钱数
示例1:
输入
3
1 0 223
2 0 323
3 2 1203
输出
0 105
说明
2的最终收入等于323+1203/10015=323+ 180
0的最终收入等于(323+ 180 + 223)/100
15= 105
示例2:
输入
4
1 0 100
2 0 200
3 0 300
4 0 200
输出
0 120

思路

题目未说明0一定就是顶级代理商,代理商之间的层级关系和编号大小无任何 关系。
使用两个map存放信息:
Map> proxyBusiness 存放代理商关系,key存代理商id,val存下线集合
Map pricesMap=new HashMap<>() 存每个代理商自己赚的钱
首先我们需要找到顶级代理商,在所有代理商中,如果该代理商自己没有赚钱,那么就是顶级代理商

因为题目明确说了,只能输出一行,且每行三个数的含义是:代理商代号 上级代理商代号 代理商赚的钱),如果顶级代理商要赚钱,假定其编号是4,那么其输入格式必然是:4 ? x,没有上级代理商,所以无法处理?号输入什么。因此,本文解题逻辑就是输入中不会输入顶级代理商自己赚的钱。
如果要处理含顶级代理商自己赚的钱的信息,可以假想输入是:4 4 x的格式,此时判断顶级代理商的逻辑可以考虑成:将输入第一列放入子代理商集合,输入第二行放入父代理商集合。遍历所有父代理商,如果其没有出现在子代理商的集合中,那么其是顶级代理商。

然后设计dfs递归函数:dfs(root),root代表传入的代理商编号,返回的值代表该代理商能够赚的钱

如果root没有子代理商,那么返回其自己赚的钱即可
如果有子代理商,遍历其所有子代理商,累加子代理商赚的钱sum
最后返回该代理商自己赚的前+sum/100*50即可

题解

package hwod;

import java.util.*;

public class MicroBusiness {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = Integer.parseInt(sc.nextLine());
        int[][] nums = new int[n][3];
        for (int i = 0; i < n; i++) {
            nums[i] = Arrays.stream(sc.nextLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        }
        int[] res = microBusiness(nums);
        System.out.println(res[0] + " " + res[1]);


    }
    private static Map<Integer, List<Integer>> proxyBusiness = new HashMap<>(); //存关系
    private static Map<Integer,Integer> pricesMap=new HashMap<>(); //每个代理商赚的钱
    private static int[] microBusiness(int[][] nums) {
        Set<Integer> set = new HashSet<>();//存放有哪些代理商
        for (int i = 0; i < nums.length; i++) {
            List<Integer> oldChild = proxyBusiness.getOrDefault(nums[i][1], new ArrayList<>());
            oldChild.add(nums[i][0]);
            proxyBusiness.put(nums[i][1], oldChild);
            pricesMap.put(nums[i][0], nums[i][2]);
            set.add(nums[i][1]);
            set.add(nums[i][0]);
        }
        int root = -1;
        //寻找顶级代理商,自己不赚钱的代理商
        for (Integer proxy : set) {
            if (!pricesMap.containsKey(proxy)) {
                root = proxy;
                break;
            }
        }
        int res=dfs(root);
        return new int[]{root,res};
    }

    private static int dfs(int root) {
        if(!proxyBusiness.containsKey(root)) return pricesMap.getOrDefault(root,0);
        List<Integer> subList = proxyBusiness.get(root);
        int sum = 0;
        for (int i = 0; i < subList.size(); i++) {
            Integer sub = subList.get(i);
            sum += dfs(sub);
        }
        return pricesMap.getOrDefault(root,0)+sum / 100 * 15;
    }
}


推荐

如果你对本系列的其他题目感兴趣,可以参考华为OD机试真题及题解(JAVA),查看当前专栏更新的所有题目。

你可能感兴趣的:(华为OD题库JAVA题解,华为od,java)