美团算法第一题(2021.10.17)

给定数字n表示为n种盒子,默认第1类盒子装1个糖果,第i种盒子可以装不同数量的盒子(嵌套)。
给定n和数组a,b:
a = int[n-1],b = int[n-1]
a[i]表示第i+1类盒子装下的盒子个数,b[i]表示是第几类盒子。
返回每一类盒子包含的糖果数量。

public class meituan1 {
    final int Mod = 1000000007;
//    a个数,b种类
    public int[] candy(int n,int a[],int b[]){
        int[] ans = new int[n];
        ans[0] = 1;
        int a_len = a.length;

        for (int i = 0; i < a_len; i++) {
            while(b[i]!=1)
            {
                a[i] *= a[b[i]-2]%Mod;
                b[i] = b[b[i]-2];
            }
            ans[i+1] = a[i];
        }
        return ans;
    }
}
#美团算法方向笔试##笔试题目##美团#
全部评论

相关推荐

10-11 17:30
湖南大学 C++
我已成为0offer的糕手:羡慕
点赞 评论 收藏
分享
贺兰星辰:不要漏个人信息,除了简历模板不太好以外你这个个人简介是不是太夸大了...
点赞 评论 收藏
分享
评论
2
收藏
分享
牛客网
牛客企业服务