python小练习（099）：etch-a-scetch（三）

jerryxjr1220

接上一期的python小练习

有了之前的那些函数以后，主程序就相对好写了。
不断交替进行横行和纵行的运算，剔除不可行的候选排列，最终剩下的就是答案。

完整程序：

1. # -*- coding: utf-8 -*-
   
2. """
   
3. Created on Tue May  9 20:57:18 2017
   
4. 
   
5. @author: XuJ06
   
6. """
   
7. 
   
8. def getdata(file):
   
9.     f = open(file)
   
10.     flag = -1
    
11.     size = []
    
12.     hor = []
    
13.     ver = []
    
14.     temp = [size,hor,ver]
    
15.     for i in f.readlines():
    
16.         if i[0]=='#':
    
17.             flag+=1
    
18.         elif i[0]=='\n':
    
19.             pass
    
20.         else:
    
21.             temp[flag].append([int(e) for e in i.split()])
    
22.     return (size[0][0],size[0][1],hor,ver)
    
23. 
    
24. def candidate(num,maxlen):
    
25.     '''''针对给出的排列方式，递归获得所有的备选排列 点亮为# '''
    
26.     candi = []
    
27.     for i in range(maxlen+2-sum(num)-len(num)):
    
28.         cans = '_'*i + '@'*num[0]
    
29.         if len(num)==1:
    
30.             tail = '_'*(maxlen-len(cans))
    
31.             candi.append(cans+tail)
    
32.         else:
    
33.             tail = ['_'+j for j in candidate(num[1:], maxlen-len(cans)-1)]
    
34.             candi.extend([cans + j for j in tail])
    
35.     return candi
    
36. 
    
37. def findinit(x,y,hnum,vnum):
    
38.     '''''找出每一行和每一列的备选排列'''
    
39.     candidateH = []
    
40.     candidateV = []
    
41.     for i in range(y):
    
42.         candidateH.append(candidate(hnum[i],x))
    
43.     for i in range(x):
    
44.         candidateV.append(candidate(vnum[i],y))
    
45.     return candidateH, candidateV
    
46. 
    
47. def check(candi,pos,flag):
    
48.     '''''对某一行（列）在给定位置和给定状态选择不冲突的备选方案'''
    
49.     newcandi = []
    
50.     for i in candi:
    
51.         if i[pos]== flag:
    
52.             newcandi.append(i)
    
53.     return newcandi
    
54. 
    
55. def solver(file):
    
56.     xsize,ysize,hnum,vnum = getdata(file)
    
57.     candiH, candiV = findinit(xsize,ysize,hnum,vnum)
    
58.     result = []
    
59.     for i in range(xsize):
    
60.         result.append([' ' for j in range(ysize)])
    
61.     count = 0
    
62.     target = xsize*ysize
    
63.     while count < target:
    
64.         for index in range(xsize):#先针对每一行
    
65.             i = candiH[index]
    
66.             if i == 'done':
    
67.                 continue
    
68.             elif len(i) == 1:
    
69.                 for j in range(ysize):
    
70.                     if result[index][j] == ' ':
    
71.                         count+=1
    
72.                         result[index][j] = i[0][j]
    
73.                         candiV[j] = check(candiV[j],index,i[0][j])
    
74.                 i = 'done'
    
75.             else:
    
76.                 for j in range(ysize):
    
77.                     if result[index][j] != ' ':
    
78.                         continue
    
79.                     for k in i[1::]:
    
80.                         if k[j]!=i[0][j]:
    
81.                             break
    
82.                     else:
    
83.                         result[index][j] = i[0][j]
    
84.                         count+=1
    
85.                         candiV[j] = check(candiV[j],index,i[0][j])
    
86. 
    
87.         for index in range(ysize):#再针对每一列，除个别变量外，代码相同
    
88.             i = candiV[index]
    
89.             if i == 'done':
    
90.                 continue
    
91.             elif len(i) == 1:
    
92.                 for j in range(xsize):
    
93.                     if result[j][index] == ' ':
    
94.                         count+=1
    
95.                         result[j][index] = i[0][j]
    
96.                         candiH[j] = check(candiH[j],index,i[0][j])
    
97.                 i = 'done'
    
98.             else:
    
99.                 for j in range(xsize):
    
100.                     if result[j][index] != ' ':
     
101.                         continue
     
102.                     for k in i[1::]:
     
103.                         if k[j]!=i[0][j]:
     
104.                             break
     
105.                     else:
     
106.                         result[j][index] = i[0][j]
     
107.                         count+=1
     
108.                         candiH[j] = check(candiH[j],index,i[0][j])
     
109.     for line in result:
     
110.         print(''.join(line))
     
111. 
     
112. print(solver('up.txt'))

复制代码

输出：

1. ___________________@@@_@@_______
   
2. __________________@@@@@@@@______
   
3. _________________@@@@@@@@@@_____
   
4. _________________@@@___@__@_____
   
5. _________________@@@@@_@@_@_____
   
6. _________________@@@@@_@@_@_____
   
7. ________________@@@@___@__@_____
   
8. _____________@@@@@@@@@@@@@@@____
   
9. ___________@@@@@@@@@@@@@@@@@@@__
   
10. __________@@@@@@_@@@@@@@@@@@@@@_
    
11. _________@@@@@@___@_@@@@@@@@@@@@
    
12. _________@@@@@@_____@_@@@@@@@@@@
    
13. ________@@@@@@@___@@__@_@@@@@@@@
    
14. ________@@@@@@_@_@_@@___@_@_@__@
    
15. ________@@@@@___@__@@@@________@
    
16. ________@@@@@_@@@@_@_@@@@_@_@_@_
    
17. ________@@@@@___@_@___@@@@@@@@__
    
18. _________@@@@@_@@__@___@@@@@@@@_
    
19. _________@@@@@@__@__@@___@_@@@__
    
20. __________@@@@@@_@@@__@@_____@__
    
21. ___________@@@@@@___@___@@@@@___
    
22. @___________@@@@@@__@@@_________
    
23. @@___________@@@@@@@___@@_______
    
24. @@@__@@@___@@@@@@@@@@_@@@@______
    
25. @@@@@@@@@_@@@@@@@@@@@@____@_____
    
26. @@@@@@@@@@@@@@@@@@@@@@____@_____
    
27. _@@@@@@@@@@@@@@@@@@@@@_@@@@_____
    
28. __@__@@@@@@@@@@@@@@@@@____@_____
    
29. ___@@__@@@@@@@@_@@@@@_____@_____
    
30. ________@@_____@@_____@@@@______
    
31. __________@@@@@__@@_@___@_______
    
32. ___________________@@@@@________

复制代码