飄逸的python – 賽程表演算法

最近德甲英超西甲各大聯賽重燃戰火，想起之前寫過的一段產生賽程表的代碼，用python來寫這類東西太舒服了。

這個演算法叫做蛇環演算法。

即，把所有球隊排成一個環形(2列)，左邊對陣右邊，第一支隊伍不動，其他隊伍順時針迴圈，這樣就肯定不重複了。
為了方便說明，假設有8支球隊a到h。像下面那樣按環形排好。
a h
|   |
b g
|   |
c  f
|   |
d-e
這樣，第1輪的對陣就是，(a,h)(b,g)(c,f)(d,e)。
下一輪的時候，第一支球隊a不動，其它球隊像齒輪一樣順時針走一格。
a b
|   |
c h
|   |
d g
|   |
e-f
這樣，第2輪的對陣就是，(a,b)(c,h)(d,g)(e,f)。
齒輪繼續滑動，直到回到原點，這樣每支球隊都跟其它所有7支球隊對陣了。

千言不如一碼。拿英超為例。

from collections import dequeimport randomdef build_schedule(_teamarr):    scheduleobj = dict.fromkeys(range(1,20))    fixpos = _teamarr[0]    ring = _teamarr[1:]    ring = deque(ring)    #前半賽程，1-19輪(round)    for round in range(1,20):        #第1支球隊不動，再加上輪轉(rotate)的環        teams = [fixpos] + list(ring)        #切成2列，左邊主隊，右邊客隊        home, away = teams[:len(teams)/2],teams[len(teams)/2:]        away = away[::-1]        #隨機打亂主客隊        scheduleobj[round] = [(x,y) if random.random()>=0.5 else (y,x) for x,y in zip(home,away)]        ring.rotate(1)    #後半賽季對陣跟前半賽季一樣，但主客隊對調    for round in range(20,39):        scheduleobj[round] = [(y,x) for x,y in scheduleobj[round-19]]    return scheduleobjif __name__ == '__main__':    teamarr = [u'曼聯', u'阿斯頓維拉', u'切爾西', u'西漢姆', u'富勒姆',                u'熱刺', u'利物浦', u'南安普頓', u'埃弗頓', u'諾維奇',                u'紐卡斯爾', u'曼城', u'斯托克城', u'桑德蘭', u'水晶宮',                u'西布羅姆維奇', u'阿森納', u'赫爾城', u'卡迪夫城', u'斯旺西']    scheduleobj = build_schedule(teamarr)    print u'---聯賽第1輪---'    for h,a in scheduleobj[1]:        print u'{}-{}'.format(h,a)    print u'---聯賽第2輪---'    for h,a in scheduleobj[2]:        print u'{}-{}'.format(h,a)