飘逸的python - 赛程表算法

时间:2023-01-05 04:47:07

最近德甲英超西甲各大联赛重燃战火,想起之前写过的一段生成赛程表的代码,用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 deque
import random
def 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 scheduleobj if __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)

飘逸的python - 赛程表算法