Задача к ЕГЭ по информатике на тему «прочие прототипы» №8

На вход программы поступает последовательность из N  натуральных чисел, все числа в последовательности различны. Рассматриваются всевозможные непустые подмножества, состоящие из элементов последовательности. Необходимо найти количество подмножеств длины 3, в которых сумма элементов кратна 3.

Пример входных данных:

Первая строка входного файла содержит число N  – общее количество чисел. Каждая из следующих N  строк содержит натуральные числа, не превышающих 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)

Ответ: 357
Оцените статью
Я решу все!