COVID Pandemic and Long Queue

Author: nirmalnishant645

Diff for: CodeChef/COVID-Pandemic-and-Long-Queue.py

@@ -0,0 +1,66 @@
+'''
+Due to the COVID pandemic, people have been advised to stay at least 6 feet away from any other person.
+Now, people are lining up in a queue at the local shop and it is your duty to check whether they are all following this advice.
+
+There are a total of N spots (numbered 1 through N) where people can stand in front of the local shop.
+The distance between each pair of adjacent spots is 1 foot.
+Each spot may be either empty or occupied; you are given a sequence A1,A2,…,AN, where for each valid i, Ai=0 means that the i-th spot is empty, while Ai=1 means that there is a person standing at this spot.
+It is guaranteed that the queue is not completely empty.
+
+For example, if N=11 and the sequence A is (0,1,0,0,0,0,0,1,0,0,1), then this is a queue in which people are not following the advice because there are two people at a distance of just 3 feet from each other.
+
+You need to determine whether the people outside the local shop are following the social distancing advice or not.
+As long as some two people are standing at a distance smaller than 6 feet from each other, it is bad and you should report it, since social distancing is not being followed.
+
+Input
+The first line of the input contains a single integer T denoting the number of test cases. The description of T test cases follows.
+The first line of each test case contains a single integer N.
+The next line contains N space-separated integers A1,A2,…,AN.
+Output
+For each test case, print a single line containing the string "YES" if social distancing is being followed or "NO" otherwise (without quotes).
+
+Constraints
+1≤T≤100
+1≤N≤100
+0≤Ai≤1 for each valid i
+at least one spot is occupied
+
+Subtasks
+Subtask #1 (100 points): original constraints
+
+Example Input
+3
+3
+1 0 1
+7
+1 0 0 0 0 0 1
+11
+0 1 0 0 0 0 0 1 0 0 1
+Example Output
+NO
+YES
+NO
+Explanation
+Example case 1: The first and third spots are occupied and the distance between them is 2 feet.
+
+Example case 2: The first and seventh spots are occupied and the distance between them is 6 feet.
+'''
+T = int(input())
+while T:
+    N = int(input())
+    count = 5
+    flag = True
+    arr = list(map(int, input().split()))
+    for i in arr:
+        if i and count >= 5:
+            count = 0
+        elif i and count < 5:
+            flag = False
+            break
+        elif not i:
+            count += 1
+    if flag:
+        print('YES')
+    else:
+        print('NO')
+    T -= 1