面试宝典_Python.常规算法.0001.在圆圈和框框分别填入1~8数字各一次?

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2016-12-24 12:19:01
# @Author  : 李满满 (xmdevops@vip.qq.com)
# @Link    : http://xmdevops.blog.51cto.com/ # @Version : $Id$

from __future__ import absolute_import
# 说明: 导入公共模块
import pprint
import itertools
# 说明: 导入其它模块

def calculation(max_num):
result = []
combinations = itertools.permutations(
xrange(1, max_num + 1),
max_num
)
for item in combinations:
flag = True
for index in range(0, max_num, 2):
x = item[index]
y = item[index + 1]
z = item[0] if index == max_num - 2 else item[index + 2]
if abs(z - x) != y:
flag = False
break
if flag:
result.append(item)

return result

if __name__ == '__main__':
result = calculation(8)
pprint.pprint(result)