31-SeongHoonC

[SeongHoonC]

🔗 문제 링크

성원이의 다이어트
다이어트

✔️ 소요된 시간

30분

✨ 수도 코드

오랜만에 풀다보니 투포인터를 떠올리는데 좀 걸렸네요.

head = 2
tail = 1

int x = (head + tail) * (head - tail) = head ^ 2 - tail ^ 2 이기 때문에

1. x 가 G 보다 크다면 tail 증가
2. x 가 G 보다 작다면 head 증가
3. x 가 G 와 같다면 리스트에 추가합니다.

[tgyuuAn]

아... 이거 투포인터로도 해결할 수 있군요....?

저는 집합 자료구조 써서 그냥 계속 있는지 탐색했어요...

import math
import sys

def input(): return sys.stdin.readline().rstrip()

G = int(input())
square_numbers_set = set()
square_numbers_list = list()
prev = 0
for step in range(1,100_001):
    now = step**2
    if (now-prev) > G: break
    square_numbers_set.add(now)
    square_numbers_list.append(now)
    prev = now

flag = True
for now in square_numbers_list:
    if now - G in square_numbers_set: 
        flag = False
        print(int(math.sqrt(now)))

if flag: print(-1)
    
# G = 현재 몸무게 ^2 - 원래 몸무게 ^2
# X + G = Y
# G = Y - X

와 투포인터로도 풀 수 있었군요.. 왜 못떠올렸지....