def can_form_count_sequence(n, counts):
# Sprawdzamy, czy istnieje wyliczanka pasująca do podanego ciągu
total_count = sum(counts)
# Sprawdzamy, czy liczba wskazań jest zgodna z zasadami
if total_count == 0:
return False
# Sprawdzamy, czy skrajne zabawki mają odpowiednią liczbę wskazań
if counts[0] > 0 and counts[1] == 0:
return False
if counts[-1] > 0 and counts[-2] == 0:
return False
# Sprawdzamy, czy liczba wskazań dla pozostałych zabawek jest zgodna
for i in range(1, n - 1):
if counts[i] > 0 and (counts[i - 1] == 0 and counts[i + 1] == 0):
return False
return True
def main():
import sys
input = sys.stdin.read
data = input().splitlines()
t = int(data[0]) # Liczba dni
results = []
index = 1
for _ in range(t):
n = int(data[index]) # Liczba zabawek
counts = list(map(int, data[index + 1].split())) # Liczba wskazań dla zabawek
index += 2
if can_form_count_sequence(n, counts):
results.append("TAK")
else:
results.append("NIE")
# Wyświetlenie wyników
print("\n".join(results))
if __name__ == "__main__":
main()
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 40 41 42 43 44 45 | def can_form_count_sequence(n, counts): # Sprawdzamy, czy istnieje wyliczanka pasująca do podanego ciągu total_count = sum(counts) # Sprawdzamy, czy liczba wskazań jest zgodna z zasadami if total_count == 0: return False # Sprawdzamy, czy skrajne zabawki mają odpowiednią liczbę wskazań if counts[0] > 0 and counts[1] == 0: return False if counts[-1] > 0 and counts[-2] == 0: return False # Sprawdzamy, czy liczba wskazań dla pozostałych zabawek jest zgodna for i in range(1, n - 1): if counts[i] > 0 and (counts[i - 1] == 0 and counts[i + 1] == 0): return False return True def main(): import sys input = sys.stdin.read data = input().splitlines() t = int(data[0]) # Liczba dni results = [] index = 1 for _ in range(t): n = int(data[index]) # Liczba zabawek counts = list(map(int, data[index + 1].split())) # Liczba wskazań dla zabawek index += 2 if can_form_count_sequence(n, counts): results.append("TAK") else: results.append("NIE") # Wyświetlenie wyników print("\n".join(results)) if __name__ == "__main__": main() |
English