1. 程式人生 > >Python入門習題----生日相同

Python入門習題----生日相同

【問題描述】

在一個有180人的大班級中,存在兩個人生日相同的概率非常大,現給出每個學生的名字,出生月日。試找出所有生日相同的學生。  

【輸入】

第一行為整數n,表示有n個學生,n<=180。此後每行包含一個字串和兩個整數,分別表示學生的名字(名字第一個字母大寫,其餘小寫,不含空格,且長度小於20)和出生月(1<=m<= 12)日(1 <=d<=31)。名字、月、日之間用一個空格分隔

【輸出】

每組生日相同的學生,輸出一行,其中前兩個數字表示月和日,後面跟著所有在當天出生的學生的名字,數字、名字之間都用一個空格分隔。對所有的輸出,要求按日期從前到後的順序輸出。 對生日相同的名字,按名字從短到長按序輸出,長度相同的按字典序輸出。如果沒有任何生日相同的學生,輸出:None  。

【樣例輸入】

6
Avril 3 2
Candy 4 5
Tim 3 2
Sufia 4 5
Lagrange 4 5
Bill 3 2

【樣例輸出】

3 2 Tim Bill Avril
4 5 Candy Sufia Lagrange
【樣例說明】
樣例輸出中,3月2日有Tim, Bill, Avril三個人在當日出生。

參考程式碼


"""
Created on Mon Oct 22 14:53:45 2018

@author: Dina
"""

n = int(input())

#--------- 把輸入的學生資訊用列表儲存
studen_list=[]
for i in range(n):
    temp = input().split()
    studen_list.append(temp)
for i in range(n):
    studen_list[i][1]=int(studen_list[i][1])
    studen_list[i][2]=int(studen_list[i][2])

#---------  把 月份和日期 都轉換為 0102 這樣的格式,為了下面做鋪墊
for i in range(n):
    if studen_list[i][1]<10 :
        studen_list[i][1] = '0'+str(studen_list[i][1])
    if studen_list[i][2]<10 :
        studen_list[i][2] = '0'+str(studen_list[i][2])
    
#--------- 把學生的資訊用字典儲存,以生日為 key ,姓名為 value 
studen_dict ={}
for i in range(n):    
    day = str(studen_list[i][1])+ str(studen_list[i][2])
    if day in studen_dict :
        studen_dict[day] += studen_list[i][0]+','    
    else:
        studen_dict[day] = studen_list[i][0]+','
        

#--------- 排序,按照日期從小到達進行排序
a = sorted(studen_dict.items(),key = lambda x:x[0])

#--------- 輸出有生日相同的學生資訊
count = 0
for i in range(len(a)):   
    temp1 = a[i][1].split(',') #以逗號分隔姓名,temp1 的型別為列表
    temp1.remove('') 
    
    # 對學生的姓名進行排序先按名字從短到長排,在長度相同的情況按字典順序排
    temp1.sort(key=lambda x: (len(x), x))
    
    temp=''
    if len(temp1)>1 :         
        for v in temp1:            
            temp += v +" "           
        # a[i][0] 的型別是字串;a[i][0][0:2]是取出字串 a[i][0] 中的前兩位
        print(str(int(a[i][0][0:2])) +' '+str(int(a[i][0][2:4])) +' '+ temp)
    else:
        count +=1

if  count == n :
    print('None')

提交可通過: