fork download
copy 
  1. # your code goes here
  2. # cook your dish here
  3. t = int(input())
  4. for _ in range(t):
  5.     n = int(input())
  6.     a = list(map(int, input().split()))
  7.     s = 0
  8.     good = True
  9.     for i in a:
  10.         s += i
  11.         if s % 3 == 0:
  12.             good = False
  13.             break
  14.     if good:
  15.         print("Yes")
  16.         continue
  17.     flag = False
  18.     for k in range(1, n + 1):
  19.         s = 0
  20.         good = True
  21.         for i in range(k - 1, -1, -1):
  22.             s += a[i]
  23.             if s % 3 == 0:
  24.                 good = False
  25.                 break
  26.         if not good:
  27.             continue
  28.         for i in range(k, n):
  29.             s += a[i]
  30.             if s % 3 == 0:
  31.                 good = False
  32.                 break
  33.         if good:
  34.             flag = True
  35.             break
  36.  
  37.     print("yes" if flag else "no")
  38.  
Success #stdin #stdout 0.1s 14072KB
comments (?)
stdin
copy 
6
4
1 1 2 1
2
1 2
5
1 2 2 2 1
4
1 2 1 2
5
1 1 2 1 1
1
2
stdout
copy 
Yes
no
yes
no
no
Yes
https://ideone.com/KfYqYP
language:
Python 3 (python  3.12)
created:
9 hours ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!