2018春招-今日头条笔试题-第四题（python）

题目描述：2018春招-今日头条笔试题5题（后附大佬答案-c++版）

#-*- coding:utf-8 -*-

class Magic:
    '''
        a:用于存储数组a
        b:用于存储数组b
        num:用于存储数组最后的结果
        flag:当已经没有可以移动的数据,flag置为1
    '''
    def __init__(self,a,b):
        self.a=a
        self.b=b
        self.num=0
        self.flag=0
    '''
        avg_a:存储数组a的平均值
        avg_b:存储数组b的平均值
        当avg_a>avg_b时，表示数组a中可以移动到b中，移动一个处于avg_b~avg_a之间的数，
        移动之后，重新计算avg_a、avg_b，看是否还有处于avg_b~avg_a之间的数，有，继续
        递归，没有，flag置为1；当avg_a<avg_b时同理。
    '''
    def fun(self):
        avg_a=float(sum(self.a)/(len(self.a)*1.0))#取浮点数
        avg_b=float(sum(self.b)/(len(self.b)*1.0))#取浮点数
        if self.flag==0:
            if avg_a>avg_b:
                self.num+=1
                self.get_num_a(avg_a,avg_b)
                self.fun()
            elif avg_b>avg_a:
                self.num+=1
                self.get_num_b(avg_a,avg_b)
                self.fun()
            else:
                return
        else:
            return

    #当avg_a>avg_b，执行该方法
    def get_num_a(self,avg_a,avg_b):
        for i in range(len(self.a)):
            if self.a[i]>=avg_b and self.a[i]<=avg_a:
                self.a.remove(self.a[i])
                self.b.append(self.a[i])
                break
            #最后一个数都不符合要求，表明其中没有在avg_b~avg_a之间的数
            if i==len(self.a)-1:
                self.flag=1
        return
    #当avg_b>avg_a，执行该方法
    def get_num_b(self,avg_a,avg_b):
        for i in range(len(self.b)):
            if self.b[i]>=avg_a and self.b[i]<=avg_b:
                self.b.remove(self.b[i])
                self.a.append(self.b[i])
                break
            # 最后一个数都不符合要求，表明其中没有在avg_b~avg_a之间的数
            if i==len(self.b)-1:
                self.flag=1
        return
if __name__=="__main__":
    n_m=raw_input()
    n=raw_input().split(' ')
    m=raw_input().split(' ')
    a=[int(i) for i in n]
    b=[int(j) for j in m]
    magic=Magic(a,b)
    magic.fun()
    print magic.num-1