나의 코드
def solution(n, words):
answer = [words[0]]
x, y = 1, 1
for i in range(1, len(words)):
x += 1
if x == n + 1:
x, y = 1, y + 1
if words[i] in answer:
return [x, y]
elif answer[-1][-1] != words[i][0]:
return [x, y]
else:
answer.append(words[i])
return [0, 0]
결과 인덱스를 x, y로 따로 연산해줬는데 그냥 나눠서 바로 구할 수 있었다.
효율 코드
return [(p%n)+1, (p//n)+1]
'코딩 테스트 (Coding Test)' 카테고리의 다른 글
[프로그래머스] 예상 대진표 - 파이썬 풀이 (0) | 2023.01.17 |
---|---|
[프로그래머스] *N개의 최소공배수 (0) | 2023.01.17 |
[프로그래머스] 짝지어 제거하기 (0) | 2023.01.17 |
[프로그래머스] 다음 큰 숫자 (0) | 2023.01.16 |
[프로그래머스] 피보나치 수 (0) | 2023.01.16 |