Openjudge 百练第4109题

 1 # -*- coding:utf-8 -*-
 2 
 3 def set_1(i, q):
 4     ‘‘‘ generate a i*i ARRAY for all relationships
 5     if there is a relation set 1 or 0
 6     return i*i ARRAY with 1 or 0
 7     ‘‘‘
 8     a = [0 for i in range(i*i)] 
 9     for j in range(len(q)):
10         n, m = q[j]
11         a[(n-1)*i+(m-1)] = 1
12         a[(m-1)*i+(n-1)] = 1
13     return a 
14     
15 def solve(i, q, r):
16     ‘‘‘ solve question
17     i is the number of people 
18     q is the set of questions
19     r is the set of relationships, the result of function set_1(); 
20     ‘‘‘
21     result = 0
22     for j in range(len(r)):
23         n, m = r[j]
24         for l in range(i):
25             if q[(n-1)*i+l] == 1 and q[(m-1)*i+l] == 1:
26                 result += 1
27         print(result)
28         result = 0
29         
30 def main():
31     d = [ 3, [3,2,2],
32              [1,3],
33              [2,3],
34              [1,2],
35              [1,3],
36              [4,3,2],
37              [1,2],
38              [2,3],
39              [1,4],
40              [2,4],
41              [1,3],
42              [5,2,1],
43              [1,2],
44              [1,4],
45              [3,4]
46         ]
47     for x in d:                         #Dispaly input 
48         print(x)
49     
50     loc = []
51     for m in range(1,len(d)):           #Get the index of every question  
52         if len(d[m])==3:
53             loc.append(m)
54 
55     for i in range(len(loc)):           #Sovle each question
56         pNum, qNum, aNum = d[loc[i]]    #slice out R and Q in d[]
57         t = loc[i]+1
58         R = d[t:t+qNum]
59         Q = d[t+qNum:t+qNum+aNum]       
60             
61         r_1 = set_1(pNum,R)             # set 1 for question
62         print(‘-------------------\nCase‘+str(i+1)+‘:‘)
63         solve(pNum, r_1, Q)
64             
65 if __name__==‘__main__‘:
66     main()
67     
68 ‘‘‘OUTPUT
69 3
70 [3, 2, 2]
71 [1, 3]
72 [2, 3]
73 [1, 2]
74 [1, 3]
75 [4, 3, 2]
76 [1, 2]
77 [2, 3]
78 [1, 4]
79 [2, 4]
80 [1, 3]
81 [5, 2, 1]
82 [1, 2]
83 [1, 4]
84 [3, 4]
85 -------------------
86 Case1:
87 1
88 0
89 -------------------
90 Case2:
91 1
92 1
93 -------------------
94 Case3:
95 0
96 ‘‘‘