阿里2020.4.13实习笔试题——蚂蚁森林之王

爱学习的程序员

于 2021-03-15 14:43:14 发布

阅读量909

点赞数

CC 4.0 BY-SA版权

分类专栏：编程算法实习

本文链接：https://blog.youkuaiyun.com/csdn0123zl/article/details/114832966

编程同时被 3 个专栏收录

34 篇文章

订阅专栏

算法

29 篇文章

订阅专栏

实习

13 篇文章

订阅专栏

本文探讨了一种特殊的投票机制，在森林中的动物们通过直接或间接的方式进行投票选择森林之王的过程。投票遵循动物之间的崇拜关系，形成树状结构。文章通过递归算法计算每个动物可能获得的最大票数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

森林里有 $n$ 只动物，编号从1到 $n$ ，编号越小的动物能力越大。他们想投票选出森林之王，每只小动物可以投票选自己，或者与自己的偶像投票相同；如果没有偶像，就只能投自己。已知小动物是否有偶像以及具体是谁，求每个人能够得到的最高票数是多少？

问题分析

崇拜的关系是传递的，如果崇拜关系当做树的一条边，题目所求即以 $i$ 为根的某棵树的节点数。此外，因为动物顺序是按照能力从大到小排序，所以后向遍历有 $a n s [A [i]] + = a n s [i]$

代码示例

n = int(input())
ans = [0] * (n + 1)
animal_list = [-1] + list(map(int, input().split()))

for i in range(n, 0, -1):
    ans[i] += 1
    ans[animal_list[i]] += ans[i]

for i in range(1, n + 1):
    print(ans[i])