[BOJ] 개미

[BOJ] 개미

Problem URL : 개미

w, h = map(int, input().split())
x, y = map(int, input().split())
t = int(input())
x += t
y += t
x %= 2*w
if w < x < 2*w:
    x = 2*w - x
y %= 2*h
if h < y < 2*h:
    y = 2*h - y
print('{} {}'.format(x, y))

Comments

처음에 다음과 같이 풀었다가 시간 초과(TLE)가 났다.

dx = 1
dy = 1
w, h = map(int, input().split())
x, y = map(int, input().split())
t = int(input())
for i in range(t):
    x += dx
    y += dy
    if x == w or x == 0:
        dx = -dx
    if y == h or y == 0:
        dy = -dy
print('{} {}'.format(x, y))