files

Author: harshitgupta028

Itertools/Iterables and Iterators.py

@@ -0,0 +1,20 @@
+from itertools import combinations
+j=1
+count=0
+lst=[]
+ind=[]
+le=int(input())
+ls=list(map(str, input().split()))
+for i in ls:
+    if i == "a":
+        ind.append(j)
+        j=j+1
+sep=int(input())
+for i in range(1,le+1):
+    lst.append(i)
+total=len(list(combinations(lst,sep)))
+for i in list(combinations(lst,sep)):
+    if i[0] in ind:
+        count = count+1
+probablity=(count)/total        
+print("%.4f" %probablity)

Itertools/Maximize It!.py

@@ -0,0 +1,25 @@
+from itertools import product
+Tc,M=map(int, input().split())
+lis=[]
+lst=(list(map(int, input().split())) [1:] for _ in range(Tc))
+z=map(lambda x: sum(i**2 for i in x)%M, product(*lst))
+print(max(z))
+
+===================================================================================================
+Alternate
+===================================================================================================
+
+from itertools import product
+Tc,M=map(int, input().split())
+lis=[]
+lst=(list(map(int, input().split())) [1:] for _ in range(Tc))
+#print(*lst)
+
+z=list(product(*lst))
+
+for i in z:
+    s=0
+    for j in i:
+        s=s+(j**2)
+    lis.append(s%M)
+print(max(lis))