魔法权值

魔法权值
给出 n 个字符串，对于每个 n 个排列 p，按排列给出的顺序(p[0] , p[1] … p[n-1])依次连接这 n 个字符串都能得到一个长度为这些字符串长度之和的字符串。所以按照这个方法一共可以生成 n! 个字符串。
一个字符串的权值等于把这个字符串循环左移 i 次后得到的字符串仍和原字符串全等的数量，i 的取值为 [1 , 字符串长度]。求这些字符串最后生成的 n! 个字符串中权值为 K 的有多少个。

注：定义把一个串循环左移 1 次等价于把这个串的第一个字符移动到最后一个字符的后面。

参考代码：

# -*- coding: utf-8 -*-
"""
Created on Sat Aug 06 19:10:02 2016
 
python测试通过
 
@author: duzejie
 
基本思想：
1.生成全排列
2.减少测试次数
a.如果字符串循环左移 i 次后得到的字符串仍和原字符串全等，
则要求i是该字符串长度的因数,所以不必测试全部i
b.对于权为k的字符串，测试通过的最小i为imin= strlen / k,
所以不必测试大于imin的i
c.测试通过的i小于imin= strlen / k，可提前退出
"""
import itertools
 
'''测试字符串循环左移 i 次后得到的字符串是否仍和原字符串全等'''
def test(s,i):
    strl = s[:i]
    strr = s[i:]
    newstr = strr + strl
    if cmp(s,newstr)==0:
        return True
         
lis = raw_input().split() #获取n,k
n=int(lis[0])
k=int(lis[1])
i = 0
mystring = [] #存放输入字符串列表
strlen = 0 #统计所有字符串长度
 
while(i<n):  #从用户输入获取各个字符串，放入mystring[]
    ms = raw_input()
    strlen += len(ms)
    mystring += [ms]
    i+=1
'''mybei[]中存放所有可能的i。如果字符串循环左移 i 次后得到的字符串仍和原字符串全等，
则要求i是该字符串长度的因数'''
mybei = [x for x in range(1,strlen+1) if strlen % x == 0]
 
  
coutk = 0
newlist = range(n)
strlist = list(itertools.permutations(newlist,n)) #存放所有字符串的可能组合方式
 
imin = strlen / k #对于权为k的字符串，测试通过的最小i为imin
 
for x in strlist : #遍历所有可能组合
    if strlen % k != 0:break
     
    s = ''
    for y in x:
        s+= mystring[y] #生成该组合的字符串
     
    for i in mybei: #对可能的i测试
        if i > imin :break
        if test(s,i):
            if i < imin:
                break
            if i == imin:
                coutk +=1
                break
 
print(coutk)