提出 #68781140

---

ソースコード 拡げる

use proconio::input;

fn main() {
    input! {
        x: i64,
    };

    let mut ans = Vec::new();
    for k in 0..=30_000_000 {
        if let Some(n) = solve(x, k) {
            assert!(n >= 0);
            ans.push(n);
            ans.push(-n - 1);
        }
    }
    ans.sort_unstable();
    ans.dedup();

    println!("{}", ans.len());
    println!(
        "{}",
        ans.iter()
            .map(|x| x.to_string())
            .collect::<Vec<_>>()
            .join(" ")
    );
}

#[allow(unused)]
fn solve(x: i64, k: i64) -> Option<i64> {
    // n^2 + n + x = k^2

    assert!(k >= 0);

    if x > k.pow(2) {
        return None;
    }

    let mut low = 0_i64;
    let mut high = f64::sqrt((k.pow(2) - x) as f64) as i64 + 1;
    assert!(low.pow(2) + low + x <= k.pow(2));
    assert!(high.pow(2) + high + x > k.pow(2));
    while low + 1 < high {
        let mid = (low + high) / 2;
        if mid.pow(2) + mid + x <= k.pow(2) {
            low = mid;
        } else {
            high = mid;
        }
    }

    if low.pow(2) + low + x == k.pow(2) {
        Some(low)
    } else {
        None
    }
}

提出情報

ジャッジ結果