На вход программы поступает последовательность из натуральных чисел, все числа в последовательности различны. Рассматриваются всевозможные непустые подмножества, состоящие из элементов последовательности. Необходимо найти количество подмножеств длины 3, в которых сумма элементов кратна 3.
Пример входных данных:
Первая строка входного файла содержит число – общее количество чисел. Каждая из следующих
строк содержит натуральные числа, не превышающих 10 000.
Пример входного файла
3
1
2
3
Для указанных данных ответом будет являться, {1, 2, 3}, то есть 1.
Решение 1
f = open(’27A.txt’)
n = int(f.readline())
nums = []
for i in range(n):
nums.append(int(f.readline()))
ans = 0
for i in range(2**n):
t = i
mn = []
for j in range(n):
if t % 2 == 1:
mn.append(nums[j])
t //= 2
if len(mn) == 3:
if sum(mn) % 3 == 0:
ans += 1
print(ans)
Решение 2
from itertools import combinations
f = open(’27A.txt’)
n = int(f.readline())
nums = []
for i in range(n):
nums.append(int(f.readline()))
ans = 0
for i in combinations(nums, 3):
if sum(i) % 3 == 0:
ans += 1
print(ans)