[LeetCode] Two Sum (Python)

🗓️ 문제 설명

• 정수 배열 nums와 정수 target이 주어졌을 때, target과 nums 내의 두 숫자가 같아지는 숫자의 인덱스를 반환해주세요.
• 제한 사항
  • 동일한 요소를 두 번 사용할 수 없음
  • 각 입력에 정확히 하나의 솔루션이 있음
• 입출력 예시

nums	target	result
[2, 7, 11, 15]	9	[0, 1]
[3, 2, 4]	6	[1, 2]
[3, 3]	6	[0, 1]

 

💻 코드

최종 코드

 

코드 풀이

 

다른 사람 풀이

풀이 1번

1. d = {} : 숫자와 인덱스를 담을 딕셔너리 생성
2. for i, j in enumerate(nums) : nums에서 인덱스와 값을 꺼내는 반복문 실행
3. r = target - j : target에서 리스트 nums에서 꺼낸 값을 뺀 후 저장
4. if r in d : return [d[r], i] : 만약 3번에서 저장한 r이 1번에서 생성한 딕셔너리 d에 있다면 r의 인덱스와 현재 값의 인덱스를 반환
5. d[j] = i : 딕셔너리 d에 값과 인덱스를 key-value 형식으로 저장