8.15 携程ctrip笔试

最新推荐文章于 2024-11-09 21:51:27 发布

Microdust_

最新推荐文章于 2024-11-09 21:51:27 发布

阅读量179

点赞数

文章标签： python 算法

本文链接：https://blog.csdn.net/weixin_44161760/article/details/108029394

版权

8.15 携程ctrip笔试

由于笔试要求不能向外泄题，所以仅放代码以供参考。
笔者很惨，第一条调试没问题但是测试没过，第二条由于分析错误导致未能写完，结束后自己编写了测试用例，调试通过
1.

def  divingBoard(a, b, k):
    if k == 0:return []
    temp = set()
    for i in range(k+1):
        s = i*a + (k-i)*b
        temp.add(s)
    res = []
    while temp:
        res.append(temp.pop())
    return sorted(res)

在这里插入图片描述

def setNode(s):
    t = s.split('|')
    tree = []
    for i in t:
        node = i.split('^')
        tree.append(node)
    return tree

def dfs(root,cur,tree,res):
    if root[2] == 'END':
        cur += int(root[1])
        res.append(cur)
        cur -= int(root[1])
    else:
        cur += int(root[1])
        next = root[2].split(',')
        for n in next:
            for i in range(len(tree)):
                if tree[i][0] == n:
                    index = i
            dfs(tree[index],cur,tree,res)

def treeMaxValue(s):   
    tree = setNode(s)
    res = []
    dfs(tree[0],0,tree,res)
    return max(res)

s = 'HEAD^0^A,B,C|A^50^END|B^20^END|C^30^D,E|D^10^F|E^20^END|F^40^END'
print(treeMaxValue(s))