python-新生舞会

解题思路:

开始想的是用对象解决，但是查询遍历的时间复杂度将达到O(n²)，想想便放弃了，然后采用字典的方法，时间复杂度为O(n)。

建立两个字典

第一个字典键值为名字，值为性别

第二个字典键值为学号，值为性别

然后查询即可。

注意事项:

参考代码:

def f(n):  
    A = dict()  
    B = dict()  
    for i in range(n):  
        name,stunum,sex = input().strip().split()  
        A.update({name:sex})  
        B.update({stunum:sex})  
  
    m = int(input().strip())  
    for i in range(m):  
        stu1,stu2 = input().strip().split()  
        try:  
            sex1 = A[stu1]  
        except:  
            sex1 = B[stu1]  
          
        try:  
            sex2 = A[stu2]  
        except:  
            sex2 = B[stu2]  
  
        if sex1 == sex2:  
            print('N')  
        else:  
            print('Y')  
  
  
if __name__ == '__main__':  
    n = int(input().strip())  
    f(n)