Завзятий колекціонер метеликів (Розбір)

Номери видів метеликів у колекції будемо зберігати у множині $s$. Тоді перевірка того, чи присутній метелик виду $x$ у Андрія Сергійовича, зводиться до перевірки належності числа $x$ множині $s$.

Реалізація алгоритму

Усі номери видів метеликів із колекції зберігаємо у множині $s$.

set<int> s;

Читаємо вхідні дані.

scanf("%d", &n);
for (i = 0; i < n; i++)
{
  scanf("%d", &x);
  s.insert(x);
}

Опрацьовуємо $m$ запитів.

scanf("%d", &m);
for (i = 0; i < m; i++)
{

Перевіряємо, чи є в колекції метелик виду $x$. Відповідь буде ствердною, якщо число $x$ належить множині $s$.

  scanf("%d", &x);
  if (s.find(x) != s.end()) puts("YES");
  else puts("NO");
}

Python реалізація

Читаємо вхідні дані.

n = int(input())
lst = list(map(int, input().split()))
m = int(input())
q = list(map(int, input().split()))

Усі номери видів метеликів із колекції зберігаємо у множині $s$.

s = set(lst)

Обробляємо $m$ запитів.

for x in q:

Перевіряємо, чи є в колекції метелик виду $x$. Відповідь буде ствердною, якщо число $x$ належить множині $s$.

  if x in s:
    print("YES")
  else:
    print("NO")