【笔试复盘】美团

1、给两串字符串A=‘A1,A2,…,An’; B=‘B1,B2,…,Bn’，将它们合并成’A1B1A2B2…AnBn’的形式
思路：直接遍历组合就行，O(n)

n = 4
A = 'SDFG'
B = 'YUIO'
res = list(zip(A,B))
ans = ''
for i in range(n):
    ans += res[i][0]+res[i][1]
print(ans)

2、给三个点的坐标，给一个信标到三个点的曼哈顿距离，求信标的具体位置。如果求出的位置有多个，则选择字典序最小的那一个。
思路：关键是根据给定的点坐标和曼哈顿距离求一个信标的备选位置集合，再比较三个点的信标备选集合重复的点坐标，如果有重复则选择字典序最小的那个坐标。

def candidate(loc,dis,n):
    candi = set()
    for dx in range(-dis,dis+1):
        dy_abs = dis - abs(dx)
        temp1 = (loc[0]+dx, loc[1]+dy_abs)
        if 0 < temp1[0] <= n and 0 < temp1[1] <= n:
            candi.add(temp1)
        temp2 = (loc[0]+dx, loc[1]-dy_abs)
        if 0 < temp2[0] <= n and 0 < temp2[1] <= n:
            candi.add(temp2)
    return list(candi)

n = 3
map_size = n
loc_1 = [2,1]
loc_2 = [2,2]
loc_3 = [2,3]
locs = [loc_1, loc_2, loc_3]
dis = [2,1,2]
each_candi = {}
use = None
flag = 1e9
for i in range(len(locs)):
    each_candi[i] = candidate(locs[i],dis[i],map_size)
    if len(each_candi[i]) < flag:
        use = i
        flag = len(each_candi[i])
out = []
nonuse = list(range(len(locs)))
nonuse.remove(use)
for el in each_candi[use]:
    if el in each_candi[nonuse[0]] and el in each_candi[nonuse[1]]:
        out.append(el)
out = sorted(out)
print(out[0])

3、小团做题，共有n题。没复习时，每个题有不同的概率做对，且每个题有不同的分值。但是为了得到更高的分数，小团准备复习，复习之后，被复习到的题100%做对。时间有限，小团只能复习m题。请问小团怎么安排复习可以得到最多的分数，这个分数是多少？输入n,m,原本每题做对的概率p，每题对应的分数a。输出复习后可以获得的最大分值。
思路：贪心。复习之后要在原来没复习的基础上获得更多的分值，自然而然需要复习那些gap更大的题目，gap=a_i*(1-p_i)，直到m题复习完。没复习到的题，则保持原来的分值。最后求和。

n,m = 2, 1
p = [0.89, 0.38 ]
q = [0.11, 0.62]
a = [445, 754]
improve = {i:q[i]*a[i] for i in range(n)}
prime = {i:p[i]*a[i] for i in range(n)}
sort_imp = sorted(improve.items(),key=lambda x:x[1],reverse=True)
sorce = 0
use = []
for i in range(m):
    sorce += a[sort_imp[i][0]]
    del prime[sort_imp[i][0]]
sorce += sum(list(prime.values()))
v = '{:.2f}'.format(sorce)
v = eval(v)
print(v)

4、有两个由数字组成的列表a、b，对应长度为n、m。为了使得这两个列表变得相同，有两种操作：（1）删除元素x，代价为abs(x）（2）使得a，b中的元素x,y变得一样，代价为abs(x-y).请问如何转换可以使得a,b变得一样，且最小代价是？输入n,m,a,b，输出最小代价。
思路：贪心。以更短的那个列表为基准（假设为b），那么先从a中寻找元素匹配b，使得修改代价最小，接着把a中多出来的元素删除。

def deal(a,b):
    count = 0
    for el in copy.deepcopy(b):
        if el in a:
            a.remove(el)
            b.remove(el)
    for i in range(len(b)):
        gap = 1e9
        pos = 0
        for j in range(len(a)):
            if abs(b[i]-a[j]) < gap:
                gap = abs(b[i]-a[j])
                pos = j
            if abs(b[i]-a[j]) > gap:
                break
        ela = a[pos]
        a.remove(ela)
        count += gap
    count += sum(a)
    return count
a = [1,3,4,7,9,10]#[-5,-3,0,1,1,6,8]
b = [0,3,5,11]#[-1,0,1,4,7]
a.sort()
b.sort()
n = len(a)
m = len(b)
if m <= n:
    count = deal(a,b)
else:
    count = deal(b,a)
print(count)