-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtwo_equal_sum_arrays.py
53 lines (31 loc) · 991 Bytes
/
two_equal_sum_arrays.py
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
46
47
48
49
50
51
52
53
from sys import stdin
t = int(stdin.readline())
for _ in range(t):
A,B,C = map(int,stdin.readline().strip().split(" "))
input_array = []
for i in range(A):
input_array.append(1)
for i in range(B):
input_array.append(2)
for i in range(C):
input_array.append(3)
left_sum = 0
for i in input_array:
left_sum+=i
right_sum = 0
N= len(input_array)
if left_sum % 2 != 0:
print('NO')
else:
for i in range(N-1,-1,-1):
if left_sum == right_sum:
break
right_sum+=input_array[i]
left_sum-=input_array[i]
if right_sum > left_sum:
left_sum+=input_array[i]
right_sum-=input_array[i]
if left_sum == right_sum:
print("YES")
else:
print("NO")