-
Notifications
You must be signed in to change notification settings - Fork 0
/
prob24_majority.py
39 lines (30 loc) · 1023 Bytes
/
prob24_majority.py
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
33
34
35
36
37
38
39
def majority(n,k,array):
split = array.split()
mid = n/2.0
tempDict = {}
for i in range(n):
num = split[i]
#print num,"num"
tempDict[num] = tempDict.get(num,0) + 1
#print tempDict[num],"current count"
if tempDict[num] > mid:
return num
elif i == n-1:
return -1
def main():
#infile = open("C:/Users/Kaiya/Desktop/scratch_rosalind.txt","rU")
infile = open("C:/Users/Kaiya/Desktop/rosalind_maj.txt","rU")
kn = infile.readline()
lines = infile.readlines()
infile.close()
knList = kn.split()
k = int(knList[0])
n = int(knList[1])
lines = [i.strip() for i in lines]
#outfile = open("C:/Users/Kaiya/Desktop/scratch_rosalind_2.txt","w")
outfile = open("C:/Users/Kaiya/Desktop/rosalind_maj_ans.txt","w")
for array in lines:
print >>outfile, majority(n,k,array),
outfile.close()
if __name__ == "__main__":
main()