COVID-Pandemic-and-Long-Queue.py

'''
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