취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 1484번 문제인 다이어트이다.
문제는 아래 링크를 확인하자.

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

50000을 넘는 자연수 X에 대하여 X와 X-1의 차는 항상 10만을 넘음을 관찰하자.

위 관찰을 이용하면, 1부터 50000까지의 제곱수를 오름차순으로 구해나가면서 각 제곱수에서 G를 뺀 수가 앞서 등장한 적이 있는지를 체크하는 것으로 문제를 해결할 수 있음을 확인할 수 있다. 이는 set 등의 자료구조를 이용하면 간단하게 해낼 수 있다.

아래는 제출한 소스코드이다.

#include <iostream>
#include <set>
using namespace std;
typedef long long ll;

ll G;
set<ll> st;
bool chk;

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

	cin >> G;

	for (ll i = 1; i < 50001LL; i++) {
		if (st.count(i * i - G)) cout << i << '\n', chk = 1;
		st.insert(i * i);
	}

	if (!chk) cout << -1;
}