提出 #68784567

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define repr(i, n) for (int i = n-1; i >=0; i--)
#define repsg(i, s, g) for (int i = s; i < (int)(g); i++)
using ll = long long;
using P = pair<int,int>;
using Graph = vector<vector<int>>;
using mint = modint1000000007;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);

  ll X; cin >> X;

  vector<ll> ans;
  // n**2 + n + X = k**2
  // 平方完成 + 4倍 + 整理
  // 4k**2 - (2n+1)**2 = 4X - 1
  // (2k-(2n+1))*(2k+(2n+1)) = 4X - 1

  // a * b = 4X -1 && a, b:奇数 && a, bは符号一緒
  // b - a = 2k + 2n +1 - 2k +2n +1 = 4n +2
  // b - a - 2 = 4n
  // ★ n = (b-a-2)/4　

  // 4X - 1の約数を列挙すれば、★から求まる
  ll Y = 4*X - 1;
  ll L = llabs(Y);
  for(ll d=1; d*d<=L; d++){
    if(L % d) continue;
    ll e = Y / d;
    
    ll a = (Y> 0) ?  d : -d;
    ll b = (Y> 0) ? (Y/ d) : -(Y/ d);
    ans.push_back((b - a - 2) / 4);

    // 36 = 6*6 みたいなときは片方だけ
    if (d != e) {
      ll a = (Y> 0) ?  e : -e;
      ll b = (Y> 0) ? (Y/ e) : -(Y/ e);
      ans.push_back((b - a - 2) / 4);
    }
  }

  sort(ans.begin(), ans.end());
  ans.erase(unique(ans.begin(), ans.end()), ans.end());

  for(auto a : ans) cout << a << ' ';
  cout << endl;
  
  return 0;
}

提出情報

ジャッジ結果