blob: 9a711f1971ab0f1385610011bf661cdb6c0076d2 (
plain)
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
|
# Uses python3
import sys
def getPartition3(a):
if len(a) < 3:
return 0
sum = 0
for i in a:
sum = sum + i;
if sum % 3 != 0:
return 0
W = int(sum / 3)
values = optimalWeight2DArray(W, a)
count = 0
for i in range(1, len(a) + 1):
for w in range(1, W + 1):
if values[w][i] == W:
count = count + 1
if count >= 3:
return 1
else:
return 0
def optimalWeight2DArray(W, w2):
j = len(w2)
value = [[0 for _ in range(j + 1)] for _ in range(W + 1)]
for i in range(1, j + 1):
for w in range(1, W + 1):
value[w][i] = value[w][i - 1]
if w2[i - 1] <= w:
val = value[w - w2[i - 1]][i - 1] + w2[i - 1]
if value[w][i] < val:
value[w][i] = val
return value
if __name__ == '__main__':
input = sys.stdin.read()
n, *A = list(map(int, input.split()))
print(getPartition3(A))
|