fork download
  1. #include <bits/stdc++.h>
  2. signed main() {
  3. int n;
  4. std::cin >> n;
  5. long long a[n];
  6. for (int i = 0; i < n; ++i) {
  7. std::cin >> a[i];
  8. a[i] = a[i] * (a[i] * (a[i] + 1) + 1);
  9. }
  10. std::sort(a, a+n);
  11. int q;
  12. std::cin >> q;
  13. while(q--){
  14. long long x;
  15. std::cin >> x;
  16. int i = std::lower_bound(a, a+n, x) - a;
  17. if(i == n || a[i] != x)
  18. std::cout << "NO\n";
  19. else
  20. std::cout << "YES\n";
  21. }
  22. }
Success #stdin #stdout 0.01s 5320KB
stdin
5
1 5 9 13 2
3
13
3
155
stdout
NO
YES
YES