Advertisement
cosenza987

asdas

Jul 7th, 2023
738
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.02 KB | None | 0 0
  1. // Date / Time: 2023-07-07 11:54:23
  2.  
  3. //Слава Україні, Героям слава
  4.  
  5. #include <bits/stdc++.h>
  6.  
  7. using namespace std;
  8.  
  9. int main() {
  10.     ios_base::sync_with_stdio(false);
  11.     cin.tie(0);
  12.     unordered_set<long long> st;
  13.     for(long long i = 2; i <= (int)1e6; i++) {
  14.         long long cur = i * i, s = 1 + i + i * i;
  15.         while(true) {
  16.             long long tmp = cur * i;
  17.             if(cur != tmp / i) break;
  18.             cur = tmp;
  19.             if(s > (long long)1e18 - cur) break;
  20.             s += cur;
  21.             st.insert(s);
  22.         }
  23.     }
  24.     int t;
  25.     cin >> t;
  26.     while(t--) {
  27.         long long n;
  28.         cin >> n;
  29.         if(n < 7) {
  30.             cout << "NO\n";
  31.             continue;
  32.         }
  33.         long long k = sqrtl(n);
  34.         if(k * (k + 1) == n - 1) {
  35.             cout << "YES\n";
  36.             continue;
  37.         }
  38.         if(st.find(n) != st.end()) {
  39.             cout << "YES\n";
  40.         } else {
  41.             cout << "NO\n";
  42.         }
  43.     }
  44.     return 0;
  45. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement