백준 1484: 다이어트 [C/C++]

https://www.acmicpc.net/problem/1484

1484번: 다이어트

#include <iostream>
#include <algorithm>
#include <set>
using namespace std;
#define ll long long
static const auto fastio = []() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    return 0;
};
ll n, cnt;
int main() {
    cin >> n;
    ll s[50001];
    for (ll i = 1; i < 50001; i++) s[i] = i * i;
    for (ll i = 1; i < 50001; i++) {
        auto iter1 = lower_bound(s + 1, s + i, i * i - n);
        auto iter2 = upper_bound(s + 1, s + i, i * i - n);
        if (*iter1 != *iter2) {
           cout << i << "\n"; cnt++;
        }
    }
    cout << (cnt == 0 ? "-1" : "");
    return 0;
}
 

간단한 이분탐색 문제입니다.

lower_bound와 upper_bound의 특성을 활용하여 풀었습니다.